[英]How do I use the python-decorator package to decorate classmethods?
我有一个装饰器,我想用来装饰类方法。 在下面的示例中,@ 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.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)
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.