Fix main app migration pathway

This commit is contained in:
Sam Chow 2018-06-21 14:48:40 -04:00
parent 561522c6d3
commit aff1a08a83
4 changed files with 23 additions and 8 deletions

View file

@ -5,13 +5,19 @@ from alembic.script import ScriptDirectory
from alembic.environment import EnvironmentContext
from alembic.migration import __name__ as migration_name
def run_alembic_migration(db_uri, log_handler=None):
def run_alembic_migration(db_uri, log_handler=None, setup_app=True):
if log_handler:
logging.getLogger(migration_name).addHandler(log_handler)
config = Config()
config.set_main_option("script_location", "data:migrations")
config.set_main_option("db_uri", db_uri)
if setup_app:
config.set_main_option('alembic_setup_app', 'True')
else:
config.set_main_option('alembic_setup_app', '')
script = ScriptDirectory.from_config(config)
def fn(rev, context):