python学习之文件综合操作作业

1、编写用户登录接口

1）输入账号密码完成验证，验证通过后输出"登录成功"
2）可以登录不同的用户
3）同一账号输错三次锁定，（提示：锁定的用户存入文件中，这样才能保证程序关闭后，该用户仍然被锁定）

user_info = {
    'egon':{'password':'123','counter':0},
    'tank':{'password':'321','counter':0},
    'alex':{'password':'888','counter':0},
    'vincent':{'password':'777','counter':0}
}
flag = True
while flag:
    inp_name = input('请输入你的账号:').strip()
    inp_pwd = input('请输入你的密码:').strip()
    file_name = 'locked_users.txt'
    with open(r'{}'.format(file_name),mode='rt',encoding='utf-8') as f:
        locked_names = f.readlines()
    for locked_name in locked_names:
        locked_name = locked_name.strip()
        if inp_name == locked_name:
            print('账号{}已锁定。'.format(inp_name))
            flag = False
            break
    else:
        if inp_name in user_info:
            if inp_pwd == user_info.get(inp_name).get('password'):
                print('登录成功')
                break
            else:
                print('密码错误。')
                user_info.get(inp_name)['counter'] += 1
            if user_info.get(inp_name).get('counter') == 3:
                print('账号{}3次密码输入错误，账号已锁定。'.format(inp_name))
                file_name = 'locked_users.txt'
                with open(r'{}'.format(file_name),mode='at',encoding='utf-8') as f:
                    f.write('{}\n'.format(inp_name))
                break
        else:
            print('账号不存在，请重新输入。')

2、编写程序实现用户注册后，可以登录。

while True:
    msg = """
    0 退出
    1 登录
    2 注册
    """
    print(msg)
    cmd = input('请输入命令编号>>: ').strip()
    if not cmd.isdigit():
        print('必须输入命令编号的数字，傻叉')
        continue
    if cmd == '0':
        break
    elif cmd == '1':
        # 登录功能代码（附加：可以把之前的循环嵌套，三次输错退出引入过来）
        count = 0
        flag = True
        while flag:
            inp_name = input('请输入你的账号:').strip()
            inp_pwd = input('请输入你的密码:').strip()
            with open(r'locked_name.txt',mode='rt',encoding='utf-8') as f:
                locked_names = f.readlines()
            for locked_name in locked_names:
                locked_name = locked_name.strip()
                if inp_name == locked_name:
                    print('账号 {} 已锁定。'.format(inp_name))
                    flag = False
                    break
            else:
                with open(r'db.txt',mode='rt',encoding='utf-8') as f:
                    users = f.readlines()
                user_info = {}
                for user in users:
                    user_name,password = user.strip().split(':')
                    user_info[user_name] = password
                if inp_name in user_info:
                    if inp_pwd == user_info.get(inp_name):
                        print('登录成功。')
                        break
                    else:
                        print('密码错误。')
                        count += 1
                    if count ==3:
                        print('账号 {} 3次输入密码错误，账号已锁定。'.format(inp_name))
                        with open(r'locked_name.txt',mode='at',encoding='utf-8') as f:
                            f.write('{}\n'.format(inp_name))
                        break
                else:
                    print('你输入的账号不存在，请重新输入。')
    elif cmd == '2':
        # 注册功能代码
        inp_name = input('请输入你要注册的账号:').strip()
        inp_pwd = input('请为你的账号设置密码:').strip()
        file_name = 'db.txt'
        with open(r'{}'.format(file_name),mode='rt',encoding='utf-8') as f:
            users = f.readlines()
        user_names = []
        for user in users:
            user_name,*_ = user.strip().split(':')
            user_names.append(user_name)
        if inp_name in user_names:
            print('账号已存在，注册失败。')
        else:
            with open(r'{}'.format(file_name),'at',encoding='utf-8') as f:
                f.write('{}:{}\n'.format(inp_name,inp_pwd))
            print('注册成功。')
    else:
        print('输入的命令不存在')