diff options
Diffstat (limited to 'mediagoblin/db/migrations/env.py')
-rw-r--r-- | mediagoblin/db/migrations/env.py | 14 |
1 files changed, 3 insertions, 11 deletions
diff --git a/mediagoblin/db/migrations/env.py b/mediagoblin/db/migrations/env.py index 712b6164..c5f88219 100644 --- a/mediagoblin/db/migrations/env.py +++ b/mediagoblin/db/migrations/env.py @@ -47,22 +47,14 @@ def run_migrations_online(): and associate a connection with the context. """ - engine = engine_from_config( - config.get_section(config.config_ini_section), - prefix='sqlalchemy.', - poolclass=pool.NullPool) - - connection = engine.connect() + connection = config.attributes["session"].get_bind() context.configure( connection=connection, target_metadata=target_metadata ) - try: - with context.begin_transaction(): - context.run_migrations() - finally: - connection.close() + with context.begin_transaction(): + context.run_migrations() if context.is_offline_mode(): run_migrations_offline() |