timer

timer.py

import time
import logging


def timer(func):
    def inner(*args, **kwargs):
        start = time.time()
        rv = func(*args, **kwargs)
        logger.info(f"Time taken to execute {func}: {time.time() - start}")
        return rv
    return inner