clearer error message for missing psycopg

1 files changed, 6 insertions, 2 deletions

diff --git a/python/skytools/psycopgwrapper.py b/python/skytools/psycopgwrapper.py
index c15ee1e8..2127f950 100644
--- a/python/skytools/psycopgwrapper.py
+++ b/python/skytools/psycopgwrapper.py
@@ -76,8 +76,12 @@ try:
 
 except ImportError:
     # use psycopg 1
-    from psycopg import connect as _pgconnect
-    from psycopg import QuotedString
+    try:
+        from psycopg import connect as _pgconnect
+        from psycopg import QuotedString
+    except ImportError:
+        print "Please install psycopg2 module"
+        sys.exit(1)
 
 def connect_database(connstr):
     """Create a db connection with connect_timeout option.