python mysql封装类

import pymysql

class MySQLWrapper:
    def __init__(self, host, user, password, database):
        self.connection = pymysql.connect(host=host, user=user, password=password, database=database)
        self.cursor = self.connection.cursor()

    def execute(self, query, params=None):
        if params:
            self.cursor.execute(query, params)
        else:
            self.cursor.execute(query)
        self.connection.commit()

    def fetch_all(self, query, params=None):
        if params:
            self.cursor.execute(query, params)
        else:
            self.cursor.execute(query)
        return self.cursor.fetchall()

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

# 使用示例
if __name__ == "__main__":
    db = MySQLWrapper(host='localhost', user='root', password='password', database='testdb')
    db.execute("CREATE TABLE IF NOT EXISTS users (id INT AUTO_INCREMENT PRIMARY KEY, name VARCHAR(255))")
    db.execute("INSERT INTO users (name) VALUES (%s)", ('Alice',))
    results = db.fetch_all("SELECT * FROM users")
    for row in results:
        print(row)
    db.close()