diff options
Diffstat (limited to 'python/skytools/psycopgwrapper.py')
| -rw-r--r-- | python/skytools/psycopgwrapper.py | 8 |
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. |