有没有可能将装饰器类应用于类的成员函数呢?根据之前的分析,似乎是可以的,代码:
from functools import wraps
class Profiled:
def __init__(self, func):
wraps(func)(self)
def __call__(self, *args, **kwargs):
print("call")
return self.__wrapped__(*args, **kwargs)
class Spam:
@Profiled
def bar(self, x):
print(self, x)
s = Spam()
s.bar(1)
执行结果:
call
Traceback (most recent call last):
File "decorator146.py", line 21, in <module>
s.bar(1)
File "decorator146.py", line 12, in __call__
return self.__wrapped__(*args, **kwargs)
TypeError: bar() missing 1 required positional argument: 'x'
似乎出现了问题。
当装饰器类应用于类成员函数时,类成员函数变成什么样了?
from functools import wraps
class Profiled:
def __init__(self, func):
wraps(func)(self)
def __call__(self, *args, **kwargs):
print("call")
return self.__wrapped__(*args, **kwargs)
class Spam:
@Profiled
def bar(self, x):
print(self, x)
s = Spam()
print(s)
print(s.bar)
print(s.__dict__)
print(Spam.__dict__)
运行结果:
<__main__.Spam object at 0x7f363ceae550>
<__main__.Profiled object at 0x7f363d10d6a0>
{}
{'__module__': '__main__', 'bar': <__main__.Profiled object at
0x7fd313c686a0>, '__dict__': <attribute '__dict__' of 'Spam' objects>,
'__weakref__': <attribute '__weakref__' of 'Spam' objects>, '__doc__':
None}
发现此时bar
的属性竟然变为一个类的属性!
非常类似通过以下代码定义的Spam
class Spam:
bar = Profiled()
此时我们继续理解之前将类作为装饰器应用于成员函数时。
s.bar(1)
调用分为2个步骤:
- s.bar 获取bar的类实例
- s.bar(1),对类实例进行调用
第一条需要根据之前所描述,需要实现__get__
函数
第二条需要实现__call__
属性
备注:其实包含3步,还有一步是__init__
正确的方法
import types
from functools import wraps
class Profiled:
def __init__(self, func):
wraps(func)(self)
def __call__(self, *args, **kwargs):
print("call")
return self.__wrapped__(*args, **kwargs)
def __get__(self, instance, cls):
if instance is None:
print('1')
print(self)
return self
else:
print('2')
print(types.MethodType(self, instance))
return types.MethodType(self, instance)
class Spam:
@Profiled
def bar(self, x):
print(self,x)
s = Spam()
# print(s)
# print(s.bar)
# print(s.__dict__)
# print(Spam.__dict__)
s.bar(1)
运行结果:
2
<bound method Spam.bar of <__main__.Spam object at 0x7fc08562c5c0>>
call
<__main__.Spam object at 0x7fc08562c5c0> 1
结果是预期的,
至于__get__
函数的三个参数的意义详解可以参考Python官网文档,这里只是大概说明
instance表示所在的类
,上述代码就是Spam
,cls
参数指的就是Profiled
至于types.MethodType
表示的意义就是什么就查看官网文档就行。
当然这里还有一些问题似乎无法理解,继续看看types.MethodType
等就可以理解。这里不再累述了。
小结
将装饰器类应用于类的成员函数其实是最复杂的场景,当然装饰器应用的场景远不止提到的这几种,在<Python学习手册>里就介绍了很多种,可以看看教材中其余装饰器的使用,不过基本原理就是连载里所提到的。