summaryrefslogtreecommitdiff
path: root/python/skytools/scripting.py
diff options
context:
space:
mode:
authorMarko Kreen2011-04-06 17:42:31 +0000
committerMarko Kreen2011-04-06 17:44:52 +0000
commitf6a8d2ceec20e2e97ba05094827d4bdbda4ef0f6 (patch)
tree7a245ebdad567611855762a91b9bb07e23d3a148 /python/skytools/scripting.py
parent72d2fd72df5837c33e69b509daaad17b0413f1d7 (diff)
Avoid direct symbol imports
Always use import skytools def xx(): skytools.foo this allows maximum laziness when importing.
Diffstat (limited to 'python/skytools/scripting.py')
-rw-r--r--python/skytools/scripting.py18
1 files changed, 8 insertions, 10 deletions
diff --git a/python/skytools/scripting.py b/python/skytools/scripting.py
index 90125f58..298c1efa 100644
--- a/python/skytools/scripting.py
+++ b/python/skytools/scripting.py
@@ -6,10 +6,8 @@
import sys, os, signal, optparse, time, errno, select
import logging, logging.handlers, logging.config
-from skytools.config import *
-from skytools.psycopgwrapper import connect_database
-from skytools.quoting import quote_statement
-import skytools.skylog, psycopg2
+import skytools
+import skytools.skylog
try:
import skytools.installer_config
@@ -232,7 +230,7 @@ class DBCachedConn(object):
# new conn?
if not self.conn:
self.isolation_level = isolation_level
- self.conn = connect_database(self.loc)
+ self.conn = skytools.connect_database(self.loc)
self.conn.my_name = self.name
self.conn.set_isolation_level(isolation_level)
@@ -458,7 +456,7 @@ class DBScript(object):
print("need config file, use --help for help.")
sys.exit(1)
conf_file = self.args[0]
- return Config(self.service_name, conf_file, override = self.cf_operride)
+ return skytools.Config(self.service_name, conf_file, override = self.cf_operride)
def init_optparse(self, parser = None):
"""Initialize a OptionParser() instance that will be used to
@@ -704,7 +702,7 @@ class DBScript(object):
self.log.info("got KeyboardInterrupt, exiting")
self.reset()
sys.exit(1)
- except psycopg2.Error, d:
+ except skytools.DBError, d:
self.send_stats()
if d.cursor and d.cursor.connection:
cname = d.cursor.connection.my_name
@@ -791,7 +789,7 @@ class DBScript(object):
def _exec_cmd(self, curs, sql, args, quiet = False):
"""Internal tool: Run SQL on cursor."""
- self.log.debug("exec_cmd: %s" % quote_statement(sql, args))
+ self.log.debug("exec_cmd: %s" % skytools.quote_statement(sql, args))
curs.execute(sql, args)
ok = True
rows = curs.fetchall()
@@ -801,7 +799,7 @@ class DBScript(object):
msg = row['ret_note']
except KeyError:
self.log.error("Query does not conform to exec_cmd API:")
- self.log.error("SQL: %s" % quote_statement(sql, args))
+ self.log.error("SQL: %s" % skytools.quote_statement(sql, args))
self.log.error("Row: %s" % repr(row.copy()))
sys.exit(1)
level = code / 100
@@ -816,7 +814,7 @@ class DBScript(object):
self.log.warning("%s" % (msg,))
else:
self.log.error("%s" % (msg,))
- self.log.debug("Query was: %s" % quote_statement(sql, args))
+ self.log.debug("Query was: %s" % skytools.quote_statement(sql, args))
ok = False
return (ok, rows)