diff options
Diffstat (limited to 'migrations/env.py')
| -rw-r--r-- | migrations/env.py | 22 |
1 files changed, 13 insertions, 9 deletions
diff --git a/migrations/env.py b/migrations/env.py index 7215fefe..4f31b751 100644 --- a/migrations/env.py +++ b/migrations/env.py @@ -9,15 +9,18 @@ config = context.config # Interpret the config file for Python logging. # This line sets up loggers basically. -#fileConfig("./alembic.ini") +# fileConfig("./alembic.ini") # add your model's MetaData object here # for 'autogenerate' support # from myapp import mymodel # target_metadata = mymodel.Base.metadata from flask import current_app -config.set_main_option('sqlalchemy.url', current_app.config.get('SQLALCHEMY_DATABASE_URI')) -target_metadata = current_app.extensions['migrate'].db.metadata + +config.set_main_option( + "sqlalchemy.url", current_app.config.get("SQLALCHEMY_DATABASE_URI") +) +target_metadata = current_app.extensions["migrate"].db.metadata # other values from the config, defined by the needs of env.py, @@ -25,6 +28,7 @@ target_metadata = current_app.extensions['migrate'].db.metadata # my_important_option = config.get_main_option("my_important_option") # ... etc. + def run_migrations_offline(): """Run migrations in 'offline' mode. @@ -43,6 +47,7 @@ def run_migrations_offline(): with context.begin_transaction(): context.run_migrations() + def run_migrations_online(): """Run migrations in 'online' mode. @@ -52,18 +57,17 @@ def run_migrations_online(): """ connectable = engine_from_config( config.get_section(config.config_ini_section), - prefix='sqlalchemy.', - poolclass=pool.NullPool) + prefix="sqlalchemy.", + poolclass=pool.NullPool, + ) with connectable.connect() as connection: - context.configure( - connection=connection, - target_metadata=target_metadata - ) + context.configure(connection=connection, target_metadata=target_metadata) with context.begin_transaction(): context.run_migrations() + if context.is_offline_mode(): run_migrations_offline() else: |