python基础——装饰器

装饰器：本质是函数，（装饰其他函数）就是为其他函数添加附加功能。
装饰器用途：在不修改原函数源代码和调用方式的情况下，为原函数添加新的功能。
原则：1.不能修改被装饰的函数的源代码
2.不能修改被装饰的函数的调用方式
高阶函数+嵌套函数=》装饰器

举例
这种不能传递参数，不是通用型，通俗的说就是在函数外套一个函数，把原函数当做参数传递进去，并在里头调用加上要实现的功能，最后返回内层函数的地址。

import time
#装饰器
def timer(func):#外层函数，func是形参，把原函数的地址传递进来
    def deco():
        start_time=time.time()#实现的新功能
        func()#地址加了括号就是调用，即调用原函数
        stop_time=time.time()#实现的新功能
        print("the func run time is ,(stop_time-start_time))"#实现的新功能
    return deco#返回内层函数的地址
@timer #test1=timer(test1)，必须要有这个，这个其实就是test1=deco
def test1():
    time.sleep(3)
    print("int the test1")
test1()#test1()=deco()，做了一个转换，之所以这样做的原因是deco()不能直接用，它是一个局部函数

通用型，可以传递任意个参数，应用场景：有些函数需要添加相同的功能，但是它们当中有些需要传递参数进去（不同的原函数有不同的功能，但它们添加的新功能相同），有些不需要，这个时候就需要实现能兼容传递参数和不传递参数的装饰器，下面的例子就是这样的一个通用型装饰器。

import time
def timer(func):
    def deco(*args,**kwargs):#注意这里需要传递参数
        start_time=time.time()
        func(*args,**kwargs)#注意这里需要传递参数
        stop_time=time.time()
        print("the total time is:",(stop_time-start_time))
    return deco
@timer# @timer要紧挨着被装饰函数的上方
def test1():
    time.sleep(3)
    print("int the test1")
@timer
def test2(name,age):#这个原函数需要传递参数
    time.sleep(3)
    print("test2:",name,age)
test1()
test2("Jorocco",23)

这种情况是装饰器带有参数，当我们的外层函数有参数的时候，此时需要在外面再加一层

import time
user,passwd="Jorocco","123ab"
def auth(auth_type):
   def outer_wrapper(func):
        def wrapper(*args,**kwargs):
            if auth_type=="local":
                username=input("Username:").strip()
                password=input("Password:").strip()
                if user==username and passwd==password:
                    print("\033[32;1mUser has passed authentication\033[0m")
                    res=func(*args,**kwargs)#1号位置，有返回值，就需要有这个
                    print("---after authenticaion")
                    return res#1号位置，有返回值，就需要有这个，不然就没有1号位置的返回结果
                else:
                    exit("\033[31;1mInvalid usrname or password\033[0m")
            elif auth_type=="ldap":
                print("gameover")
        return wrapper
   return outer_wrapper

def index():
    print("welcome to index")
@auth(auth_type="local")#auth_type="local"有这个参数
def home():
    print("welcome to home")
    return 3# 1号位置，有返回值
@auth(auth_type="ldap")
def bbs():
    print("welcome to bbs")

index()
print(home())
bbs()