better postgres support, interactive mode provides a bit more information before you play with it.

percious · percious · commit 1b6a9d719c7e · 2009-09-02T15:39:56.000-06:00
diff --git a/sqlautocode/constants.py b/sqlautocode/constants.py
@@ -47,6 +47,10 @@ def getModel(metadata):
     model = Model()
 """
 
+PG_IMPORT = """\
+from sqlalchemy.databases.postgres import *
+"""
+
 FOOTER_Z3C = """
     return model
 """
@@ -102,6 +106,7 @@ def getModel(metadata):
 try:
     from IPython.Shell import IPShellEmbed
     print 'Success! Type <ctrl-d> to exit.'
+    print 'Available models:%%s'%%%s
     ipshell = IPShellEmbed()
     ipshell()
 except:
diff --git a/sqlautocode/declarative.py b/sqlautocode/declarative.py
@@ -113,6 +113,8 @@ def __repr__(self):
         if not isinstance(engine, basestring):
             engine = str(engine.url)
         s.write(constants.HEADER_DECL%engine)
+        if 'postgres' in engine:
+            s.write(constants.PG_IMPORT)
         for table in tables:
             s.write('%s = %s\n\n'%(table.name, self._table_repr(table)))
         
@@ -123,7 +125,7 @@ def __repr__(self):
         if self.config.example or self.config.interactive:
             s.write(constants.EXAMPLE_DECL%models[0].__name__)
         if self.config.interactive:
-            s.write(constants.INTERACTIVE)
+            s.write(constants.INTERACTIVE%[model.__name__ for model in models])
         return s.getvalue()
 
     @property
