diff -r 203be9022b46 -r 16dc5dec9c36 pgtoolkit/toolbase.py --- 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: