mirror of
https://gitlab.archlinux.org/archlinux/aurweb.git
synced 2025-02-03 10:43:03 +01:00
aecb649473
First off: This commit changes the default development database backend to mysql. sqlite, however, is still completely supported with the caveat that a user must now modify config.dev to use the sqlite backend. While looking into this, it was discovered that our SQLAlchemy backend for mysql (mysql-connector) completely broke model attributes when we switched to utf8mb4_bin (binary) -- it does not correct the correct conversion to and from binary utf8mb4. The new, replacement dependency mysqlclient does. mysqlclient is also recommended in SQLAlchemy documentation as the "best" one available. The mysqlclient backend uses a different exception flow then sqlite, and so tests expecting IntegrityError has to be modified to expect OperationalError from sqlalchemy.exc. So, for each model that we define, check keys that can't be NULL and raise sqlalchemy.exc.IntegrityError if we have to. This way we keep our exceptions uniform. Signed-off-by: Kevin Morris <kevr@0cost.org>
64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
import argparse
|
|
|
|
import alembic.command
|
|
import alembic.config
|
|
|
|
import aurweb.db
|
|
import aurweb.schema
|
|
|
|
|
|
def feed_initial_data(conn):
|
|
conn.execute(aurweb.schema.AccountTypes.insert(), [
|
|
{'ID': 1, 'AccountType': 'User'},
|
|
{'ID': 2, 'AccountType': 'Trusted User'},
|
|
{'ID': 3, 'AccountType': 'Developer'},
|
|
{'ID': 4, 'AccountType': 'Trusted User & Developer'},
|
|
])
|
|
conn.execute(aurweb.schema.DependencyTypes.insert(), [
|
|
{'ID': 1, 'Name': 'depends'},
|
|
{'ID': 2, 'Name': 'makedepends'},
|
|
{'ID': 3, 'Name': 'checkdepends'},
|
|
{'ID': 4, 'Name': 'optdepends'},
|
|
])
|
|
conn.execute(aurweb.schema.RelationTypes.insert(), [
|
|
{'ID': 1, 'Name': 'conflicts'},
|
|
{'ID': 2, 'Name': 'provides'},
|
|
{'ID': 3, 'Name': 'replaces'},
|
|
])
|
|
conn.execute(aurweb.schema.RequestTypes.insert(), [
|
|
{'ID': 1, 'Name': 'deletion'},
|
|
{'ID': 2, 'Name': 'orphan'},
|
|
{'ID': 3, 'Name': 'merge'},
|
|
])
|
|
|
|
|
|
def run(args):
|
|
aurweb.config.rehash()
|
|
|
|
# Ensure Alembic is fine before we do the real work, in order not to fail at
|
|
# the last step and leave the database in an inconsistent state. The
|
|
# configuration is loaded lazily, so we query it to force its loading.
|
|
if args.use_alembic:
|
|
alembic_config = alembic.config.Config('alembic.ini')
|
|
alembic_config.get_main_option('script_location')
|
|
alembic_config.attributes["configure_logger"] = False
|
|
|
|
engine = aurweb.db.get_engine(echo=(args.verbose >= 1))
|
|
aurweb.schema.metadata.create_all(engine)
|
|
feed_initial_data(engine.connect())
|
|
|
|
if args.use_alembic:
|
|
alembic.command.stamp(alembic_config, 'head')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser(
|
|
prog='python -m aurweb.initdb',
|
|
description='Initialize the aurweb database.')
|
|
parser.add_argument('-v', '--verbose', action='count', default=0,
|
|
help='increase verbosity')
|
|
parser.add_argument('--no-alembic',
|
|
help='disable Alembic migrations support',
|
|
dest='use_alembic', action='store_false')
|
|
args = parser.parse_args()
|
|
run(args)
|