Как аргумент который принимает декоратор перекинуть вместо аргумента декорируемой функции?

Question

[Марк]

def dec_arg(arg):

    def decorator(func):
        def wrapper(*args, **kwargs):
            res = func(*args, **kwargs)
            return res
        return wrapper
    return decorator

@dec_arg(20)
def sqr(num, n):

    return (num**2 + n)


print(sqr(1, 45))

Мне интересно узнать, можно ли что, к примеру вместо нашей n, был аргумент который я передал моему декоратору?

Answer 1

[Wispik]

def dec_arg(arg):

    def decorator(func):
        def wrapper(*args, **kwargs):
            res = func(dec_n=arg ,*args, **kwargs)
            return res
        return wrapper
    return decorator

@dec_arg(20)
def sqr(num, n, **kwargs):

    return (num**2 + kwargs['dec_n'])


print(sqr(1, 45))

Answer 2

[Михаил Кростелев]

def dec_arg(arg):
    def decorator(func):
        def wrapper(*args, **kwargs):
            res = func(*args, arg, **kwargs)
            return res
        return wrapper
    return decorator

@dec_arg(20)
def sqr(num, n):

    return (num**2) + n


print(sqr(1))