一、装饰器的基本语法和使用
-
什么是装饰器
是一种编程的设计模式,python中只要是可调用的对象,都可以当成装饰器来用
-
什么是装饰器的开放封闭原则
官方定义是:软件实体应该是可扩展的,但不可以进行修改,也就是说,对外扩展是开放的,而对修改是封闭的
-
装饰器的语法
被调用函数加上:@装饰器函数,
例如:
@pytest.mark
@classmethod -
装饰器的作用
在不更改原功能函数内部代码,并且不改变调用方法的情况下,为原代码扩展新的功能
-
装饰器装饰的原理
- 调用装饰器:并且把被装饰的函数,当成参数传入到装饰器中,保存在封闭作用域内
- 被装饰的函数执行时:执行的是装饰器中的内层函数,在内层函数中可以进行新的功能代码的扩展,然后再调用原功能函数
例如:
我现在定义一个函数work1,调用函数会打印原函数功能def work1(): print("----写代码---")
那我要在他的基础上扩展新功能,扩展的新功能就是装饰器函数
我在定义一个闭包函数decorator
- decorator 函数 :传进来的参数 func 就是被调用函数
- wrapper 函数 :执行的是装饰器中的内层函数,在内层函数中对新功能代码的扩展,然后再调用原功能函数
def decorator(func): def wrapper(): # 新功能代码的扩展 print("---开机,打开Pycharm--") # 调用被装饰的函数 func() # 新功能代码的扩展 print("---关机,提交代码---") return wrapper
最终的实现装饰器的调用:
# 装饰器函数 def decorator(func): def wrapper(): # 新功能代码的扩展 print("---开机,打开Pycharm--") # 调用被装饰的函数 func() # 新功能代码的扩展 print("---关机,提交代码---") return wrapper # 函数1 @decorator -----装饰器 def work1(): print("----写代码---") work1() # 调用原函数
执行结果:
二、装饰器的基本结构(无参数和返回结果)
-
装饰器的基本结构:
def decorator(func): def wrapper(): print("在功能函数调用之前扩展的代码1") print("在功能函数调用之前扩展的代码2") # 调用被装饰的函数 func() print("在功能函数调用之后扩展的代码1") print("在功能函数调用之后扩展的代码2") return wrapper # 外层函数return返回的是 内层函数
-
实例:假如实现一个可以统计任意函数执行时间的装饰器
# 无参数和返回结果的装饰器 def coubnt_time(func): def wrap(): # 获取函数调用之前的时间 start_time = time.time() # 调用被装饰的函数 func() # 获取函数调用之后的时间 end_start = time.time() print("执行时间:", end_start - start_time) return wrap # 被装饰的函数1 @coubnt_time # ----装饰器 def time_work2(): print("这是time_work2的---stert-------") time.sleep(1) print("这是time_work2的---end-------") # 函数调用 time_work2()
执行结果:
三、装饰器装饰带有参数和返回结果的函数实现
-
如果被调用的函数有参数和返回结果,而定义的装饰器没有去处理,就会报错 :
TypeError: time_work() takes 0 positional arguments but 2 were given:函数有2个参数,但是装饰器没有参数传递
-
带有参数和返回结果的装饰器怎么处理
1.在第二层函数的参数传递需要加上被调用函数的参数,有几个参数传几个参数
2.在第一次函数的参数值,在被调用时也需要传递被调用函数的参数,有几个参数传几个参数,并使用result接收返回值
3.装饰器中的代码执行完后,将原功能函数的结果返回出去 -
带参数和结果处理的基本结构如下
# 装饰器(带参数和结果处理) def decorator(func): # 加上被调用函数的参数,有几个参数传几个参数 def wrapper(a,b): print("在功能函数调用之前扩展的代码1") print("在功能函数调用之前扩展的代码2") # 调用被装饰的函数,并使用result接收返回值 result = func(a,b) print("在功能函数调用之后扩展的代码1") print("在功能函数调用之后扩展的代码2") # 装饰器中的代码执行完后,将原功能函数的结果返回出去 return result return wrapper
-
实例:还是实现一个可以统计任意函数执行时间的装饰器
# 无参数和返回结果的装饰器 def coubnt_time(func): # 加上被调用函数的参数,有几个参数传几个参数 def time_work(a,b): # 获取函数调用之前的时间 start_time = time.time() # 调用被装饰的函数,并使用result接收返回值 result = func(a,b) # 获取函数调用之后的时间 end_start = time.time() print("执行时间:", end_start - start_time) # 装饰器中的代码执行完后,将原功能函数的结果返回出去 return result return time_work # 被装饰的函数1有参数传递 @coubnt_time # ----装饰器 def work(a,b): time.sleep(1) return a + b # 函数调用 print(work(1,2))
执行结果:
四、装饰器的通用结构(无论是否有参数)
-
那如果定义的装饰器传递的参数是写死的,被调用的A函数有参数和返回结果需要,而B函数没有参数需要传递,对装饰器没有去做处理,就会报错:
TypeError: time_work() takes 0 positional arguments but 2 were given:缺少 2 个必需的位置参数传递
-
针对这种有不同的函数,有的需要传参数,有的不需要,装饰器应该如何实现:
解决这种情况:在参数传递这里就不能够写死,需要灵活传参,这就要用到 不定长参数 (*args, **kwargs)去传递。
1.在第二层函数处需要加上被调用函数的参数,加上 不定长参数(*args, **kwargs)
2.在第一次函数的参数值,在被调用时也需要加上不定长参数(*args, **kwargs),并使用result接收返回值
3.装饰器中的代码执行完后,将原功能函数的结果返回出去 -
实现通用的装饰器的作用:
1.可以装饰有参数的函数,也可以装饰没有参数的函数
2.不管函数有没有返回值都适用 -
通用装饰器的基本结构:
def decorator(func): # 传不定长参数(*args, **kwargs) def wrapper(*args,**kwargs): print("在功能函数调用之前扩展的代码1") print("在功能函数调用之前扩展的代码2") # 传不定长参数(*args, **kwargs),调用被装饰的函数, 并使用result接收返回值 result = func(*args,**kwargs) print("在功能函数调用之后扩展的代码1") print("在功能函数调用之后扩展的代码2") # 装饰器中的代码执行完后,将原功能函数的结果返回出去 return result return wrapper
-
实例:还是实现一个可以统计任意函数执行时间的装饰器
# 通用装饰器 def count_time(func): # 传不定长参数(*args, **kwargs) def time_work(*args,**kwargs): # 获取函数调用之前的时间 start_time = time.time() # # 传不定长参数(*args, **kwargs),调用被装饰的函数, 并使用result接收返回值 result = func(*args,**kwargs) # 获取函数调用之后的时间 end_start = time.time() print("执行时间:", end_start - start_time) # 装饰器中的代码执行完后,将原功能函数的结果返回出去 return result return time_work # 函数1:有参数,需要返回结果 @count_time def count_number(a,b): time.sleep(1) return a * b # 函数2:无参数,无返回结果 @count_time def count_number1(): for i in range(2): time.sleep(1) # 函数3:很多参数传递,需要返回结果 @count_time def count_number2(aa,bb,cc,dd,ee,ff,gg=1000): print(aa,bb,cc,dd,ee,ff,gg) print(count_number(11,22)) count_number1() count_number2(1,2,3,4,5,6)
执行结果:
五、装饰器带参数的实现
-
之前都是被调用的函数带了参数,那如果装饰器也带了参数,如果装饰器没有去接收,会报错:
由于在装饰器的参数传的是:字符串类型的参数,就会报 TypeError: ‘str’ object is not callable:“str”对象不可调用
-
装饰器带了参数,整个装饰器应该如何实现
当装饰器带有参数去传递,例如:@decorator(‘henry’)
解决方式是:需要在定义装饰器函数时,需要额外定义一个函数去接收装饰器传进来的参数
-
支持参数传递的装饰器结构
def decorator(name): ----最外层的参数,接收装饰器传进来的值 def wrapper2(func): ----第二层参数func,接收被装饰器的函数 def wrapper3(*args, **kwargs): ----第三层参数,接收被装饰的函数,调用时传进来的参数 print("装饰器扩展代码1") result = func(*args,**kwargs) print("装饰器扩展代码2") ---返回result 结果 return result ---返回第三层结果 return wrapper3 ---返回第二层结果 return wrapper2
-
实例:还是实现一个可以统计任意函数执行时间的装饰器
# 支持参数传递的装饰器实现 def decorator(name): """最外层的参数,接收装饰器传进来的值""" def wrapper2(func): """第二层参数func,接收被装饰器的函数""" def wrapper3(*args, **kwargs): """第三层参数,接收被装饰的函数,调用时传进来的参数""" print(name) # 获取函数调用之前的时间 start_time = time.time() # 传不定长参数(*args, **kwargs),并使用result接收 result = func(*args,**kwargs) # 获取函数调用之后的时间 end_start = time.time() print("执行时间:", end_start - start_time) return result # 返回第三层结果 return wrapper3 # 返回第二层结果 return wrapper2 # 装饰器带参数 @decorator('henry') def work(): time.sleep(1) print("python6666") # 装饰器相当于执行以下代码: # work = decorator('henry')(work) work()
执行结果:
六、装饰器如何装饰类
- 装饰器不仅仅可以装饰函数,还可以装饰类
- 闭包形式的装饰器装饰类:可以在类实例对象的时候,进行功能扩展
-
装饰器装饰类与装饰函数一样,没有区别
# 通用装饰器 def decorator(item): def wrapper(*args, **kwargs): print("装饰器扩展代码1") result = item(*args,**kwargs) # 给 result.name = "henry" print("装饰器扩展代码2") return result return wrapper # 定义一个类 # @decorator class Mytest: a = 100 # 类属性 # # 装饰器相当于执行以下代码: # Mytest = decorator(Mytest) m = Mytest() print(m.name)
七、装饰器的副作用
-
装饰器的副作用
1.装饰器在装饰之后会出现副作用,无法获取原功能函数的特殊属性
2.特殊所属性,例如:函数名字,文档注释 -
如何消除装饰器的副作用
1.使用functools库中的 wraps 模块
2.使用 @wraps(原函数) 消除 -
实例,消除装饰器的副作用
首先定义一个通用装饰器函数:
def count_time(func): def time_work(*args,**kwargs): # 获取函数调用之前的时间 start_time = time.time() # 调用原功能函数,并使用result接收返回值 result = func(*args,**kwargs) # 获取函数调用之后的时间 end_start = time.time() print("执行时间:", end_start - start_time) # 装饰器中的代码执行完后,将原功能函数的结果返回出去 return result return time_work
- 在定义几个函数,不调用装饰器函数,去打印函数1和函数2 的几个特殊属性
# 这个函数1:带有参数需要传递 def work1(a,b): '''这是work1的文档注释''' time.sleep(1) return a + b # 这个函数2:没有参数需要传递 def work2(): '''这是work2的文档注释''' for i in range(2): time.sleep(1) # 打印函数名称和注释文字 print(work1.__name__, work1.__doc__) print(work2.__name__, work2.__doc__)
打印的结果显示各自获取到对应函数的属性:
- 这次去调用装饰器函数函数,去打印函数1和函数2 的几个特殊属性看看
# 这个函数1:带有参数需要传递 @count_time def work1(a,b): '''这是work1的文档注释''' res = a + b time.sleep(1) return res # 这个函数2:没有参数需要传递 @count_time def work2(): '''这是work2的文档注释''' for i in range(2): time.sleep(1) # 装饰器装饰之后会出现副作用,无法获取原功能函数的特殊属性 print(work1.__name__, work1.__doc__) print(work2.__name__, work2.__doc__)
打印的结果显示注释为空,函数名是同一个:
-
怎么解决
只需要增加一行代码,使用 @wraps(原函数) 消除即可,获取到