--- a/pgtoolkit/toolbase.py Tue Sep 06 17:55:15 2011 +0200
+++ b/pgtoolkit/toolbase.py Fri Sep 09 11:56:37 2011 +0200
@@ -1,11 +1,16 @@
import argparse
+import logging
from pgtoolkit import pgmanager, config
+from pgtoolkit.coloredformatter import ColoredFormatter
+from pgtoolkit.highlight import highlight
class ToolBase:
def __init__(self, name, desc):
self.parser = argparse.ArgumentParser(description=desc)
+ self.parser.add_argument('-d', dest='debug', action='store_true',
+ help='Debug mode - print database queries.')
self.config = config.ConfigParser()
self.config.add_argument('databases', type=dict)
@@ -17,6 +22,14 @@
def init(self):
self.config.load('pgtoolkit.conf')
self.args = self.parser.parse_args()
+ if self.args.debug:
+ handler = logging.StreamHandler()
+ format = ColoredFormatter(
+ highlight(1,7,0)+'%(asctime)s %(levelname)-5s'+highlight(0)+' %(message)s', '%H:%M:%S')
+ handler.setFormatter(format)
+ handler.setLevel(logging.DEBUG)
+ logger = logging.getLogger('pgmanager')
+ logger.addHandler(handler)
def buildconn(self, name, targetname):
with self.pgm.cursor('meta') as curs: