python_63_装饰器6

#decorator意思:1.装饰器  2.语法糖
import time
user,passwd='qi','123'
def auth(func):
    def wrappper(*args, **kwargs):
        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 authentiction')
            return res
            #func(*args, **kwargs)#如果将以上三步换成本步,则打印不出from home,可以换成 return func(*args, **kwargs)
        else:
            exit('\033[31;1mInvalid username or password\033[0m')
    return wrappper

def index():#首页(不需要登录)
    print('Welcome to index page!')
@auth
def home(auth_type='local'):#主页(需要登录)
    print('Welcome to home page!')
    return 'from home'
@auth
def bbs(auth_type='ldap'):#论坛页(不需要登录 )
    print('Welcome to bbs page!')

index()
print(home())#调用home相当于调用wrapper()
bbs()

 

posted @ 2018-01-29 23:05  耐烦不急  阅读(116)  评论(0编辑  收藏  举报