Python 编写装饰器，为多个函数加上认证的功能（用户的账号密码来源于文件）

Python 编写装饰器，为多个函数加上认证的功能

实例题目

• 编写装饰器，为多个函数加上认证的功能（用户的账号密码来源于文件），要求登录成功一次，后续的函数都无需再输入用户名和密码

用户文件

用户的账户名和密码存放在 user.txt 这个文件里
格式如下，用逗号隔开账号和密码

这里只放了一个用户，你当然可以放多个, 但要注意每一行只能放一个

定义用户状态

首先使用一个字典，存放用户状态

    account={                   # 存放用户状态信息，认证成功后进行修改
        "user":None,
        "init":False,
    }

读取账号密码的文件

这里的usename，password 是下面登陆模块生成的，利用这两个和文件内容验证

    def read_file(username,password):
        with open("user.txt","r",encoding="utf-8") as f_r:
            for i in f_r:
                i = i.strip().split(",")
                # 将内容先去除前后空格和\n(换行符) 再利用 ，号进行分割 返回一个列表
                if username == i[0] and password == i [1]:
                    return True                                  # 如果账户密码正确变返回 True
                else:
                    return False                                 # 如果账户密码正确变返回 False

验证账号密码

将之前写的读取文件模块加到下面，username 和 password 在这里起到作用

    def login(funx):
        def inner():
            if account["init"] == True:
                print("欢迎回来\n")
                funx()
            elif account["init"] == False:
                while True:                            # 加一个循环，输错了能重输 (精髓)
                    username = input("用户名：")
                    password = input("密码：")
                    if read_file(username,password):   # 如果返回的是True值则执行，False反之
                        print("登陆成功，欢迎\n")
                        account["user"] = username     # 修改用户状态信息
                        account["init"] = True
                        funx()
                        break
                    else:
                        print("账户名或密码错误,请重新输入")
        return inner

要验证功能的函数

终于写完验证模块啦，现在只需要把其加在要验证的函数上就行了
这里我随便写了几个

    def one():
        print(f"{'儿童频道':-^24}")
    @login                              #使用@添加装饰器，原理等于 two= login(two)
    def two():
        print(f"{'游戏频道':-^24}")
    @login
    def three():
        print(f"{'动漫频道':-^24}")

    one()
    two()
    three()

    print(account)      #打印出用户状态，方便查看