day11 登录注册

# author_lorry   
# date: 2018/11/19

mean_page ="""
=====欢迎进入学生管理系统====
    "❤1",  登录
    "❤2",  注册
    "❤3",  退出
============================
"""
def log_passwod_infor():
    qqq = []
    qqqqq = []
    qq_key = []
    qq_value = []
    with open("user_info.txt", encoding="utf-8") as m:
        b = m.read()
        aa = b.split("\n")
        b = len(aa)
    index = 0
    while index <= b - 2:
        qq = []
        aaa = aa[index].split(":")
        qq.append(aaa[0])
        qq.append(aaa[1])
        index += 1
        qqq.append(qq)
    for item in qqq:
        qqqq = {}
        qqqq.update([item])
        qqqqq.append(qqqq)
    for item in qqqqq:
        for key in item:
            qq_key.append(key)
    for item in qqqqq:
        for key in item:
            qq_value.append(item[key])
    return qq_key,qq_value


def registered_infor(num:int):
    if num == 1:
        while True:
            your_name = input("请输入用户名：")
            user_zhanghao, user_passwo = log_passwod_infor()
            if your_name not in user_zhanghao:
                print("账号还未注册，请先注册！")
                break
            else:
                your_password = input("请输入密码：")
                user_name_index = user_zhanghao.index(your_name)
                user_password = user_passwo[user_name_index]
                if your_password ==  user_password:
                    print("登陆成功！")
                else:
                    print("invalid password or username!please once again")
    else:
        flag = True
        while flag:
            user_name = input("请自定义账号：")
            with open("user_info.txt",'r',encoding="utf-8") as m:
                content = m.readline()
                if len(content) == 0:
                    password = input("请输入密码：")
                    repassword = input("请再次输入密码：")
                    if password == repassword:
                        with open("user_info.txt", "a", encoding="utf-8") as m:
                            m.write("%s:%s\n" % (user_name, password))
                            print("注册成功！")
                            break
                    else:
                        print("账号密码不一致，请重新输入！")
                else:
                    user_zhanghao,user_passwo = log_passwod_infor()  #   qqqqq
                    if user_name not in user_zhanghao:
                        print("还未注册！可以注册！")
                        password = input("请输入密码：")
                        repassword = input("请再次输入密码：")
                        if password == repassword:
                            with open("user_info.txt", "a", encoding="utf-8") as m:
                                m.write("%s:%s\n" % (user_name, password))
                                print("注册成功！")
                                flag = False
                        else:
                            print("账号密码不一致，请重新输入！")
                    else:
                        print("您已注册，请登录！")
                        break


while True:
    print(mean_page)
    caozuo = input("请输入编号进行管理：")
    if caozuo.isdigit():
        registered_infor(int(caozuo))
    else:
        print("无效输入！请重新输入！")