Apply fix from Alexis-benoist#80

eveith · Oct 6, 2021 · 6855f88 · 6855f88
1 parent 4f9dbda
commit 6855f88
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/eralchemy/sqla.py b/eralchemy/sqla.py
@@ -47,7 +47,14 @@ def table_to_intermediary(table):
     """Transform an SQLAlchemy Table object to it's intermediary representation. """
     return Table(
         name=table.fullname,
-        columns=[column_to_intermediary(col) for col in table.c._data.values()]
+        columns=[
+            column_to_intermediary(col)
+            for col in getattr(
+                table.c,
+                "_colset",
+                getattr(table.c, "_data", {}).values()
+            )
+        ]
     )