diff options
Diffstat (limited to 'bin/sql_db.py')
| -rw-r--r-- | bin/sql_db.py | 133 |
1 files changed, 133 insertions, 0 deletions
diff --git a/bin/sql_db.py b/bin/sql_db.py new file mode 100644 index 00000000000..005d66fec53 --- /dev/null +++ b/bin/sql_db.py @@ -0,0 +1,133 @@ +############################################################################## +# +# Copyright (c) 2004 TINY SPRL. (http://tiny.be) All Rights Reserved. +# +# $Id: pooler.py 1310 2005-09-08 20:40:15Z pinky $ +# +# WARNING: This program as such is intended to be used by professional +# programmers who take the whole responsability of assessing all potential +# consequences resulting from its eventual inadequacies and bugs +# End users who are looking for a ready-to-use solution with commercial +# garantees and support are strongly adviced to contract a Free Software +# Service Company +# +# This program is Free Software; you can redistribute it and/or +# modify it under the terms of the GNU General Public License +# as published by the Free Software Foundation; either version 2 +# of the License, or (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program; if not, write to the Free Software +# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. +# +############################################################################## + +import psycopg +import tools +import sys,os + +#try: +# import decimal +#except ImportError: +# from tools import decimal + + +import re + +class fake_cursor: + nbr = 0 + _tables = {} + def __init__(self, db, con, dbname): + self.db = db + self.obj = db.cursor() + self.con = con + self.dbname = dbname + + def execute_dont_run(self,*args): + if not fake_cursor.nbr % 100: + print 'sql: ',fake_cursor.nbr + res = re.match('^select.* from ([a-zA-Z_]+) .*$', args[0], re.I) + if res: + fake_cursor._tables.setdefault(res.group(1), 0) + fake_cursor._tables[res.group(1)] += 1 + + #else: + # if len(args)>1: + # print 'sql: ',fake_cursor.nbr, args[0], args[1] + # else: + # print 'sql: ',fake_cursor.nbr, args[0] + + if not fake_cursor.nbr % 5000: + ct = [] + for t,c in fake_cursor._tables.items(): + ct.append([c,t]) + ct.sort() + ct.reverse() + print 'After %d queries' % (fake_cursor.nbr,) + for line in ct[:50]: + print ' %s: %d' % (line[1], line[0]) + + #if len(args)>1: + # print 'sql: ',fake_cursor.nbr, args[0], args[1] + #else: + # print 'sql: ',fake_cursor.nbr, args[0] + + fake_cursor.nbr += 1 + return self.obj.execute(*args) + + def close(self): +# print "close cursors fno:", [i.fileno() for i in self.db.cursors] + self.obj.close() + + # This force the cursor to be freed, and thus, available again. It is + # important because otherwise we can overload the server very easily + # because of a cursor shortage (because cursors are not garbage + # collected as fast as they should). The problem is probably due in + # part because browse records keep a reference to the cursor. + del self.obj +# print "after close cursors fno:", [i.fileno() for i in self.db.cursors] + + def __getattr__(self, name): +# print 'LOOK',name + return getattr(self.obj, name) + +class fakedb: + def __init__(self, truedb, dbname): + self.truedb = truedb + self.dbname = dbname + + def cursor(self): + return fake_cursor(self.truedb, {}, self.dbname) + +def decimalize(symb): + if symb is None: return None + if isinstance(symb, float): + return decimal.Decimal('%f' % symb) + return decimal.Decimal(symb) + +def db_connect(db_name): + host = tools.config['db_host'] and "host=%s" % tools.config['db_host'] or '' + port = tools.config['db_port'] and "port=%s" % tools.config['db_port'] or '' + name = "dbname=%s" % db_name + user = tools.config['db_user'] and "user=%s" % tools.config['db_user'] or '' + password = tools.config['db_password'] and "password=%s" % tools.config['db_password'] or '' + tdb = psycopg.connect('%s %s %s %s %s' % (host, port, name, user, password), serialize=0) + fdb = fakedb(tdb, db_name) + return fdb + +def init(): + #define DATEOID 1082, define TIMESTAMPOID 1114 see pgtypes.h + psycopg.register_type(psycopg.new_type((1082,), "date", lambda x:x)) + psycopg.register_type(psycopg.new_type((1083,), "time", lambda x:x)) + psycopg.register_type(psycopg.new_type((1114,), "datetime", lambda x:x)) + #psycopg.register_type(psycopg.new_type((700, 701, 1700), 'decimal', decimalize)) + +psycopg.register_type(psycopg.new_type((1082,), "date", lambda x:x)) +psycopg.register_type(psycopg.new_type((1083,), "time", lambda x:x)) +psycopg.register_type(psycopg.new_type((1114,), "datetime", lambda x:x)) + |