写一个函数,用来添加数据库中的表中的数据
时间: 2024-05-11 21:17:06 浏览: 10
好的,我可以给你一个基本的函数模板,你可以根据你的具体情况进行修改和补充:
```python
import pymysql
def add_data_to_table(data, table_name):
'''
data: 要添加的数据,类型为字典,键为表中的列名,值为要添加的数据
table_name: 要添加数据的表名
'''
# 连接数据库
db = pymysql.connect(host="localhost", user="root", password="123456", database="test")
# 创建游标
cursor = db.cursor()
# 构造 SQL 语句
keys = ','.join(data.keys())
values = ','.join(['%s']*len(data))
sql = 'INSERT INTO {} ({}) VALUES ({})'.format(table_name, keys, values)
# 执行 SQL 语句
try:
cursor.execute(sql, tuple(data.values()))
db.commit()
print('数据添加成功!')
except Exception as e:
db.rollback()
print('数据添加失败,错误信息为:', e)
finally:
# 关闭游标和数据库连接
cursor.close()
db.close()
```
这个函数使用了 pymysql 库来连接 MySQL 数据库,你需要根据你自己的情况进行修改,比如数据库的主机名、用户名、密码、数据库名等。函数的参数 data 是一个字典,键为表中的列名,值为要添加的数据;table_name 是要添加数据的表名。在函数内部,我们使用了 SQL 的 INSERT INTO 语句来添加数据,使用了 try...except...finally 语句块来捕获可能出现的异常,并在最后关闭游标和数据库连接。