Python练习—验证用户及密码是否正确

代码：

dic = {"root": "123456", "admin": "123", "wyq": "firewall"}
count = 3
# print(dic["root"]) 

def login():
    global count
    while count > 0:
        answer1 = input("请输入你的用户名:")
        answer2 = input("请输入你的密码：")
        if answer1 in dic:  
            if dic[answer1] == answer2:  
                print("您输入的密码正确，登录成功")
                return  
            else:
                count = count - 1  
                if count > 0:  
                    print(f"您输入的密码错误，还有{count}次机会,请重新输入...")
                else:
                    print("密码错误，结束程序")
                    return
        else:  
            count = count - 1
            if count > 0:
                print(f"您输入的用户名：{answer1} 未注册，还有{count}次机会。请确认用户名密码是否存在")
    print("用户名不存在，程序退出") 


if __name__ == '__main__':
    login()

返回：

当用户密码都正确时，直接退出程序

当用户名不存在时，提示未注册，允许三次试错

当用户存在时，输入对应的密码错误后，提示输入密码错误，允许三次试错