Python deprecation decorator

python_deprecation_helper.py

def deprecated(arg):
    if callable(arg):
        return deprecated_auto_message(arg)
    else:
        return deprecated_user_message(arg)

def deprecated_auto_message(func):
    def wrapper(*args, **kwargs):
        message = "Function {} is deprecated.".format(func.__name__)
        warn_deprecation(message, 3)
        return func(*args, **kwargs)

    return wrapper

def deprecated_user_message(message):
    def decorator(func):
        def wrapper(*args, **kwargs):
            warn_deprecation(message, 3)
            return func(*args, **kwargs)

        return wrapper

    return decorator

def warn_deprecation(message, stacklevel):
    from warnings import warn
    warn(message, DeprecationWarning, stacklevel=stacklevel)