用python写一个用户注册登录的代码

在Python中编写简单的用户注册和登录功能，通常会涉及到数据存储（如文件、数据库等），以及验证过程。这里我们可以使用内置的数据结构（例如字典）来模拟一个简单的内存数据库。以下是一个基本的示例，使用命令行交互：

class UserDatabase:
    def __init__(self):
        self.users = {}

    # 用户注册
    def register(self, username, password):
        if username not in self.users:
            self.users[username] = password
            print(f"User {username} registered successfully.")
        else:
            print("Username already exists.")

    # 用户登录
    def login(self, username, password):
        if username in self.users and self.users[username] == password:
            print("Login successful.")
        else:
            print("Invalid credentials or user does not exist.")


# 创建用户数据库实例
db = UserDatabase()

# 用户注册
db.register("user1", "password1")

# 用户登录
db.login("user1", "password1")
db.login("user1", "wrong_password")  # 验证错误密码