From: Marcin Kuzminski <marcin@python-works.com>
Date: Wed, 8 Feb 2012 03:40:51 +0000 (+0200)
Subject: try to get engine from Base if it's defined
X-Git-Tag: rel_0_2_2~10^2~3
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=2b3b6d7c6aa066d43608a9654986ea7def677eb8;p=thirdparty%2Fsqlalchemy%2Falembic.git

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 bfc789c4..90b65d01 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