python装饰器

装饰器

概念：可以在不修改原来代码的情况下（函数原有的功能或者类原有的功能），为需要被装饰的函数或者类增加新的功能或者添加限制调剂以及帮助输出

常用种类

• 函数的装饰器
• 类的装饰器
• 不管是哪种 类型的装饰器设计模式原则是：开放封闭的原则（对外扩展开发，对内关闭修改）

1.1 装饰器的定义

装饰器的必要构成条件

• 装饰器本身是一个函数
• 装饰器的返回值一定是一个函数的引用（函数的名字）
• 装饰器只能必须一定是只有一个形参
• 形参是用来接收函数的引用（需要被装饰的函数的名字）

装饰器本质上是一个闭包函数

#装饰器实现登录
def check(func): # func=shopping
    def inner():
         print("开始进入登录页面")
         print("扫码登录")
         print("登录成功")
         func()   #func=shopping func()=shopping()
    return inner


#原有的购物功能
def shopping():
    print("商品加入购物车")
    print("付款")


#原始使用方式：调用装饰器，传递需要被装饰的函数作为实参
shopping1=check(shopping)  # 2---3(inner函数不会被执行，但是加载到内存)--8--18
#check(shopping)返回结果：shopping1=inner
shopping1() #shopping1()=innner() 3-4-5-6-7--12--13--14

 1.2 装饰器的语法糖使用

python提供了装饰器的精简的书写格式：语法糖格式

使用@符号加装饰器的名字即可：@check

注意：装饰器语法糖一定要写在被调用函数的上面

#装饰器实现登录
def check(func): # func=shopping
     def inner():
          print("开始进入登录页面")
          print("扫码登录")
          print("登录成功")
          func()   #func=shopping func()=shopping()
     return inner


 #原有的购物功能
@check #说明：shopping=check(shopping)----shopping=inner
def shopping():
     print("商品加入购物车")
     print("付款")

#shopping()=inner()
shopping()

#被装饰的函数赋值给了func形参
#func=shopping
#装饰器被调用完返回结果是inner
#shopping=inner
#shopping()=inner()

语法糖@装饰器名 执行方式

• 在整个模块开始运行代码的时候，会优先执行装饰器代码，装饰器不需要手动调用，自动执行
• 装饰器的本质就是一个闭包函数，只不过传递过去的实参是被装饰的函数的引用（函数的名字）

1.3 装饰器的使用场景

• 统计一个函数的执行时间
• 输出日志信息
• 缓存处理
• 实现路由

import time

#统计一个函数的执行时间

def get_time(func):
    def inner():
        begin=time.time()
        func()
        end=time.time()
        print(f"被装饰的函数执行的时间为{end-begin}")

    return  inner

@get_time
def func1():
    for i in range(1,1000000):
        print(i)

func1()

 1.4 带有参数的装饰器

#用带有参数的装饰器完成计数器的加减乘除功能
def mark(flag):#flag用来接收装饰器的实参
    def out_func(fn): #用来接收被装饰器的函数引用
        def inner_func(num1,num2): #需要被装饰的函数的实参
            if flag=="+":
                print(f"您输入的装饰器实参是{flag},正在努力进行加法运算中")
                result=fn(num1,num2)
            elif flag=="-":
                print(f"您输入的装饰器实参是{flag},正在努力进行减法运算中")
                result=fn(num1,num2)
            else:
                result="您输入的运算符号不在范围内"
            return result

        return inner_func
    return out_func

# 带有参数的装饰器代码执行流程：
#2---->16---->3-->15--->4--->5(8)--->6(9)(11)---7(10)(12)---(13)

#装饰器语法糖的调用过程
#mark=mark("+")----mark=out_func
# @mark("+") 
# def add(a,b):
#     result=a+b
#     return result
#-------------------------------------
# @out_func
# def add(a,b):
#     result=a+b
#     return result
#add=out_func(add)=inner_func
#add=inner_func()
@mark("+")
def add(a,b):
    result=a+b
    return result

@mark("-")
def sub(a,b):
    result=a - b
    return result

#add=inner_func()
print(add(22,56))
print(sub(15,6))