a small fix

omersaeed · Oct 10, 2012 · f21534b · f21534b
1 parent dd7ec58
commit f21534b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/spire/schema/__init__.py b/spire/schema/__init__.py
@@ -4,9 +4,6 @@
 from sqlalchemy.orm.exc import MultipleResultsFound, NoResultFound
 from sqlalchemy.schema import PrimaryKeyConstraint, UniqueConstraint
 
-from spire.schema.fields import (Array, Boolean, Date, DateTime, Decimal, Email,
+from spire.schema.fields import *
-    Enumeration, Float, ForeignKey, Hstore, Identifier, Integer, Serialized, 
-    Text, Time, Token, UUID)
-
 from spire.schema.model import *
 from spire.schema.schema import *
diff --git a/spire/schema/fields.py b/spire/schema/fields.py
@@ -8,6 +8,13 @@
 
 from spire.util import uniqid
 
+__all__ = ('Array', 'ArrayType', 'Boolean', 'BooleanType', 'Date', 'DateType',
+    'DateTime', 'DateTimeType', 'Decimal', 'DecimalType', 'Email', 'EmailType',
+    'Enumeration', 'EnumerationType', 'Float', 'FloatType', 'ForeignKey',
+    'Hstore', 'HstoreType', 'Identifier', 'Integer', 'IntegerType', 'Serialized',
+    'SerializedType', 'Text', 'TextType', 'Time', 'TimeType', 'Token', 'TokenType',
+    'UUID', 'UUIDType')
+
 class TypeDecorator(TypeDecorator):
     def __repr__(self):
         return '%s()' % type(self).__name__