diff --git a/alembic/env.py b/alembic/env.py index 779be371..3b05e9a6 100755 --- a/alembic/env.py +++ b/alembic/env.py @@ -15,12 +15,8 @@ # License for the specific language governing permissions and limitations # under the License. from __future__ import with_statement - -import os import sys sys.path.append("./") - - from alembic import context from sqlalchemy import engine_from_config, pool from logging.config import fileConfig @@ -44,6 +40,7 @@ target_metadata = None # my_important_option = config.get_main_option("my_important_option") # ... etc. + def run_migrations_offline(): """Run migrations in 'offline' mode. @@ -62,23 +59,21 @@ def run_migrations_offline(): with context.begin_transaction(): context.run_migrations() + def run_migrations_online(): """Run migrations in 'online' mode. In this scenario we need to create an Engine - and associate a connection with the context. - - """ + and associate a connection with the context.""" engine = engine_from_config( - config.get_section(config.config_ini_section), - prefix='sqlalchemy.', - poolclass=pool.NullPool) + config.get_section(config.config_ini_section), + prefix='sqlalchemy.', + poolclass=pool.NullPool) connection = engine.connect() context.configure( - connection=connection, - target_metadata=target_metadata - ) + connection=connection, + target_metadata=target_metadata) try: with context.begin_transaction(): @@ -86,6 +81,7 @@ def run_migrations_online(): finally: connection.close() + if context.is_offline_mode(): run_migrations_offline() else: