Python小白学习之函数装饰器

装饰器

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))

执行结果:

Python小白学习之函数装饰器

使用装饰器:

下面程序分为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()

执行结果如下:

Python小白学习之函数装饰器

可以看到执行结果和不使用装饰器一样的。看上去使用装饰器复杂了,实际上,对于大量执行计时器操作,就不用重复定义变量并打印结果了,我们使用装饰器只需改变少量实参即可即可省去重复操作,因为我们把一部分重复操作封装在装饰函数中,@装饰函数名。

带参数的被装饰函数上面是不带参数的装饰器,下面是带参数的装饰器,注意: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(,)

执行结果如下:

Python小白学习之函数装饰器

带参数的装饰器(带参数的语法糖)

不改变内部被装饰函数,通过在语法糖(@装饰函数名)后面添加实参和并且在装饰函数再加一层外层函数

 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)
 # 装饰器带参数;在语法糖上带参数
 # 可以看到一层嵌套函数,出入的参数满了,我们可以在最外面加一层嵌套函数

执行结果如下:

Python小白学习之函数装饰器


小白学习笔记,如有理解错误,望批评指正,大佬止步。
上一篇:ethereumjs/ethereumjs-vm-3-StateManager


下一篇:配置sudo日志审计