0

我有一些函数，在正常情况下，这些函数是使用用户输入提供的参数调用的。但是，使用某些系列参数调用其中一些函数是有效的，这些参数是在运行时根据某些系统状态确定的。

我希望用户能够有选择地指示我的程序使用所有有效输入调用这些函数并返回每次调用的结果。我认为一个装饰器可以像激活开关一样工作，用于具有另一个装饰器的函数，该装饰器指示要使用的参数系列会很好地工作。

此外，我需要保留函数签名和元数据。这对我的程序的运行至关重要。

这是我尝试过的，但它不起作用。它基于这个例子。

>>> from decorator import decorator
>>> def the_list():
...     return ["foo", "bar", "baz"]
... 
>>> import itertools
>>> @decorator
... def do_all(func):
...     # this will do nothing (besides wrap in a tuple) unless func is decorated with @gets_arg_from
...     if hasattr(func, 'get_from'):
...         return tuple(func(*args) for args in itertools.product(*(list_fun() for list_fun in func.get_from)))
...     else:
...         return (func(),)
... 
>>> def gets_arg_from(*list_funcs):
...     # this will do nothing to func unless decorated with @do_all
...     def gets_arg_from(func, *args, **kwargs):
...         func.get_from = list_funcs
...         return func(*args, **kwargs)
...     return decorator(gets_arg_from)
... 
>>> @gets_arg_from(the_list)
... def print_func(word):
...     # this will print its argument unless decorated with @do_all
...     # at that point it will print every element returned by the_list()
...     print word
... 
>>> print_func("foo")
foo
>>> all = decorator(do_all, print_func)
>>> all()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: print_func() takes exactly 1 argument (0 given)
>>> print_func.get_from
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
AttributeError: 'function' object has no attribute 'get_from'

我的预期是：

>>> all()
("foo", "bar", "baz")

我注意到的是错误的：

  1. gets_arg_from不将get_from属性添加到func.
  2. 关于我使用该符号的某些事情@gets_arg_from(the_list)是错误的。它认为我正在尝试传递两个参数（但为什么这会成为问题呢？）

至于我的动机，我想到了装饰器，因为实际上有数百个这样的例程，它们的实现细节（以及它们的功能正确性）经常变化，我不想用它inspect来推理做什么基于它们的参数名称，我也不想do_all对每个有意义的函数的功能进行硬编码。类方法可能有效，但就我的目的而言，它们是语义设计的。此外，为了其他可能需要维护我的代码的人，我认为让他们应用装饰器而不用担心其余部分比使用某些参数名称或将函数放在某个类中或任何。我意识到这个问题可能听起来很奇怪，所以我认为这个脚注可能会让我看起来不那么像个疯子。

4

1 回答 1

1

这不是在做你想做的事吗？

import functools
from itertools import product

def the_list():
    return ["foo", "bar", "baz"]


def do_all(func):
    if hasattr(func, 'get_from'):
        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            return tuple(func(*args) for args in
                         product(*(lf() for lf in func.get_from)))
        return wrapper
    return func

def gets_arg_from(*list_funcs):
    def decorator(func):
        func.get_from = list_funcs
        return func
    return decorator

@gets_arg_from(the_list)
def print_func(word):
    return word

print print_func('foo')

all = do_all(print_func)
print all()

编辑：解释

这两个代码段是相同的：

@deco
def func(...):
    some code

是相同的

func = deco(lambda ...: some code)

@something 只是函数调用和匿名函数创建的语法糖......

我将逐步解释下一个代码和平中发生的事情：

@gets_arg_from(the_list)
def print_func(word):
     return word

  1. 首先 Python 创建一个匿名函数，它接收一个参数word并有一个只返回 thisword的函数体（或执行函数体所做的任何事情）

  2. 然后函数get_arg_from被调用并the_list作为参数传递给它

  3. get_arg_from创建一个decorator函数并返回它

  4. decorator调用从返回的函数get_arg_from（这是语法糖）作为参数传递func在步骤 1 中创建的匿名函数。

  5. decorator只是将list_funcs元组分配给get_from匿名函数的属性并返回匿名函数

  6. 函数的返回值decorator被赋值给一个变量print_func

类似的效果可以通过以下方式实现：

def __anonimous(word):
    return word
__decorator = gets_arg_from(the_list)
print_func = __decorator(__anonimous)

所以基本上gets_arg_from不是装饰器，它是一个返回装饰器的函数。

do_all另一方面是一个装饰器，它接收一个函数作为参数，并返回原始函数（如果函数没有属性get_from）或wrapper替换原始函数的函数（如果它有get_from属性）。

您可以在此处找到更多示例。

于 2013-08-02T19:09:12.993 回答