timerproxy.py
45 lines
| 1.5 KiB
| text/x-python
|
PythonLexer
|
r4116 | # -*- coding: utf-8 -*- | ||
# 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 3 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, see <http://www.gnu.org/licenses/>. | ||||
r547 | import time | |||
r1191 | import logging | |||
|
r4116 | from sqlalchemy.interfaces import ConnectionProxy | ||
r1191 | log = logging.getLogger('timerproxy') | |||
r547 | BLACK, RED, GREEN, YELLOW, BLUE, MAGENTA, CYAN, WHITE = xrange(30, 38) | |||
r1307 | ||||
r547 | def color_sql(sql): | |||
COLOR_SEQ = "\033[1;%dm" | ||||
COLOR_SQL = YELLOW | ||||
normal = '\x1b[0m' | ||||
r1360 | return ''.join([COLOR_SEQ % COLOR_SQL, sql, normal]) | |||
r547 | ||||
r1307 | ||||
r547 | class TimerProxy(ConnectionProxy): | |||
r1186 | ||||
r547 | def __init__(self): | |||
super(TimerProxy, self).__init__() | ||||
r1186 | ||||
def cursor_execute(self, execute, cursor, statement, parameters, | ||||
context, executemany): | ||||
r547 | now = time.time() | |||
try: | ||||
r1191 | log.info(color_sql(">>>>> STARTING QUERY >>>>>")) | |||
r547 | return execute(cursor, statement, parameters, context) | |||
finally: | ||||
total = time.time() - now | ||||
r1191 | log.info(color_sql("<<<<< TOTAL TIME: %f <<<<<" % total)) | |||