python作业day1—用户登陆

跟着老男孩培训写的第一次作业

需求：

1、用户可以通过用户名和密码登陆；

2、密码输入错误三次后即锁定用户

3、把用户名、密码、锁定后的用户放在文件里

4、如果没有该用户提示还没有注册

5、使用锁定的用户登陆会提示此用户已经被锁定了

#! /usr/bin/env python
# -*- coding:utf-8 -*-

username = raw_input("username: ")
password = raw_input("password: ")


# 把用户文件放到字典里面，更加容易匹配
def read_config():
    with open(r'E:\coding\oldboy\practice_day1\user_info', 'r') as f:
        user_dict={}
        for line in f.readlines():
            user_info = line.strip().split('=')
            user_dict[user_info[0]] = user_info[1]
            # user_dict['password'] = user_info[3]
    return user_dict


# 检查输入的用户是否在黑名单里，如果在，显示已经在黑名单里面
def chack_locked():
    with open(r'E:\coding\oldboy\practice_day1\lock_info', 'r') as f_lock_info:
        if username in f_lock_info.read():
            print("You are locked")
            exit(0)


# 密码输入错误后重新输入，超过三次则会停止输入密码
def input_again():
    count = 0
    while count < 2:
        print("It is wrong password")
        password = raw_input("password again: ")
        if password in read_config()['password']:
            print('congratulition')
            break
        else:
            count += 1
    else:
        print("Sorry")
    return count


# 如果输入密码错误三次，则自动把用户添加到黑名单里面
def locked(count):
    if count == 2:
        with open('E:\coding\oldboy\practice_day1\lock_info', 'a') as f_lock:
            f_lock.write(username)


if __name__ == '__main__':
    chack_locked()
    user_dict = read_config()
    if username == user_dict['username'] and password == user_dict['password']:
        print("Welcome to login")
    elif username != user_dict['username']:
        print("No user, please sign up")
    elif username == user_dict['username'] and password != user_dict['password']:
        count = input_again()
        locked(count)

仍然存在的问题：

1、用户文件里无法添加多个用户，只能有一个用户登陆，把用户存放字典时，因为字典的键唯一，所以只能保存后面的用户信息