Source code for luna_tasks.crutches_on_wheels.utils.timer

"""Timers

Timer decorators

"""
import time
from functools import wraps
from typing import Callable

from vlutils.helpers import isCoroutineFunction

from ..utils.log import Logger


def printLogger(self: object, func: callable, duration: float) -> None:
    """
    Print timing to log

    Args:
        self: class instance with logger
        func: measured function
        duration: calling duration of function in secs
    """
    if hasattr(self, "logger"):
        self.logger.debug("{} time: {}".format(func.__qualname__, round(duration, 5)))
    else:
        Logger().debug("{} time: {}".format(func.__qualname__, round(duration, 5)))


def timer(func: callable) -> Callable:
    """
    Decorator for function work time estimation.

    Args:
        func: decorating function
    Returns:
        decorated function.
    """
    if isCoroutineFunction(func):

        @wraps(func)
        async def wrap(*func_args, **func_kwargs):
            start = time.perf_counter()
            res = await func(*func_args, **func_kwargs)
            end = time.perf_counter()
            printLogger(func_args and func_args[0], func, end - start)
            return res

    else:

        @wraps(func)
        def wrap(*func_args, **func_kwargs):
            start = time.perf_counter()
            res = func(*func_args, **func_kwargs)
            end = time.perf_counter()
            printLogger(func_args and func_args[0], func, end - start)
            return res

    return wrap