From f58df5e45f95e8886dd37844dbb26508c63ea902 Mon Sep 17 00:00:00 2001
From: norohind <60548839+norohind@users.noreply.github.com>
Date: Wed, 8 Dec 2021 21:05:27 +0300
Subject: [PATCH] utils.measure: add name to display

---
 model/__init__.py | 4 ++--
 utils.py          | 5 +++--
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/model/__init__.py b/model/__init__.py
index 6e10f20..e3aa162 100644
--- a/model/__init__.py
+++ b/model/__init__.py
@@ -9,5 +9,5 @@ logger = get_main_logger()
 model: AbstractModel = PostgresModel()
 
 if config.log_level == 'DEBUG':
-    model.get_diff_action_id = utils.measure(model.get_diff_action_id)
-    model.get_activity_changes = utils.measure(model.get_activity_changes)
+    model.get_diff_action_id = utils.measure(model.get_diff_action_id, 'model.get_diff_action_id')
+    model.get_activity_changes = utils.measure(model.get_activity_changes, 'model.get_activity_changes')
diff --git a/utils.py b/utils.py
index 628c97e..b1fdce9 100644
--- a/utils.py
+++ b/utils.py
@@ -141,7 +141,7 @@ def _get_bearer() -> str:
     return bearer
 
 
-def measure(function: callable):
+def measure(function: callable, name_to_display: str):
     """
     Decorator to measure function (method) execution time
     Use as easy as
@@ -150,6 +150,7 @@ def measure(function: callable):
     def im_function_to_measure():
         ....
 
+    :param name_to_display:
     :param function:
     :return:
     """
@@ -158,7 +159,7 @@ def measure(function: callable):
         start = time.time()
         result = function(*args, **kwargs)
         end = time.time()
-        print(f'{function.__name__}: {(end - start) * 100} ms')
+        print(f'{name_to_display}:{function.__name__}: {(end - start) * 100} ms')
         return result
 
     return decorated