装饰器

使用装饰器：不改变函数本身并新增功能
装饰器：本质也是一个函数(嵌套函数+高阶函数)
    1、返回值：函数对象（内层函数）
    2、在被装饰函数之前@装饰器函数
    3、被装饰函数需要传参：装饰器内部函数也需要传参，写成inner(*args,**kwargs)适用多个被装饰函数
    4、装饰器需要传参：则在装饰器外需再套一层函数
例子：
　　

# 被装饰函数有参数时，在装饰器内层函数也要有对应参数
import time
def timer(func):
    def inner(*args,**kwargs):
        start_time = time.time()
        func(*args,**kwargs)
        stop_time = time.time()
        print('the func run time is %s'%(stop_time-start_time))
    return inner

@timer  # test1=timer(test1) return inner 即test1=inner
def test1(arg):
    time.sleep(1)
    print('in the test1,arg is %s'%arg)
@timer
def test2():
    time.sleep(1)
    print('in the test2')
test1('123') # 执行到这里时 test1('123')=inner('123')
test2()

装饰器本身需要传参时，需要再嵌套一层函数

# 装饰器本身需要传参
user,passwd = 'admin','123456'
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 username == user and password == passwd:
                    # 设置颜色开始 ：33[显示方式;前景色;背景色m
                    print('33[0;32;47;1mUser has passed authentication33[0m')
                    res = func(*args,**kwargs)
                    print('---after authentication')
                    return res # 被装饰函数有返回值
                else:
                    print('33[0;31;47;1mInvalid username or password33[0m')
            elif auth_type == 'ldap':
                print('不需要验证')
        return wrapper
    return outer_wrapper

def index():
    print('welcome to index page')
@auth(auth_type='local')  #home = wrapper :auth() -->return outer_wrapper -->outer_wrapper()-->return 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())
bbs()