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