diff --git a/Week04/decorators_alisher_shalenov.py b/Week04/decorators_alisher_shalenov.py new file mode 100644 index 00000000..af105b1f --- /dev/null +++ b/Week04/decorators_alisher_shalenov.py @@ -0,0 +1,27 @@ +import time +import tracemalloc + +def performance(fn): + if not hasattr(performance, "counter"): + performance.counter = 0 + performance.total_time = 0 + performance.total_mem = 0 + + def wrapper(*args, **kwargs): + performance.counter += 1 + + tracemalloc.start() + start_time = time.time() + + result = fn(*args, **kwargs) + + end_time = time.time() + current, peak = tracemalloc.get_traced_memory() + tracemalloc.stop() + + performance.total_time += (end_time - start_time) + performance.total_mem += peak + + return result + + return wrapper