python入门之装饰器

'''
#装饰器(
#定义: 本质是函数,功能就是用来装饰其它函数(就是为其他函数添加附加功能))
#原则:1.不能修改被装饰的函数的源代码 2.被修饰的函数的调用方式不能被修改

#实现装饰器知识储备：
#1.函数即“变量” 2.高阶函数 3.嵌套函数 4.高阶函数+嵌套函数=装饰器

#----------函数即“变量”----------------------

#定义一个函数就等于把函数体复制给了一个变量

#-----------高阶函数---------------------------
#满足两个条件之一：1.把一个函数名当作实参传给另外一个函数(在不修改被装饰函数源代码的情况下为其添加功能)
                 2.返回值中包含函数名(不修改函数的调用方式)


import time

def timmer(func):
    def warpper(*args,**kwargs):
        start_time = time.time()
        func()
        stop_time = time.time()
        print('time is %s' %(stop_time-start_time))
    return warpper()
@timmer

def test1():
    time.sleep(3)
    print('in the test1')

test1()
'''
'''------高阶函数1------
import  time
def bar():
    print('in the bar')
def test1(func):

    start_time = time.time()
    func() #run bar
    stop_time = time.time()
    print("the func time is %s" %(stop_time-start_time))

test1(bar)
--------------高阶函数2---------------------
import time
def bar():
    time.sleep(3)
    print('in the bar')

def test2(func):
    print(func)
    return func
print(test2(bar))
bar()

---------------嵌套函数-------------
def foo():
    print('in the bar')
    def bar():
        print('in the bar')
    bar()
foo()


x =0
def grandpa():
    x=1
    def daa():
        x=2
        def son():
            x=3
            print(x)
        son()
    daa()
grandpa()

'''

#装饰器(需求：在不修改代码的情况下给test1增加一个功能)
import time

def timer(func):
    def deco():
        start_time = time.time()
        func()
        stop_time = time.time()
        print("the func run time is %s" %(stop_time-start_time))
    return deco

@timer #就等于 test1=timer(test1) 哪个函数使用新功能就写在哪个函数上面
def test1():
    time.sleep(3)
    print('in the test1')

@timer
def test2():
    time.sleep(3)
    print('in the test2')

test1()
test2()

---------装饰器案例----------------

import time
user,passwd = 'sun','123'
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":
                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)  # from home
                    print("---after authenticaion ")
                    return res
                else:
                    exit("\033[31;1mInvalid username or password\033[0m")
            elif auth_type == "ldap":
                print("搞毛线ldap,不会。。。。")

        return wrapper
    return outer_wrapper

def index():
    print("welcome to index page")
@auth(auth_type="local") # home = wrapper()
def home():
    print("welcome to home  page")
    return "from home"

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

index()
print(home()) #wrapper()
bbs()