diff options
| author | Marcin Kuzminski <marcin@python-works.com> | 2012-02-08 05:40:51 +0200 |
|---|---|---|
| committer | Marcin Kuzminski <marcin@python-works.com> | 2012-02-08 05:40:51 +0200 |
| commit | 2b3b6d7c6aa066d43608a9654986ea7def677eb8 (patch) | |
| tree | 5cd9357aca6827c5517d7b3099d81e831418b4bc | |
| parent | 9afb841880cbe8bb241b661f5536a0e4800bf7ea (diff) | |
| download | alembic-2b3b6d7c6aa066d43608a9654986ea7def677eb8.tar.gz | |
try to get engine from Base if it's defined
| -rw-r--r-- | alembic/templates/pylons/env.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/alembic/templates/pylons/env.py b/alembic/templates/pylons/env.py index bfc789c..90b65d0 100644 --- a/alembic/templates/pylons/env.py +++ b/alembic/templates/pylons/env.py @@ -58,7 +58,13 @@ def run_migrations_online(): """ - connection = meta.engine.connect() + if hasattr(meta, 'engine'): + connection = meta.engine.connect() + elif hasattr(meta, 'Base'): + connection = meta.Base.metadata.bind.connect() + else: + raise Exception(('Unable to fetch connection from meta %s' + 'Try to define engine or Base in meta file') % meta) context.configure( connection=connection, target_metadata=target_metadata |