mirror of
https://github.com/norohind/SquadsActivityMonitor.git
synced 2026-10-22 17:12:08 +03:00
utils.measure: add name to display
This commit is contained in:
2 files changed
+5
-4
No files matched your search
+2
-2
@@ -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')
|
||||
@@ -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
|
||||
|
||||
Reference in new issue
Block a user