Python装饰器3

user,passwd='coke','123456'
#服务器验证，本地验证
def auth(auth_type):    #如果decorator本身需要传入参数，以便解决不同函数需求，用一个装饰器的情况
    def out_wrapper(func):
        def wrapper(*args, **kwargs):   #可以解决被加装饰器的函数有多个参数的情况
            if auth_type=='local':
                username=input('input your username').strip()
                password=input('input your password').strip()
                if user == username and passwd ==password:
                    print('\033[32;1mUser has passed authentication\033[0m')
                    res=func(*args,**kwargs)    #可以解决被加装饰器的函数存在return的情况
                    print('------after authentication------')
                    return res  #考虑到func函数有返回值
                else:
                    exit('\033[31;1mInvalid username or password\033[0m')
            elif auth_type=='ldap':
                print('this is in the bbs')
        return wrapper
    return out_wrapper

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

@auth(auth_type="local")        #装饰器即为函数
def home():#-------->旧版本函数==func
    print('welcome to home page')
    return 'from home'

@auth(auth_type="ldap")
def bbs():#-------->旧版本函数==func
    print('welcome to bbs page')
    
home()      #-------->新版本函数==wrapper  也是auth('local')(home)
bbs()       #-------->新版本函数==wrapper