diff --git a/ingestion/utils.py b/ingestion/utils.py index 7a4fb4e..b837d42 100644 --- a/ingestion/utils.py +++ b/ingestion/utils.py @@ -1,5 +1,6 @@ import logging import time +from datetime import timedelta def st_time(func): @@ -7,16 +8,19 @@ def st_time(func): st decorator to calculate the total time of a func """ - def st_func(*args, **keyArgs): + def st_func(*args, **kwargs): t1 = time.time() - r = func(*args, **keyArgs) + r = func(*args, **kwargs) t2 = time.time() + elapsed = t2 - t1 logging.info( - f"Function={func.__name__}, args={args}, " + f"Function={func.__name__}, args={args}, kwargs={kwargs}, " f"start_time={time.strftime('%Y-%m-%d %H:%M:%S', time.gmtime(t1))}, " f"end_time={time.strftime('%Y-%m-%d %H:%M:%S', time.gmtime(t2))}" ) - logging.info(f"Function={func.__name__}, args={args}, total_time={t2 - t1}") + logging.info( + f"Function={func.__name__}, args={args}, kwargs={kwargs}, elapsed_time={str(timedelta(seconds=elapsed))}" + ) return r return st_func