Update decorators_fevzi_bagriacik.py

Author: fevzibagriacik

Week04/decorators_fevzi_bagriacik.py

@@ -1,20 +1,16 @@
 import time
 import tracemalloc
-from functools import update_wrapper
+from functools import wraps
 
 
 class performance:
-    """
-    A decorator which measures the performance of functions and
-    also saves some statistics.
-    """
+    counter = 0
+    total_time = 0.0
+    total_mem = 0
 
     def __init__(self, func):
         self.func = func
-        self.counter = 0
-        self.total_time = 0.0
-        self.total_mem = 0
-        update_wrapper(self, func)
+        wraps(func)(self)
 
     def __call__(self, *args, **kwargs):
         tracemalloc.start()
@@ -26,8 +22,8 @@ def __call__(self, *args, **kwargs):
         current, peak = tracemalloc.get_traced_memory()
         tracemalloc.stop()
 
-        self.counter += 1
-        self.total_time += (end_time - start_time)
-        self.total_mem += peak
+        performance.counter += 1
+        performance.total_time += (end_time - start_time)
+        performance.total_mem += peak
 
         return result