try to get engine from Base if it's defined

diff --git a/alembic/templates/pylons/env.py b/alembic/templates/pylons/env.py
index bfc789c47a081d593587868296df6e3fbd8737bb..90b65d01ef7bc0dd145714af952e3ae83b9d302e 100644 (file)
--- 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