day2023-3-23-文件操作之登录注册系统作业

"""
Author:Ren Yulian
Create Time: 2023/3/23 17:51
Happy every day!
"""

print('=========================================')
print(' ❀❀欢迎来到xxx管理系统❀❀ ')
print(' ❤ 1. 登              录')
print(' ❤ 2. 注              册')
print(' ❤ 3. 退              出')
print('=========================================')

option = input('请选择(1~3):')

if option == '2':
    user_name = input('请输入账号：')
    pw = input('请输入密码：')

    try:
        f = open('files/xxx管理系统.txt', encoding='utf-8')
        all_users = eval(f.read())
        f.close()
    except FileNotFoundError:
        all_users = []

    users = []
    pws = []
    for x in all_users:
        users.append(x['user'])
        pws.append(x['pw'])

    if user_name not in users and pw not in pws:
        print('注册成功！')
        all_users.append({'user': user_name, 'pw': pw})
        f = open('files/xxx管理系统.txt', 'w', encoding='utf-8')
        f.write(str(all_users))
        f.close()
    elif user_name in users:
        print('注册失败！该账号已经注册过！')


if option == '1':
    user_name = input('请输入账号：')
    pw = input('请输入密码：')

    f = open('files/xxx管理系统.txt', encoding='utf-8')
    all_users = eval(f.read())
    users = []
    pws = []
    for x in all_users:
        users.append(x['user'])
        pws.append(x['pw'])
    f.close()

    if {'user':user_name, 'pw':pw} in all_users:
        print('登录成功！')

    if {'user': user_name, 'pw': pw} not in all_users:
        if user_name in users and pw not in pws:
            print('登陆失败！密码错误！')
        elif user_name not in users:
            print('登录失败！该账号不存在！')


if option == '3':
    print('退出登录！')


"""
1.持久化数据：所有注册过的账号
2.设计文件：files/userInfo.txt
方案1：{用户名1:密码1, 用户名2:密码2, ...}
方案2：[{'username':用户名1, 'pw':密码1}, {'username':用户名2, 'pw':密码2}, ...]
"""


def read_file(file_path):
    try:
        with open(file_path, encoding='utf-8') as f:
            return f.read()
    except FileNotFoundError:
        return None


def register():
    # 1.提示用户输入账号和密码
    username = input('请输入账号：')
    password = input('请输入密码：')

    # 2.判断账号是否已经注册过
    # 1)获取所有已经注册过的账号
    result = read_file('files/userInfo.txt')
    if result:
        all_user = eval(result)
    else:
        all_user = []
    # 2)判断输入账号是否在已经注册过的
    for user in all_user:
        if user['username'] == username:
            print('注册失败！该账号已经注册过！')
            return            # 直接结束这个注册操作

    # 3.完成注册
    all_user.append({'username':username, 'pw':password})
    with open('files/userInfo.txt', 'w', encoding='utf-8') as f:
        f.write(str(all_user))
    print('注册成功！')


def login():
    # 1.输入账号和密码
    username = input('请输入账号：')
    password = input('请输入密码：')

    # 2.判断是否登录成功
    # 1）获取已经注册过的用户信息
    result = read_file('files/userInfo.txt')
    if not result:
        print('登录失败！该账号不存在！')
        return
    # 2）判断用户名是否已经注册并且输入的密码和保存密码是否一致
    all_user = eval(result)
    for user in all_user:
        if user['username'] == username:
            if user['pw'] == password:
                print('登录成功！')
                return True
            else:
                print('登录失败！密码错误！')
            return False

    print('登录失败！该账号不存在！')
    return False


# 方法2
def main_page():
    page = read_file('files/mainPage.txt')
    while True:
        # 1.打印主页内容
        print(page)

        # 2.给出选择
        value = input('请选择(1~3):')

        # 3.根据不同的选择程序做出不同的反应
        if value == '1':
            result = login()
            if result:
                break
        elif value == '2':
            register()
        else:
            exit()   # 退出程序

    print(read_file('files/succeed.txt'))


# 调用函数进行登录注册操作
main_page()



if __name__ == '__main__':
    main_page()