Question

我熟悉python装饰器，并想知道一般情况。

让我们说我有一个生成一个人的功能：

def GeneratePerson(name, surname):
    return {"name": name, "surname": surname}

现在，让我们说我想让装饰者添加一个年龄：

def WithAge(fn):
    def AddAge(*args):
        person = fn(args[0], args[1])
        person[age] = args[2]
        return person
    return AddAge

然后我会打电话给

@AddAge
GeneratePerson("John", "Smith", 42)

然而，我发现这有点违反直觉。如果我查看＆＃34; GeneratePerson＆＃34;的签名，我看到它只需要一个名字和一个姓氏参数。年龄（42）是装饰者要求的财产。我觉得语法（Java风格）为：

@AddAge(42)
GeneratePerson("John", "Smith")

可能更可取。

我很欣赏我可以使用kwargs解决这个问题，将堆栈调用为：

@WithAge
@AddGender
GeneratePerson(name="John", surname="Smith", age="42", gender="M")

但有没有办法直接将参数传递给装饰器函数（例如：@Age(42)而不是通过装饰函数间接传递它？）

Answer 1

注意：装饰器在定义时应用，而不是在运行时应用。这意味着你写的时候是：

@AddAge
def GeneratePerson(name, lastname):
    ...

这足以回答你的上一个问题：不可能将参数传递给装饰器而不是装饰函数，除非你希望它对于装饰函数的所有调用都是相同的。

但是在Python 3中，可以在装饰函数中添加一个参数：

def AddAge(param_name='age'):
    # define the new parameter
    param = inspect.Parameter(param_name,
                  inspect.Parameter.POSITIONAL_OR_KEYWORD)
    def wrapper(f):
        sig = inspect.signature(f)
        params = list(sig.parameters.values())
        params.append(param)                     # add the new parameter
        sig2 = sig.replace(parameters = params)  # to a new signature
        def wrapped(*args, **kwargs):
            bound = sig2.bind(*args, **kwargs)   # process the parameters
            args = bound.args                    #  both positional
            kwargs = bound.kwargs                #  and keywords
            if param_name in kwargs:             # extract the age
                age = kwargs[param_name]         #  either keyword
                del kwargs[param_name]
            else:
                args = list(args)                #  or positional
                age = args.pop()
            obj = f(*args, **kwargs)             # call the original function
            obj.age = age                        # add the attribute
            return obj
        wrapped.__signature__ = sig2             # attach the new signature
        return wrapped
    return wrapper

然后你可以这样使用

>>> f = AddAge()(GeneratePerson)
>>> p = f('John', 'Smith', 42)
>>> p.name
'John'
>>> p.lastname
'Smith'
>>> p.age
42

它甚至可以支持函数调用中的关键字：

>>> p = f('John', age=42, lastname="Smith")

会给出预期的结果。唯一的限制是原始函数不应具有名为age的参数。如果有，则必须将另一个名称传递给装饰器。

Python装饰器参数

1 个答案: