封装一个sql插入工具
时间: 2024-01-25 22:05:15 浏览: 126
好的,我可以为您封装一个Python的SQL插入工具,使用MySQL作为数据库。以下是示例代码:
```python
import mysql.connector
def insert_data(table_name, data):
# 连接MySQL数据库
mydb = mysql.connector.connect(
host="localhost",
user="yourusername",
password="yourpassword",
database="yourdatabase"
)
# 获取游标
mycursor = mydb.cursor()
# 构造SQL语句
sql = "INSERT INTO " + table_name + " ("
for key in data:
sql += key + ","
sql = sql[:-1] + ") VALUES ("
for value in data.values():
sql += "'" + value + "',"
sql = sql[:-1] + ")"
# 执行SQL语句
mycursor.execute(sql)
# 提交事务
mydb.commit()
# 关闭游标和连接
mycursor.close()
mydb.close()
print(mycursor.rowcount, "record inserted.")
```
这个函数接受两个参数,第一个是表名,第二个是要插入的数据。在函数内部,我们先连接到MySQL数据库,然后构造SQL语句,最后执行SQL语句并提交事务。请注意,这个示例代码并没有做任何输入验证和错误处理,您需要根据实际情况进行适当的修改。
阅读全文