__init__.py

class DecoratorHelper:
    """This class is intended for decorating called objects.
    Adds attributes that store arguments to an object and returns it."""

    def wraps(self, wrapper, wrapped):

        try:
            wrapper.__doc__ = wrapped.__doc__
            wrapper.__name__ = wrapped.__name__
            wrapper.__module__ = wrapped.__module__
            wrapper.__qualname__ = wrapped.__qualname__
            wrapper.__annotations__ = wrapped.__annotations__
        except Exception:
            pass

    def __init__(self, *args, **kwargs):
        self.function = None
        self.decorator_args = None
        self.function_args = None
        self.pre_function = None
        self.post_function = None
        self.first_call = True

        if hasattr(args[0], '__call__'):
            self.function = args[0]
            self.__decorator_with_args = False
            self.wraps(self, self.function)
        else:
            self.decorator_args = *args, kwargs
            self.__decorator_with_args = True

    def __make_function(self, ):

        if hasattr(self.pre_function, '__call__'):
            self.pre_function()

        if type(self.function_args[0]) != dict:
            self.res = self.function(self.function_args)
        else:
            self.res = self.function()

        if hasattr(self.post_function, '__call__'):
            self.post_function()
        return self.res

    def __call__(self, *args, **kwargs):

        if hasattr(args[0], '__call__'):
            self.function = args[0]
            self.function_args = *args[1:], kwargs
        else:
            self.function_args = *args, kwargs

        self.wraps(self, self.function)

        if self.__decorator_with_args and self.first_call:
            self.first_call = False
            return self
        else:
            return self.__make_function()