Python 第十五章 有参装饰器 +多个装饰器装饰一个函数

有参装饰器

# def auth(argv):
#     def wrapper(func):
#         def inner(*args,**kwargs):
#             func(*args,**kwargs)
#         return inner
#     return wrapper


# login_dic = {
#     "username": None,
#     "flag": False
# }
# def auth(argv): # argv == foo
#     def wrapper(func):
#         def inner(*args,**kwargs):
#             if login_dic["flag"]:
#                 func(*args,**kwargs)
#             else:
#                 if argv == "QQ":
#                     user = input("username:")
#                     pwd = input("password:")
#                     if user == "alex" and pwd == "alex123":  # qq
#                         login_dic["flag"] = True
#                         login_dic["username"] = user
#                         func(*args,**kwargs)
#                     else:
#                         print("用户名或密码错误!")
#                 elif argv == "微信":
#                     user = input("username:")
#                     pwd = input("password:")
#                     if user == "1351101501" and pwd == "alex":  # 微信
#                         login_dic["flag"] = True
#                         login_dic["username"] = user
#                         func(*args, **kwargs)
#                     else:
#                         print("用户名或密码错误!")
#                 elif argv == "抖音":
#                     user = input("username:")
#                     pwd = input("password:")
#                     if user == "alexdsb" and pwd == "alex":  # 抖音
#                         login_dic["flag"] = True
#                         login_dic["username"] = user
#                         func(*args, **kwargs)
#                     else:
#                         print("用户名或密码错误!")
#                 else:
#                     user = input("username:")
#                     pwd = input("password:")
#                     if user == "alexdsb@dsb.com" and pwd == "alex":  # 邮箱
#                         login_dic["flag"] = True
#                         login_dic["username"] = user
#                         func(*args, **kwargs)
#                     else:
#                         print("用户名或密码错误!")
#
#         return inner
#     return wrapper
#
# # 错误的案例
# @auth # foo = auth(foo)
# def foo():
#     print("这是一个被装饰的函数")
#
# foo()


login_dic = {
    "username": None,
    "flag": False
}
# 正确的案例
msg = """
QQ
微信
抖音
邮箱
请输入您要选择登陆的app:
"""
chose = input(msg).upper()

def auth(argv):
    def wrapper(func):
        def inner(*args,**kwargs):
            if login_dic["flag"]:
                func(*args,**kwargs)
            else:
                if argv == "QQ":
                    print("欢迎登陆QQ")
                    user = input("username:")
                    pwd = input("password:")
                    if user == "alex" and pwd == "alex123":  # qq
                        login_dic["flag"] = True
                        login_dic["username"] = user
                        func(*args,**kwargs)
                    else:
                        print("用户名或密码错误!")
                elif argv == "微信":
                    print("欢迎登陆微信")
                    user = input("username:")
                    pwd = input("password:")
                    if user == "1351101501" and pwd == "alex":  # 微信
                        login_dic["flag"] = True
                        login_dic["username"] = user
                        func(*args, **kwargs)
                    else:
                        print("用户名或密码错误!")
                elif argv == "抖音":
                    print("来了,老弟!")
                    user = input("username:")
                    pwd = input("password:")
                    if user == "alexdsb" and pwd == "alex":  # 抖音
                        login_dic["flag"] = True
                        login_dic["username"] = user
                        func(*args, **kwargs)
                    else:
                        print("用户名或密码错误!")
                else:
                    print("欢迎登陆dab邮箱")
                    user = input("username:")
                    pwd = input("password:")
                    if user == "alexdsb@dsb.com" and pwd == "alex":  # 邮箱
                        login_dic["flag"] = True
                        login_dic["username"] = user
                        func(*args, **kwargs)
                    else:
                        print("用户名或密码错误!")

        return inner
    return wrapper
@auth("QQ")
def foo():
    print("这是一个被装饰的函数")

# wrapper = auth(chose)
# foo = wrapper(foo)
foo()



"""
# @auth(chose) 相等于以下两行代码的解构
# wrapper = auth(chose)
# foo = wrapper(foo)
"""

多个装饰器装饰一个函数

def wrapper1(func):
    def inner1(*args,**kwargs):
        print(1)
        func(*args,**kwargs)
        print(11)
    return inner1

def wrapper2(func):  # func == foo
    def inner2(*args,**kwargs):
        func(*args, **kwargs)
        print(22)
    return inner2

def wrapper3(func):
    def inner3(*args,**kwargs):
        print(3)
        func(*args, **kwargs)
        print(33)
    return inner3

# @wrapper1  # 1 11
# @wrapper3  # 3 33
# @wrapper2  #  8 22

# def foo():
#     print(8)

# foo = wrapper2(foo)  # foo == inner2
# foo = wrapper3(foo)  # inner3 = wrapper3(inner2)
# foo = wrapper1(foo)  # inner1 = wrapper1(inner3)
# foo()                # inner1()


# foo = wrapper3(foo)  # foo == inner3
# foo = wrapper2(foo)  # foo = wrapper2(inner3)  foo == inner2
# foo = wrapper1(foo)  # foo = wrapper1(inner2)


# 被装饰的函数正上方有多个装饰器,先执行离被装饰函数最近的装饰器