diff --git a/Week04/decorators_ilyas_bugra_filoglu b/Week04/decorators_ilyas_bugra_filoglu new file mode 100644 index 00000000..0b40daa1 --- /dev/null +++ b/Week04/decorators_ilyas_bugra_filoglu @@ -0,0 +1,31 @@ +import time +import tracemalloc +from functools import wraps + +def performance(func): + # initialize stats on the decorator itself + if not hasattr(performance, "counter"): + performance.counter = 0 + if not hasattr(performance, "total_time"): + performance.total_time = 0.0 + if not hasattr(performance, "total_mem"): + performance.total_mem = 0 + + @wraps(func) + def wrapper(*args, **kwargs): + tracemalloc.start() + tracemalloc.reset_peak() + start = time.perf_counter() + + try: + return func(*args, **kwargs) + finally: + elapsed = time.perf_counter() - start + _, peak = tracemalloc.get_traced_memory() + tracemalloc.stop() + + performance.counter += 1 + performance.total_time += elapsed + performance.total_mem += peak + + return wrapper