python-装饰器

装饰器

• 装饰器是高阶函数，是对传入函数功能的装饰，即进行增强

无参装饰器

• 是一个函数
• 函数作为形参
• 返回值为函数
• 可以使用@functionname来简化调用

1. 现有一个加法函数，想对其进行输出方式的加强，重定义一个logger函数，在函数内调用已经定义的加法函数，从而达到对功能的加强

def add(x,y):
	return x + y

def logger(fn):
	print('begin')
	x = fn(4,5)
	print('end')
	return x
#调用过程为：
logger(add)

2. 以上形式将函数的传参写在函数内部，使用不灵活，因此将传参做以下改动，实现参数的配置

def add(x,y):
	return x + y

def logger(fn,*args,**args):
	print('begin')
	x = fn(*args,**args)
	print('end')
	return x
#调用过程为：
logger(add,5,y=6)

3. 将改变传参后的logger函数进行柯里化

def add(x,y):
	return x + y

def logger(fn):
	def wrapper(*args,**args):
		print('begin')
		x = fn(*args,**args)
		print('end')
		return x
	return wrapper

#调用过过程为：
logger(add)(5,y=6)
换种写法为：
add = logger(add)
add(5,y=6)#此处的add并非原始的add函数，而是logger(add)，即logger(add)

4. 柯里化后的函数可作为一个装饰器来使用，使用装饰器可写为以下形式

def logger(fn):
	def wrapper(*args,**args):
		print('begin')
		x = fn(*args,**args)
		print('end')
		return x
	return wrapper

@logger #等同于add = logger(add)
def add(x,y):
	return x + y

#调用时为：
add(5,6)

@logger即为装饰器语法

带参装饰器

• 是一个函数
• 函数作为形参
• 返回值是一个不带参数的装饰器函数
• 可以使用@functionname(参数列表)来进行调用，参数列表可省，但小括号不可省略

若要对add函数的执行时长进行判断并比较，则另外需要加一个duration参数，首先想到的写法是将duration参数作为logger的传参，写法如下：

def logger(fn,duration):
	def wrapper(*args,**args):
		print('begin')
		start = detetime.datetime.now()
		x = fn(*args,**args)
		delta = (datetime.datetime.now()-start).total_second()
		print('so slow') if delta > duretion
		print('end')
		return x
	return wrapper

以上形式不可用无参装饰器进行调用，需再次对其进行柯里化，写法如下：

'''
获取函数执行时长，并对超阈值的函数进行记录
'''
def logger(duration):
	def _logger(fn):
		def wrapper(*args,**args):
			print('begin')
			start = detetime.datetime.now()
			x = fn(*args,**args)
			delta = (datetime.datetime.now()-start).total_second()
			print('so slow') if delta > duretion
			print('end')
			return x
		return wrapper
	return _logger

@logger(5)#相当于logger(5)(add)
def add(x,y):
	return x + y

一般在工程实现中不会编写超过三层的嵌套函数，若此时需再次加入参数只需在logger的形参中进行添加，无需再次进行柯里化

def logger(duration,x,y):
	def _logger(fn):
		def wrapper(*args,**args):
			print('begin')
			start = detetime.datetime.now()
			x = fn(*args,**args)
			delta = (datetime.datetime.now()-start).total_second()
			print('so slow') if delta > duretion
			print('end')
			return x
		return wrapper
	return _logger

#调用如下：
@logger(5,9,2)#相当于logger(5,9,2)(add)
def add(x,y):
	return x + y