moved psycopg2 reference from scripting module to actual script using it

2 files changed, 6 insertions, 3 deletions

diff --git a/python/skytools/scripting.py b/python/skytools/scripting.py
index ed7dfa03..77c1bec5 100644
--- a/python/skytools/scripting.py
+++ b/python/skytools/scripting.py
@@ -957,8 +957,7 @@ class DBScript(BaseScript):
         sql_retry_formula_a = self.cf.getint("sql_retry_formula_a", 1)
         sql_retry_formula_b = self.cf.getint("sql_retry_formula_b", 5)
         sql_retry_formula_cap = self.cf.getint("sql_retry_formula_cap", 60)
-        import psycopg2
-        elist = exceptions or (psycopg2.OperationalError,)
+        elist = exceptions or tuple([])
         stime = time.time()
         tried = 0
         dbc = None
diff --git a/scripts/simple_local_consumer.py b/scripts/simple_local_consumer.py
index 50177097..e78ba0b4 100755
--- a/scripts/simple_local_consumer.py
+++ b/scripts/simple_local_consumer.py
@@ -22,12 +22,15 @@ Config::
 
 import sys
 
+import psycopg2
+
 import pkgloader
 pkgloader.require('skytools', '3.0')
 
 import pgq
 import skytools
 
+
 class SimpleLocalConsumer(pgq.LocalConsumer):
     __doc__ = __doc__
 
@@ -57,7 +60,8 @@ class SimpleLocalConsumer(pgq.LocalConsumer):
         payload['pgq.ev_extra4'] = ev.ev_extra4
 
         self.log.debug(self.dst_query, payload)
-        retries, curs = self.execute_with_retry('dst_db', self.dst_query, payload)
+        retries, curs = self.execute_with_retry('dst_db', self.dst_query, payload,
+                                                exceptions = (psycopg2.OperationalError,))
         if curs.statusmessage[:6] == 'SELECT':
             res = curs.fetchall()
             self.log.debug(res)