utils.measure: add name to display

This commit is contained in:
norohind 2021-12-08 21:05:27 +03:00
parent 54ce44de15
commit f58df5e45f
Signed by: norohind
GPG Key ID: 01C3BECC26FB59E1
2 changed files with 5 additions and 4 deletions

View File

@ -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')

View File

@ -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