Create decorators_fevzi_bagriacik.py

Author: fevzibagriacik

Week04/decorators_fevzi_bagriacik.py

@@ -0,0 +1,33 @@
+import time
+import tracemalloc
+from functools import wraps
+
+
+def performance(func):
+    """
+    A decorator which measures the performance of functions and
+    also saves some statistics.
+    """
+
+    @wraps(func)
+    def wrapper(*args, **kwargs):
+        tracemalloc.start()
+        start_time = time.perf_counter()
+
+        result = func(*args, **kwargs)
+
+        end_time = time.perf_counter()
+        current, peak = tracemalloc.get_traced_memory()
+        tracemalloc.stop()
+
+        wrapper.counter += 1
+        wrapper.total_time += (end_time - start_time)
+        wrapper.total_mem += peak
+
+        return result
+
+    wrapper.counter = 0
+    wrapper.total_time = 0.0
+    wrapper.total_mem = 0
+
+    return wrapper