我有一个装饰器,我想用它来装饰类方法.在下面的示例中,@mydec装饰器可以正常工作,但是在使用help()或pydoc时,它不会保留函数签名.为了解决这个问题,我研究了使用@decorator python-decorator包:
import functools
import decorator
@decorator.decorator
def mydec(func):
@functools.wraps(func)
def inner(cls, *args, **kwargs):
# do some stuff
return func(cls, *args, **kwargs)
return inner
class Foo(object):
@classmethod
@mydec
def bar(cls, baz='test', qux=None):
print (baz, qux)
Foo.bar()
不幸的是,这导致以下异常:
Traceback (most recent call last):
File "/tmp/test.py", line 21, in <module>
Foo.bar()
File "<string>", line 2, in bar
TypeError: mydec() takes exactly 1 argument (4 given)
解决方法:
您不再需要提供自己的包装器,只需在内部函数上使用@ decorator.decorator即可,该函数需要一个额外的第一个位置参数,即包装的函数:
@decorator.decorator
def mydec(func, cls, *args, **kwargs):
# do some stuff
return func(cls, *args, **kwargs)
装饰器包不对装饰器使用闭包,而是将包装函数作为参数传递.
演示:
>>> @decorator.decorator
... def mydec(func, cls, *args, **kwargs):
... # do some stuff
... return func(cls, *args, **kwargs)
...
>>> class Foo(object):
... @classmethod
... @mydec
... def bar(cls, baz='test', qux=None):
... print (baz, qux)
...
>>> Foo.bar()
('test', None)