python3--装饰器高级学习版

__author__ = "Aaron Fan"
import time #导入time模块
user,passwd = 'alex','abc123'   #用户名密码
def auth(auth_type):    #创建auth函数，创建auth_type形参
    #print("auth func:",auth_type)   #打印形参auth_type的值
    def outer_wrapper(func):    #创建outer_wrapper函数，创建func形参
        def wrapper(*args, **kwargs):   #创建wrapper函数，创建非固定参数*args,**kwargs
            #print("wrapper func args:", *args, **kwargs)   #打印传进来的这些非固定参数的值
            if auth_type == "local":    #如果auth函数中的形参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    #这里主要是为了显示home()函数中的return值：from home
                    print("---after authenticaion ")    #打印一句话
                    return res  #返回定义的res变量的值

                else:   #否则账号或者密码错误
                    exit("\033[31;1mInvalid username or password\033[0m")
            elif auth_type == "ldap":   #如果auth函数中的形参auth_type的值等于ldap
                print("搞毛线ldap,不会。。。。") #ldap的处理语句，这里暂时先用一句print代替了，记得后面练习是需要去细化一下

        return wrapper  #return wrapper
    return outer_wrapper    #return outer_wrapper

# index主页，不需要用户名密码，可以直接登录，所以这里也无需装饰器
def index():
    print("welcome to index page")

#home页面，需要通过local方式来匹配用户登陆
@auth(auth_type="local") # home = wrapper()
                        #注意@auth代表auth(),而@auth(auth_type="local")代表outer_wrapper()
                        #所以这里其实是指 home = outer_wrapper(home) = wrapper
def home():
    print("welcome to home  page")
    return "from home"

#bbs页面，需要通过ldap方式来匹配用户登陆
@auth(auth_type="ldap")
def bbs():
    print("welcome to bbs  page")

index()
print(home()) #其实执行的是wrapper()  这个方式会显示home的return值
#home()  #其实执行的是wrapper()   这个不会显示return值
bbs()   #其实执行的是wrapper()