装饰器
2018-10-25 13:49:37
装饰器从字面意思就是用来装饰的,在函数可以理解为:在函数中,我们不想影响原来的函数功能,又想给函数添加新的功能,这时候我们就用到了装饰器。
一般函数操作(不用装饰器):
import time print(time.time()) def i_can_sleep(): time.sleep(3) start = time.time() i_can_sleep() stop = time.time() print("函数运行了%s"%(stop - start))
执行结果:
使用装饰器:
下面程序分为4步走(外部函数、返回嵌套函数、嵌套函数+语法糖(@装饰函数简写)+被装饰函数+调用被装饰的函数);若有大量的调用函数i_can_sleep并查看程序运行了多长时间,我们就嘚不厌其烦的定义新的start和stop变量,然后打印时间
为此,我们就需要更简单的方法,装饰器,把这些重复的工作封装到装饰函数中也就是封装到@装饰函数名(语法糖)
代码如下:
################################### import time def timer(fun):#参数是我们要传入的被装饰函数,记下面的i_can_sleep函数 #1 #此处我们需要在函数内部再定义一个内部函数,即嵌套函数 def decorator(): start = time.time() fun() #调用被装饰的函数,下面的i_can_sleep() stop = time.time() print('被装饰的函数i_can_sleep程序执行时间为:',stop-start) #给被装饰函数添加计时器功能 return decorator @timer #装饰函数 # 2 def i_can_sleep():#我们要装饰的函数,即被装饰函数 # 3 time.sleep(3) i_can_sleep() #最后一步调用程序 # 4 # 其实上面的第四部也可以按下面写,只是装饰器简化下面的步骤。 #next = timer(i_can_sleep);next()
执行结果如下:
可以看到执行结果和不使用装饰器一样的。看上去使用装饰器复杂了,实际上,对于大量执行计时器操作,就不用重复定义变量并打印结果了,我们使用装饰器只需改变少量实参即可即可省去重复操作,因为我们把一部分重复操作封装在装饰函数中,@装饰函数名。 带参数的被装饰函数上面是不带参数的装饰器,下面是带参数的装饰器,注意:1、参数放在内部函数中#1;2、被装饰的函数也要加上参数 看#2;定义被装饰的函数有参数看#3、#4。3、定义被装饰函数时(语法糖下面的函数是被装饰函数),形参名字和上面的嵌套函数可以不一致。
print("---------------------------------------------------") # 带参数的装饰器 def wai(func): def nei(a,b): #参数放在内部函数中 #1 print('start') func(a,b) #被装饰函数调用也要加上参数 #2 print('stop') return nei @wai def add(m,n): #加法 #3 return print(m+n) #参数名和嵌套函数中的参数名字可以不同。 add(,) # 可以多次使用语法糖调用装饰器 @wai def sub(x,y): #减法 #4 return print(x-y) sub(,)
执行结果如下:
带参数的装饰器(带参数的语法糖)
不改变内部被装饰函数,通过在语法糖(@装饰函数名)后面添加实参和并且在装饰函数再加一层外层函数
def add_chars(argv): def wai(func): def nei(a,b): print('开始:%s'%argv) func(a,b) print('结束') return nei return wai @add_chars('加法') #语法糖名字要和多层嵌套函数最外层函数名一致,并且参数对应最外层函数形参 def add(m,n): return print(m,'+',n,'=',m+n) add(4,6) @add_chars('减法') # def sub(x,y): #减法 return print(x,'-',y,'=',x-y) sub(11,4) # 装饰器带参数;在语法糖上带参数 # 可以看到一层嵌套函数,出入的参数满了,我们可以在最外面加一层嵌套函数
执行结果如下:
小白学习笔记,如有理解错误,望批评指正,大佬止步。