Mirrors/PySnooper
1
0
Fork 0
mirror of https://github.com/cool-RR/PySnooper.git synced 2026-01-23 02:14:04 +00:00
Code Issues Projects Releases Wiki Activity

Enable multi call

Browse source
This commit is contained in:
iory 2020-04-18 09:12:44 +09:00 • committed by Ram Rachum
parent d94b0214f9
commit b886f2b504
1 changed files with 9 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

15
pysnooper/tracer.py
View file

@ -217,6 +217,7 @@ class Tracer:
for v in utils.ensure_tuple(watch_explode)
]
self.frame_to_local_reprs = {}
self.start_times = {}
self.depth = depth
self.prefix = prefix
self.thread_info = thread_info
@ -302,7 +303,7 @@ class Tracer:
'original_trace_functions', []
)
stack.append(sys.gettrace())
self.start_time = datetime_module.datetime.now()
self.start_times[id(calling_frame)] = datetime_module.datetime.now()
sys.settrace(self.trace)
def __exit__(self, exc_type, exc_value, exc_traceback):
@ -314,10 +315,10 @@ class Tracer:
self.target_frames.discard(calling_frame)
self.frame_to_local_reprs.pop(calling_frame, None)
if thread_global.depth == -1:
duration = datetime_module.datetime.now() - self.start_time
now_string = pycompat.timedelta_isoformat(duration, timespec='microseconds')
self.write('Total elapsed time: {now_string}'.format(**locals()))
start_time = self.start_times.pop(id(calling_frame))
duration = datetime_module.datetime.now() - start_time
now_string = pycompat.timedelta_isoformat(duration, timespec='microseconds')
self.write('Total elapsed time: {now_string}'.format(**locals()))
def _is_internal_frame(self, frame):
return frame.f_code.co_filename == Tracer.__enter__.__code__.co_filename
@ -364,7 +365,9 @@ class Tracer:
### Finished checking whether we should trace this line. ##############
if self.elapsed_time:
duration = datetime_module.datetime.now() - self.start_time
calling_frame = frame.f_back
duration = datetime_module.datetime.now() - self.start_times[
id(calling_frame)]
now_string = pycompat.timedelta_isoformat(
duration, timespec='microseconds') if not self.normalize else ' ' * 15
else: