summaryrefslogtreecommitdiff
path: root/django/core/db/base.py
blob: 62cca9d0be1b1da426837231f487630d2814e099 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
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)