python - 使用 Python 装饰器为方法添加方法

Question

我希望能够根据某种标准格式调用方法:

outputs = obj.meth(in_0, in_1, ...)

，其中输出是一个数组元组，每个输入是一个数组。

然而，在大多数情况下，我只返回一个数组，不想为了标准格式而被迫返回长度为 1 的元组。 (我的实际格式问题更复杂，但现在我们坚持这个解释。)

我希望能够定义一个类:

class _SomeClass(object):

    def __init__(self):
        self._amount_to_add = 1

    @single_return_format
    def add_one(self, x):
        return x+self._amount_to_add

然后可以这样调用它:

obj = _SomeClass()
assert obj.add_one(3) == 4
assert obj.add_one.standard_format(3)==(4, )

问题是:如何定义装饰器以允许这种行为？

我试过:

def single_return_format(fcn):
    fcn.standard_format = lambda *args: (fcn(*args), )
    return fcn

，但它在第二个断言行上失败了:

TypeError: add_one() takes exactly 2 arguments (1 given)

因为 add_one 需要“self”作为参数，而装饰器修改函数时对象甚至还没有创建。

Stack，我该怎么做？

---

注意事项:

1) 我知道我可以使用基类和继承来做到这一点，但是当类中有多个方法要以这种方式装饰时，这就会成为一个问题。

2) 实际问题来自使用 theano - 标准格式是 outputs, updates = fcn(*inputs)，但大多数函数不返回任何更新，所以你希望能够以自然的方式定义这些函数，但仍然可以选择根据此标准接口(interface)调用它们。

Answer 1

这确实是一个问题，因为从函数中检索“绑定(bind)”方法的方式并没有考虑这种方式。

我看到两种方式:

1. 您可以只包装函数:

   def single_return_format(fcn):
       # TODO Do some functools.wraps here...
       return lambda *args, **kwargs: (fcn(*args, **kwargs), )
   

   不要乱用 .standard_format，而只是函数的替换。所以函数可以定义自己为返回一个值，但只能作为返回元组被调用。

2. 如果这不是您想要的，您可以定义一个类来装饰方法，它会覆盖 __get__ 并以“实时方式”进行包装。当然，它也可以重新定义 __call__，以便它也可用于(独立的、非方法的)函数。