零基础入门登录功能判断

770686119

MenuNote = '''
|--- 新建用户：N/n ---|
|--- 登录账号：E/e ---|
|--- 退出程序：Q/q ---|

'''
print(MenuNote)
userInfo = {}
code = input("请输入指令代码")
flag = 0
while True:
    if flag:
        print(MenuNote)
        code = input("请输入指令代码")
    flag = 1
    # 01 新建用户模块
    if code == 'N' or code == 'n':
        userName = input("请设置用户名1：")
        # 判断用户名是否唯一
        while True:
            if userName in userInfo:
                userName = input("该用户名已经被使用，请重新设置用户名：")
            else:
                userInfo[userName] = input("请设置密码：")
                print("注册成功，赶紧试试登录吧")
                break
            
    #02 登录验证
    elif code == 'E' or code == 'e':
        userName = input("请输入用户名：")
        #判断用户名是否存在
        while True:
            if userName in userInfo:
                userPassword = input("请输入密码：")
            #判断密码是否正确
                while userPassword != userInfo[userName]:
                    userPassword = input("密码验证不正确，请重新输入密码：")
    
                print("验证通过，登录成功")
                flag = 1
                break
            else:
                count = 0
                for eachKeay in userInfo.keys():
                    count += 1
                    print("第%d位用户：%s" %(count,eachKeay))
                isNew = input("用户不存在,是否创建？（Y/N）：")
                if isNew == 'Y':
                    userInfo[userName] = input("请设置密码：")
                    print("注册成功，赶紧试试登录吧")
                    break
                else:
                    break
                        
                flag = 0
    elif code == 'Q' or code == 'q':
        print("感谢使用，再见！")
        break
    else :
        code = input("指令不存在，请重新输入")
        flag = 0