diff --git a/Week04/decorators_ayseselin_kargi.py b/Week04/decorators_ayseselin_kargi.py new file mode 100644 index 00000000..2f5bdd2c --- /dev/null +++ b/Week04/decorators_ayseselin_kargi.py @@ -0,0 +1,30 @@ +import time +import tracemalloc + +def performance(fn): + counter = 0 + total_time = 0.0 + total_mem = 0 + + def wrapper(*args, **kwargs): + nonlocal counter, total_time, total_mem + + counter += 1 + + tracemalloc.start() + start = time.perf_counter() + + result = fn(*args, **kwargs) + + end = time.perf_counter() + current, peak = tracemalloc.get_traced_memory() + tracemalloc.stop() + + total_time += (end - start) + total_mem += peak + + return result + + wrapper.get_stats = lambda: (counter, total_time, total_mem) + + return wrapper