Python timed logger decorator

decorators.py

import logging
import time

from functools import wraps
from typing import (
    Callable,
    Optional,
)


def timed_logger(logger: Optional[Callable] = None):
    """
    Decorator for logging the time taken to run a function, if no logger
    is set then default to logger.info.
    """

    # logger is optional. If it is just use a default (logger.info)
    if not callable(logger):
        logger = logging.info

    def decorator(func: Callable):
        @wraps(func)
        def wrapper(*args, **kwargs):
            start = time.time()
            result = func(*args, **kwargs)
            taken = round(time.time() - start, 2)
            logger(f'{func.__name__} ran in {taken}(seconds)')
            return result

        return wrapper
    return decorator