python-装饰器

python-装饰器

简单的来理解，装饰器就是在你执行一个函数之前，想保存一下运行日志。而依据开闭原则，直接修改原函数并不是一个明智的选择，这种既不能修改原代码，又要实现新功能的拓展，就是装饰器的主要作用。
装饰器函数
装饰器函数是一个闭包函数，内部函数调用目标函数或返回目标函数的调用，外部函数返回内部函数的引用，需要扩展的额外功能编写在内部函数中。
eg：

def wrapper(func):
	def inner(*args, **kwargs):
		#扩展新的功能：如打印日志、权限验证等
		print('这里是装饰器起作用了')
		return func(*args, **kwargs)
	return inner


@wrapper
def demo():
	print('这是目标函数')

if __name__ == '__main__':
	demo()

上述示例的运行结果为：

这里是装饰器起作用了
这是目标函数

Process finished with exit code 0

可以看到先执行了装饰器中的内容，再执行了目标函数。

类装饰器
装饰器也可写为类的形式：

class Logit:
	def __call__(self, func)
		def wrapper(*args, **kwargs):
			# 进行扩展功能的操作
			return func(*args, **kwargs)
		return wrapper 

当装饰器有传入的参数时，可在类中加上__init__初始化函数:

def __init__(self, name):
	self.name=name

额外补充
被装饰器装饰的函数，函数名会发生改变，例如第一个例子，若在程序的最后打印demo.name。那么你会得到的是inner
解决这个问题的办法是：

from functools import wraps

inner函数加上装饰器@wraps(func)