python 登陆验证

以下是一个简单的Python登录验证的例子：

# 定义一个用户字典
user1 = {'user1': 'password1', 'user2': 'password2', 'user3': 'password3'}

# 登录函数
def login():
    count = 0
    while True:
        user = input('请输入用户名：')
        password = input('请输入密码：')
        if user not in user1.keys():
            print('用户名不存在，请重新输入！')
        elif user in user1 and password == user1[user]:
            print('登录成功！')
            break
        else:
            count += 1
            if count == 3:
                print('错误次数过多，退出程序！')
                break
            else:
                print('密码错误，请重新输入！')

# 注册函数
def register():
    p = input('请输入用户名：')
    s = input('请输入密码：')
    if p not in user1.keys():
        user1[p] = s
        print('注册成功！')
    else:
        print('用户名已存在，请重新注册！')

# 主函数
def main():
    while True:
        choice = input('请选择功能：1.登录 2.注册 3.退出\n')
        if choice == '1':
            login()
        elif choice == '2':
            register()
        elif choice == '3':
            print('退出程序！')
            break
        else:
            print('输入有误，请重新输入！')

if __name__ == '__main__':
    main()