pymysql 封装插入语句

import pymysql

class MySQLHelper:
    def __init__(self, host, user, password, database):
        self.host = host
        self.user = user
        self.password = password
        self.database = database
        self.connection = None

    def connect(self):
        self.connection = pymysql.connect(
            host=self.host,
            user=self.user,
            password=self.password,
            database=self.database
        )

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

    def insert(self, table, data):
        columns = ', '.join(data.keys())
        values = ', '.join(['%s'] * len(data))
        sql = f"INSERT INTO {table} ({columns}) VALUES ({values})"
        with self.connection.cursor() as cursor:
            cursor.execute(sql, list(data.values()))
        self.connection.commit()

# 使用示例
if __name__ == "__main__":
    helper = MySQLHelper(host='localhost', user='root', password='password', database='test')
    helper.connect()
    data = {'name': 'Alice', 'age': 25}
    helper.insert('users', data)
    helper.close()