diff --git a/Week04/decorators_beyza_saridas.py b/Week04/decorators_beyza_saridas.py new file mode 100644 index 00000000..6a3b10f9 --- /dev/null +++ b/Week04/decorators_beyza_saridas.py @@ -0,0 +1,28 @@ +import time +import tracemalloc + +def performance(fn): + def _performance(*args, **kwargs): + # If the 'performance' function has not been called before, initialize static variables + if not hasattr(performance, "counter"): + setattr(performance, "counter", 0) + setattr(performance, "total_time", 0) + setattr(performance, "total_mem", 0) + + 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_mem += peak + performance.total_time += (end_time - start_time) + + return result + + return _performance