装饰器

装饰器
定义：本质是函数，装饰其他函数，就是为其他函数添加附加功能。
原则：1.不能修改被装饰函数的源代码
      2.不能修改被装饰函数的调用方式
知识储备：
1.函数即“变量”
2.高阶函数：
        a.把一个函数名以实参的形式传给另一个函数（在不修改被修饰函数源代码的情况下为其添加新功能）
        b.返回值中包含函数名（不修改函数夫人调用方式）
3.嵌套函数：在函数中定义一个函数
装饰器实例；

#Task:给test添加一个计时函数
import time
def timer(func):
    def deco():
        start_time = time.time()
        func()
        stop_time = time.time()
        print('the func time is %s '%(stop_time - start_time))
    return deco
@timer  #test = timer(test)这就是所谓语法糖
def test():
    time.sleep(3)
    print('in the test')
test()

Output:

in the test
the func time is 3.0004241466522217 

 带参数的装饰器：

这段代码将验证方式分为两种，一种local，一种ldap,此段只实现local逻辑，ldap以print函数带过。

username, passwd = "sugar", "abc"
def auth(auth_type):
    print("auth_func", auth_type )
    def outer_wrapper(func):
        def wrapper(*args, **kwargs):
            print("wrapper func args:",*args, **kwargs)
            if auth_type == "local":
                user = input("Please input username:")
                password = input("Please input passwd:")
                if user == username and password == passwd:
                    print("\033[32;1msuccessful pass\033[0m")
                    res = func(*args, **kwargs)
                    print("------afterauth----")
                    return res
                else:
                    exit("\033[31;1mInvalid username or password\033[0m")
            elif auth_type == "ldap":
                print(input("名场面：我信你个鬼，你个糟老头子，坏的很。。。"))
        return wrapper
    return outer_wrapper

def index():
    print("welcome to index page")

@auth(auth_type = "local")#先给装饰器传参数，再将home函数名作为参数传给outerwrapper()，
def home():
    print("welcome to home page")
    return "from home"

@auth(auth_type = "ldap")
def care():
    print("welcome to care page")

index()
print(home())
care()