mirror of
https://github.com/norohind/SquadsActivityMonitor.git
synced 2025-04-12 21:07:12 +03:00
utils.measure: add name to display
This commit is contained in:
parent
54ce44de15
commit
f58df5e45f
@ -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')
|
||||
|
5
utils.py
5
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
|
||||
|
Loading…
x
Reference in New Issue
Block a user