我希望能够根据某种标准格式调用方法:
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)调用它们。
最佳答案
这确实是一个问题,因为从函数中检索“绑定(bind)”方法的方式并没有考虑这种方式。
我看到两种方式:
您可以只包装函数:
def single_return_format(fcn): # TODO Do some functools.wraps here... return lambda *args, **kwargs: (fcn(*args, **kwargs), )
不要乱用
.standard_format
,而只是函数的替换。所以函数可以定义自己为返回一个值,但只能作为返回元组被调用。如果这不是您想要的,您可以定义一个类来装饰方法,它会覆盖
__get__
并以“实时方式”进行包装。当然,它也可以重新定义__call__
,以便它也可用于(独立的、非方法的)函数。
关于python - 使用 Python 装饰器为方法添加方法,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/26725593/