diff --git a/utils/timer_utils.py b/utils/timer_utils.py index 8cdb33c..48d67d1 100644 --- a/utils/timer_utils.py +++ b/utils/timer_utils.py @@ -1,19 +1,35 @@ -from time import time +from datetime import datetime -start_time_of_timers = dict() +from rich.console import Console + +console = Console() + +timers_started_ats = dict() -def start_timer(name=1, print_=True): - started_at = time() - start_time_of_timers[name] = started_at - if print_: - print(f"{name} started at: {started_at}") +def start_timer(name=1, do_print=True): + started_at = datetime.now() + timers_started_ats[name] = started_at + if do_print: + console.print(f"{name:<{len(str(name)) + 4}} ", end="") + print(f"started at: {started_at.strftime('%H:%M:%S.%f')}") return name, started_at -def stop_timer(name=1, print_=True): - ended_at = time() - took = round(ended_at - start_time_of_timers[name], 6) - if print_: - print(f"{name} ended at: {ended_at}\ttook: {took} second(s)") - return name, round(ended_at - start_time_of_timers[name], 6) +def stop_timer(name=1, do_print=True): + ended_at = datetime.now() + took = ended_at - timers_started_ats[name] + if do_print: + console.print(f"{name:<{len(str(name)) + 4}} ", end="") + print(f"{'ended at:':>11} {ended_at.strftime('%H:%M:%S.%f')} ", end="") + console.print(f"took: {took}") + return name, ended_at, took + + +if __name__ == "__main__": + from time import sleep + + start_timer() + sleep(1) + stop_timer() +