parouby
/
Pamhyr2
mirror of https://gitlab.com/pamhyr/pamhyr2
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

tools: Update timers display format (add module name).

mesh
Pierre-Antoine Rouby 2023-04-25 11:56:52 +02:00
parent 0b06c9be80
commit 937cf47256
1 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/tools.py
View File

@ -28,7 +28,7 @@ def display_timers():
global _timers
global _calls
print(f" +--{Fore.BLUE}Timers{Style.RESET_ALL}---------------------------------------------------------+")
print(f" +--{Style.BRIGHT}{Fore.BLUE}Timers{Style.RESET_ALL}------------------------------------------------------------------------------------------+")
lst = sorted(
map(
@ -40,9 +40,11 @@ def display_timers():
)
for func, time, calls in lst:
print(f" | {Fore.GREEN}{func:<32}{Style.RESET_ALL} | {time:>10.6f} sec | {calls:>5} calls |")
name = (f"{Fore.BLUE}{func.__module__}{Style.RESET_ALL}" +
f".{Style.BRIGHT}{Fore.GREEN}{func.__qualname__:<{64 - len(func.__module__)}}{Style.RESET_ALL}")
print(f" | {name} | {time:>10.6f} sec | {calls:>5} calls |")
print(" +-----------------------------------------------------------------+")
print(" +--------------------------------------------------------------------------------------------------+")
def timer(func):
"""Function wrapper to register function runtime"""
@ -59,12 +61,12 @@ def timer(func):
end_time = time.perf_counter()
run_time = end_time - start_time
if func.__qualname__ not in _timers:
_timers[func.__qualname__] = 0
_calls[func.__qualname__] = 0
if func not in _timers:
_timers[func] = 0
_calls[func] = 0
_timers[func.__qualname__] += run_time
_calls[func.__qualname__] += 1
_timers[func] += run_time
_calls[func] += 1
return value