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

Add total elapsed time

Browse source
This commit is contained in:
iory 2020-04-16 04:27:36 +09:00 • committed by Ram Rachum
parent 487fa5317e
commit 6416a11d39
1 changed files with 6 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
pysnooper/tracer.py
View file

@ -296,6 +296,7 @@ class Tracer:
'original_trace_functions', []
)
stack.append(sys.gettrace())
self.start_time = datetime_module.datetime.now()
sys.settrace(self.trace)
def __exit__(self, exc_type, exc_value, exc_traceback):
@ -307,6 +308,11 @@ class Tracer:
self.target_frames.discard(calling_frame)
self.frame_to_local_reprs.pop(calling_frame, None)
now = (datetime_module.datetime.min + (
datetime_module.datetime.now() - self.start_time)).time()
now_string = pycompat.time_isoformat(now, 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