装饰器是什么
在不改变源代码的基础上扩展函数需要的新需求,这就是装饰器。(不改变函数源代码,这也是装饰器最基本的原则)
装饰器,它本身也就是一个函数,应用高阶函数实现。
使用时一般把被装饰的函数的内存地址当参数传入装饰器函数体,通过参数调用被装饰的函数,获得或者修改其属性。
装饰器的使用
一般认为,装饰器使用遵循一定格式,即:高阶函数+高阶函数内嵌套函数。
假设有一个需求,我要知道正在运行什么函数,在运行时输出。
有一个比较粗暴的方法。
def func1():
print(func1.__name__)
pass
def func2():
print(func2.__name__)
pass
def func3():
print(func3.__name__)
pass
if name == '__main__':
func1()
func2()
func3()
这一定是对的,但是不自然就想到两个字“封装”,虽然这和装饰器大相径庭,但是两个概念确实有相通之处。
按照装饰器原则,可以这样写。
def funcname(func):
def out():
func()
print(func.__name__)
return out
@funcname
def func1():
pass
@funcname
def func2():
pass
@funcname
def func3():
pass
if name == '__main__':
func1()
func2()
func3()
装饰器的进一步使用
在之前的这个小实验中,基本上了解了装饰器的用途和写法,然而不是所有的函数都想func1这样简单,无参数。
对于含不定参数的函数,参数由收集参数获取。
def funcname(func):
def out(*args, **kwargs):
func(*args, **kwargs)
print(func.__name__)
return out
@funcname
def func1(a, b):
print(a+b)
@funcname
def func2(a, b, c):
print(a+b+c)
@funcname
def func3(a, b, c, d):
print(a+b+c+d)
if name == '__main__':
func1(1, 2)
func2(1, 2, 3)
func3(1, 2, 3, 4)