diff --git a/Week04/decorators_ege_enc.py b/Week04/decorators_ege_enc.py new file mode 100644 index 00000000..e3999333 --- /dev/null +++ b/Week04/decorators_ege_enc.py @@ -0,0 +1,27 @@ +import tracemalloc +import time +from functools import wraps + +def performance(f): + @wraps(f) + def wrapper(*args, **kwargs): + tracemalloc.start() + start_time = time.time() + + result = f(*args, **kwargs) + + end_time = time.time() + current, peak = tracemalloc.get_traced_memory() + tracemalloc.stop() + + wrapper.counter += 1 + wrapper.total_mem += peak + wrapper.total_time += (end_time - start_time) + + return result + + wrapper.counter = 0 + wrapper.total_mem = 0 + wrapper.total_time = 0 + + return wrapper