skorotkiewicz/rDrama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Make proper use of defaultdict.

Browse source
This commit is contained in:
Ben Rog-Wilhelm 2022-11-10 21:40:16 -06:00 • committed by Ben Rog-Wilhelm
parent 1437bc3092
commit 965d9b30b3
1 changed files with 3 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

7
thirdparty/sqlalchemy-easy-profile/easy_profile/profiler.py vendored
View file

@ -1,4 +1,4 @@
from collections import Counter, namedtuple, OrderedDict from collections import Counter, namedtuple, OrderedDict, defaultdict
import functools import functools
import inspect import inspect
from queue import Queue from queue import Queue
@ -164,8 +164,6 @@ class SessionProfiler:
self._stats["duration"] += query.duration self._stats["duration"] += query.duration
duplicates = self._stats["duplicates"].get(query.statement, -1) duplicates = self._stats["duplicates"].get(query.statement, -1)
self._stats["duplicates"][query.statement] = duplicates + 1 self._stats["duplicates"][query.statement] = duplicates + 1
if query.statement not in self._stats["callstacks"]:
self._stats["callstacks"][query.statement] = Counter()
self._stats["callstacks"][query.statement].update({query.callstack: 1}) self._stats["callstacks"][query.statement].update({query.callstack: 1})
return self._stats return self._stats
@ -181,7 +179,7 @@ class SessionProfiler:
self._stats["duration"] = 0 self._stats["duration"] = 0
self._stats["call_stack"] = [] self._stats["call_stack"] = []
self._stats["duplicates"] = Counter() self._stats["duplicates"] = Counter()
self._stats["callstacks"] = {} self._stats["callstacks"] = defaultdict(Counter)
def _before_cursor_execute(self, conn, cursor, statement, parameters, def _before_cursor_execute(self, conn, cursor, statement, parameters,
context, executemany): context, executemany):
@ -194,4 +192,5 @@ class SessionProfiler:
)) ))
def _stack_callback_default(self): def _stack_callback_default(self):
pprint.pprint(traceback.format_stack())
return ''.join(traceback.format_stack()) return ''.join(traceback.format_stack())