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() model: AbstractModel = PostgresModel()
if config.log_level == 'DEBUG': if config.log_level == 'DEBUG':
model.get_diff_action_id = utils.measure(model.get_diff_action_id) 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 = utils.measure(model.get_activity_changes, 'model.get_activity_changes')

View File

@ -141,7 +141,7 @@ def _get_bearer() -> str:
return bearer return bearer
def measure(function: callable): def measure(function: callable, name_to_display: str):
""" """
Decorator to measure function (method) execution time Decorator to measure function (method) execution time
Use as easy as Use as easy as
@ -150,6 +150,7 @@ def measure(function: callable):
def im_function_to_measure(): def im_function_to_measure():
.... ....
:param name_to_display:
:param function: :param function:
:return: :return:
""" """
@ -158,7 +159,7 @@ def measure(function: callable):
start = time.time() start = time.time()
result = function(*args, **kwargs) result = function(*args, **kwargs)
end = time.time() 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 result
return decorated return decorated