33 lines
919 B
Python
33 lines
919 B
Python
|
from time import time
|
||
|
|
|
||
|
|
class CursorDebugWrapper:
|
||
|
|
def __init__(self, cursor, db):
|
||
|
|
self.cursor = cursor
|
||
|
|
self.db = db
|
||
|
|
|
||
|
|
def execute(self, sql, params=[]):
|
||
|
|
start = time()
|
||
|
|
result = self.cursor.execute(sql, params)
|
||
|
|
stop = time()
|
||
|
|
self.db.queries.append({
|
||
|
|
'sql': sql % tuple(params),
|
||
|
|
'time': "%.3f" % (stop - start),
|
||
|
|
})
|
||
|
|
return result
|
||
|
|
|
||
|
|
def executemany(self, sql, param_list):
|
||
|
|
start = time()
|
||
|
|
result = self.cursor.executemany(sql, param_list)
|
||
|
|
stop = time()
|
||
|
|
self.db.queries.append({
|
||
|
|
'sql': 'MANY: ' + sql + ' ' + str(tuple(param_list)),
|
||
|
|
'time': "%.3f" % (stop - start),
|
||
|
|
})
|
||
|
|
return result
|
||
|
|
|
||
|
|
def __getattr__(self, attr):
|
||
|
|
if self.__dict__.has_key(attr):
|
||
|
|
return self.__dict__[attr]
|
||
|
|
else:
|
||
|
|
return getattr(self.cursor, attr)
|