From 9d0ef2d5035ef98624f7c605826ff41dd0dd8f76 Mon Sep 17 00:00:00 2001
From: Marko Kreen <markokr@gmail.com>
Date: Mon, 17 Sep 2007 09:04:21 +0000
Subject: [PATCH] clearer error message for missing psycopg

---
 python/skytools/psycopgwrapper.py | 8 ++++++--
 1 file 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.
-- 
2.39.5