Python：类装饰器

我在另一篇博客中介绍了Python装饰器的基本形式和原理，有兴趣的小伙伴可以看一下：

初识装饰器

我们已经知道以下两种方式是等价的：

@dec	# 方式一：使用@符号为fun添加装饰器
def fun():
	pass

# fun = dec(fun)	# 方式二： 将fun当做参数创给dec，然后由dec返回一个新的函数

观察第二种方式，它的形式和一个类生成实例是相似的，dec是可调用对象（函数和类都是可调用对象），接收参数（函数调用，类调用（即生成实例），这两个过程都能够接收参数）， 返回一个的对象（函数返回值是一个对象，类调用后生成的实例也是一个对象）

如果我们将dec写成一个类，大致形式如下：

class Dec:
	def __init__(self, fun):
		self.fun = fun

@Dec
def test_fun():
	pass

# test_fun = Dec(test_fun)

print(type(test_fun))

<class '__main__.Dec'>

此时，test_fun是Dec的一个实例，它已经不是函数了，但现在它还不是可调用对象。
一个类的实例如果要成为可调用对象，它应当实现__call__方法。
实例在调用时所接收的参数，都会传递到__call__方法中

class Dec:
	def __init__(self, fun):
		self.fun = fun
	
	def __call__(self, *args, **kwargs):
		return self.fun(*args, **kwargs)

@Dec
def test_fun():
	print('this is test_fun')

# test_fun = Dec(test_fun)

print(type(test_fun))
test_fun()

<class '__main__.Dec'>
this is test_fun

这样一来，调用test_fun这个由Dec类生成的实例，和调用未装饰前的test_fun，变得一模一样了。
知道了基本过程之后，我们要想增加额外的功能，只需要在Dec类中实现就好。

举个例子：

import time

class Cache:
	"""
	这是一个类装饰器
	被装饰的函数在调用时，返回缓存的结果
	缓存结果的过期时间是5秒
	"""
	def __init(self, fun):
		self.fun = fun
		self.res = ()
		
	def __call__(self, *args, **kwargs):
		if not self.res or time.time() - self.res[1] > 5:
			rt = self.fun(*args, **kwargs)
			self.res = (rt, time.time())
		return self.res

@Cache
def f():
	return time.time()	# 如果没有缓存，那么调用f每次都会返回不一样的值

for _ in range(10):
	print(f())
	time.sleep(1)

1570087517.895919
1570087517.895919
1570087517.895919
1570087517.895919
1570087517.895919
1570087522.8972712
1570087522.8972712
1570087522.8972712
1570087522.8972712
1570087522.8972712

总结起来，类装饰器和函数装饰器差别还是蛮大的，两者外在形式虽然一样，但是前者返回的是一个类的实例，需要在类中定义好__call__方法，来让这个实例变成可调用对象。
实例在初始化时，被装饰函数作为参数加入到了实例的初始化过程中，被绑定为该实例的某个属性，然后在__call__方法中，以某种形式来调用它即可。