注册登录系统

注册登录系统

def print_():
    print("================================")
    print("    ** 欢迎来到学生管理系统 **      ")
    print()
    print()
    print("          1.注册                 ")
    print("          2.登录                 ")
    print("          3.退出                 ")
    print()
    print("================================")


def input_():
    try:
        zh = input("请输入账号（3~6）：")
        mm = input("请输入密码（6~12）：")
        if (not 3 <= len(zh) <= 6) or (not 6 <= len(mm) <= 12):
            raise IndexError
    except IndexError:
        print("请输入正确的账号和密码长度！")
        return -1
    return [zh, mm]


def init_():
    try:
        f1 = open("stuInfo.txt", encoding="utf-8")
        f1.close()
    except FileNotFoundError:
        f1 = open("stuInfo.txt", "w", encoding="utf-8")
        f1.write("{}")
        f1.close()
        

init_()
while 1:
    print_()
    n = int(input("请输入选择（1-3）："))
    # ================注册================
    if n == 1:
        flag = input_()
        if flag != -1:
            with open("stuInfo.txt", encoding="utf-8") as f:
                info = eval(f.read())
            if flag[0] in info:
                print("该账号已被注册！")
                continue
            else:
                info.update({flag[0]: flag[1]})
                with open("stuInfo.txt", "w", encoding="utf-8") as f:
                    f.write(str(info))
                print("注册成功！")
                continue
        else:
            continue
    # ================登录================
    elif n == 2:
        flag = input_()
        if flag != -1:
            with open("stuInfo.txt", encoding="utf-8") as f:
                info = eval(f.read())

            if flag[0] not in info:
                print("此账号还未注册！")
                continue
            else:
                if info[flag[0]] == flag[1]:
                    print("登录成功！")
                else:
                    print("密码错误！")
        continue
    # ================退出================
    elif n == 3:
        print("退出登录！")
        break