装饰器

一. 装饰器

定义:假设我们要增强函数的功能，比如，在函数调用前后自动打印日志，但又不希望修改函数的定义，这种在代码运行期间动态增加功能的方式，称之为“装饰器”（Decorator）。

原则:基于开放封闭原则

封闭原则:对修改封闭，意味着类一旦设计完成，就可以独立完成其工作，而不要对类进行任何修改。

开放原则:对扩展开放，意味着有新的需求或变化时，可以对现有代码进行扩展，以适应新的情况。

1.实现目标:登录功能和index功能

def login():
    user=input("请输入用户名:")
    pwd=input("请输入密码:")
    if user=="shy" and pwd=="123":
        print("登录成功")
    else:
        print("用户名或密码错误")
def index():
    print("this is index")

2.只有登录才能调用index函数

def login():
    user=input("请输入用户名:")
    pwd=input("请输入密码:")
    if user=="shy" and pwd=="123":
        index()
    else:
        print("用户名或密码错误")
def index():
    print("this is index")
login()

结果:只能调用index函数,不灵活

3.登录后能调用各种函数

def login(func):
    user=input("请输入用户名:")
    pwd=input("请输入密码:")
    if user=="shy" and pwd=="123":
        func()
    else:
        print("用户名或密码错误")
def index():
    print("this is index")
def foo():
    print("this is foo")
login(foo)

结果:不符合开放原则,导致所有调用index,或foo的地方都要更改

4.在保证index,foo开放的前提下,添加登录功能

def login(func):
   def inner():
       user = input("请输入用户名:")
       pwd = input("请输入密码:")
       if user == "shy" and pwd == "123":
           func()
       else:
           print("用户名或密码错误")
   return inner

def index():
    print("this is index")
def foo():
    print("this is foo")
index=login(index)
index()
foo=login(foo)
foo()

结果:每一个只有登录才能调用的函数都要加上这句话index=login(index)

5.Django为其提供了快捷的方法,语法糖(装饰器)

def login(func):
   def inner():
       user = input("请输入用户名:")
       pwd = input("请输入密码:")
       if user == "shy" and pwd == "123":
           func()
       else:
           print("用户名或密码错误")
   return inner
@login
def index():
    print("this is index")
@login
def foo():
    print("this is foo")
# index=login(index)  #语法糖相当于这句话
index()

结果:在每个需要登录才能调用的方法上加@login

6.例:在每个函数执行后都打印执行该函数的所需时间

def timer(func):
    import time
    def inner():
        start=time.time()
        func()
        end=time.time()
        cost=end-start
        print(cost)
    return inner
@timer
def add():
    s=0
    for  i in range (30000000):
        s+=i
    print(s)
add()

7.带参数的装饰器

def timer(func):
    import time
    def inner(x,y):
        start=time.time()
        func(x,y)
        end=time.time()
        cost=end-start
        print(cost)
    return inner
@timer
def add(x,y):
    print(x+y)
add(50,54)

8.满足各个函数参数不等的装饰器

def timer(func):
    import time
    def inner(*args,**kwargs):
        start=time.time()
        func(*args,**kwargs)
        end=time.time()
        cost=end-start
        print(cost)
    return inner
@timer
def add(x,y):
    print(x+y)
add(50,54)
@timer
def x(x,y,z):
    print(x*y*z)
x(1,2,3)