python进阶:装饰器

一、何为装饰器?

<一>、装饰器

上一篇文章讲到了函数中的闭包,这篇文章讲讲python中的装饰器

def my_decorator(func):
	def wrapper():
		print('call wrapper')
		func()
	return wrapper

def greet():
	print('call greet')

greet = my_decorator(greet)
greet()

上述这段代码中,变量greet指向了内部函数wrapper(),而内部函数wrapper()中又会调用原函数greet(),因此,最后调用greet()时,就会先打印’call wrapper’,然后输出’call greet’
这里的函数my_decorator()就是一个装饰器,它把真正需要执行的函数greet()包裹在其中,并且改变了它的行为,但是原函数greet()不变

def my_decorator(func):
	def wrapper():
		print('call wrapper')
		func()
	return wrapper

@my_decorator
def greet():
	print('call greet')
greet()

上述代码是python装饰器更为简单的表示,@my_decorator相当于前面的greet = my_decorator(greet)语句,@称为语法糖

<二>、带有参数的装饰器

若原函数比如上述的greet()有参数怎么办?可以在对应的装饰器函数wrapper()加上相应的参数

def my_decorator(func):
	def wrapper(message):
		print('call wrapper')
		func(message)
	return wrapper

@my_decorator
def greet(message):