@returns Python

from __future__ import annotations


def return_func(
    all_value: Optional = None,
    reminder: bool = True
) -> Callable:
    """
    Initialize:
    returns = return_func()
    Optional keyword adrguments:
        all_value: value to return on all instances of decorator
        reminder: print reminder to remove decorator

    Usage:
    @returns("Value to return")
    def some_function(...):
        ...
    @returns()  # if all_value specified in initilization
    def some_function(...):
        ...
    """
    def returns(value: Optional = None) -> Callable:
        def function(fn: Callable) -> Callable:
            def wrapper(*args, **kwargs) -> Any:
                if reminder:
                    function_name = fn.__name__
                    decorator_line = fn.__code__.co_firstlineno
                    print(
                        "Don't forget to REMOVE decorator on function",
                        fn.__name__,
                        "on line",
                        decorator_line
                    )
                if all_value:
                    return all_value
                return value
            return wrapper
        return function
    return returns