Python logging decorator

log_io.py

from functools import wraps
from logging import getLogger, DEBUG

logger = getLogger(__name__)

def log_io(level: int=DEBUG, enter: bool=False, exit: bool=False):
    """
    Decorator factory that logs function input arguments and return values
    at the specified logging level.

    Usage:
        debug = @log_io(logging.DEBUG, enter=True, exit=True)
        
        @debug
        def my_func(...):
            ...
    """
    def decorator(func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            if enter:
                logger.log(level, "Calling %s with args=%s, kwargs=%s", func.__name__, args, kwargs, stacklevel=2)
            result = func(*args, **kwargs)
            if exit:
                logger.log(level, "%s returned %r", func.__name__, result, stacklevel=2)
            return result
        return wrapper
    return decorator

debug = log_io(DEBUG, enter=True, exit=True)
debug_in = log_io(DEBUG, enter=True)
debug_out = log_io(DEBUG, exit=True)