mysql注册登录通用类

import mysql.connector

class UserDAO:
    def __init__(self, host, user, password, database):
        self.conn = mysql.connector.connect(
            host=host,
            user=user,
            password=password,
            database=database
        )
        self.cursor = self.conn.cursor()

    def register(self, username, password):
        query = "INSERT INTO users (username, password) VALUES (%s, %s)"
        self.cursor.execute(query, (username, password))
        self.conn.commit()

    def login(self, username, password):
        query = "SELECT * FROM users WHERE username = %s AND password = %s"
        self.cursor.execute(query, (username, password))
        result = self.cursor.fetchone()
        return result is not None

    def close(self):
        self.cursor.close()
        self.conn.close()

# 示例使用
user_dao = UserDAO('localhost', 'root', 'password', 'testdb')
user_dao.register('testuser', 'testpass')
print(user_dao.login('testuser', 'testpass'))  # 输出: True
user_dao.close()