external/aurweb
0
0
Fork 0
mirror of https://gitlab.archlinux.org/archlinux/aurweb.git synced 2025-02-03 10:43:03 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Set up Alembic for database migrations

Browse source 
Signed-off-by: Lukas Fleischer <lfleischer@archlinux.org>
This commit is contained in:
Frédéric Mangano-Tarumi 2020-02-22 22:31:26 +01:00 • committed by Lukas Fleischer
parent 7188743fc3
commit a8a1f74a92
8 changed files with 252 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

9
aurweb/initdb.py
View file

@ -1,6 +1,8 @@
import aurweb.db
import aurweb.schema
import alembic.command
import alembic.config
import argparse
import sqlalchemy
@ -31,10 +33,17 @@ def feed_initial_data(conn):
def run(args):
# 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.
alembic_config = alembic.config.Config('alembic.ini')
alembic_config.get_main_option('script_location')
engine = sqlalchemy.create_engine(aurweb.db.get_sqlalchemy_url(),
echo=(args.verbose >= 1))
aurweb.schema.metadata.create_all(engine)
feed_initial_data(engine.connect())
alembic.command.stamp(alembic_config, 'head')
if __name__ == '__main__':