createerm.py
branchdevel
changeset 251 3caa803a2dec
parent 92 f479738b4879
child 294 0e75bd39767d
--- a/createerm.py	Sat Mar 24 01:05:34 2012 +0100
+++ b/createerm.py	Sat Mar 24 01:06:51 2012 +0100
@@ -1,13 +1,15 @@
 from iro.model import schema
 
-from sqlalchemy.orm import class_mapper
-tables = []
-for attr in schema.__tables__:
-    if attr[0] == '_': continue
+tables = schema.Base.metadata.sorted_tables
+
+tables_cls={}
+for i in dir(schema):
+    if i.startswith("__") or i == "Base":
+        continue
     try:
-        cls = getattr(schema, attr)
-        tables.append(class_mapper(cls))
-    except:
+        a = getattr(schema,i)
+        tables_cls[a.__tablename__] = a
+    except AttributeError:
         pass