Support enums with no labels. Fixes #31.

perseas · Jun 22, 2012 · 86dd5db · 86dd5db
1 parent 785b184
commit 86dd5db
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pyrseas/dbobject/dbtype.py b/pyrseas/dbobject/dbtype.py
@@ -240,6 +240,8 @@ def _from_catalog(self):
             elif kind == 'e':
                 del dbtype.type
                 self[(sch, typ)] = Enum(**dbtype.__dict__)
+                if not hasattr(self[(sch, typ)], 'labels'):
+                    self[(sch, typ)].labels = {}
             elif kind == 'c':
                 del dbtype.type
                 self[(sch, typ)] = Composite(**dbtype.__dict__)