From: Mike Bayer <mike_mp@zzzcomputing.com>
Date: Sat, 20 Sep 2014 22:33:00 +0000 (-0400)
Subject: - compat for sqla 0.7
X-Git-Tag: rel_0_7_0~67
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=3f1ba114d6d8031ee15a3cf5d1c55238b1d662ad;p=thirdparty%2Fsqlalchemy%2Falembic.git

- compat for sqla 0.7
---

diff --git a/alembic/autogenerate/compare.py b/alembic/autogenerate/compare.py
index 1f9cf77e..dbc7b9f3 100644
--- a/alembic/autogenerate/compare.py
+++ b/alembic/autogenerate/compare.py
@@ -69,7 +69,8 @@ def _compare_tables(conn_table_names, metadata_table_names,
             event.listen(
                 t,
                 "column_reflect",
-                autogen_context['context'].impl.autogen_column_reflect)
+                autogen_context['context'].impl.
+                _compat_autogen_column_reflect(inspector))
             inspector.reflecttable(t, None)
         if _run_filters(t, tname, "table", True, None, object_filters):
             diffs.append(("remove_table", t))
@@ -87,7 +88,8 @@ def _compare_tables(conn_table_names, metadata_table_names,
             event.listen(
                 t,
                 "column_reflect",
-                autogen_context['context'].impl.autogen_column_reflect)
+                autogen_context['context'].impl.
+                _compat_autogen_column_reflect(inspector))
             inspector.reflecttable(t, None)
         conn_column_info[(s, tname)] = t
 
diff --git a/alembic/ddl/impl.py b/alembic/ddl/impl.py
index 9f75acc6..3e4f6e24 100644
--- a/alembic/ddl/impl.py
+++ b/alembic/ddl/impl.py
@@ -231,6 +231,15 @@ class DefaultImpl(with_metaclass(ImplMeta)):
                                         metadata_indexes):
         pass
 
+    def _compat_autogen_column_reflect(self, inspector):
+        if util.sqla_08:
+            return self.autogen_column_reflect
+        else:
+            def adapt(table, column_info):
+                return self.autogen_column_reflect(
+                    inspector, table, column_info)
+            return adapt
+
     def autogen_column_reflect(self, inspector, table, column_info):
         """A hook that is attached to the 'column_reflect' event for when
         a Table is reflected from the database during the autogenerate