用pymysql实现一个添加数据的接口
时间: 2024-03-03 18:49:14 浏览: 23
好的,我可以为您提供一个使用Python的pymysql包来实现添加数据接口的例子代码,您可以根据自己的需求进行修改:
```python
import pymysql
# 配置数据库连接信息
host = 'localhost'
port = 3306
user = 'root'
password = 'password'
db = 'database_name'
# 连接数据库
conn = pymysql.connect(host=host, port=port, user=user, password=password, db=db)
# 获取游标
cursor = conn.cursor()
# 定义插入数据的函数
def insert_data(name, age, gender):
sql = "INSERT INTO user(name, age, gender) VALUES (%s, %s, %s)"
try:
cursor.execute(sql, (name, age, gender))
conn.commit()
print("数据插入成功!")
except Exception as e:
print("数据插入失败:", e)
conn.rollback()
# 调用插入数据的函数
name = '张三'
age = 18
gender = '男'
insert_data(name, age, gender)
# 关闭游标和连接
cursor.close()
conn.close()
```
在该代码中,我们首先定义了数据库的连接信息,然后通过pymysql包连接数据库,并获取游标。接下来定义了一个insert_data函数,用于插入数据。在函数中,我们使用了参数化的SQL语句,避免了SQL注入的风险。最后,我们调用了insert_data函数,并传入了name、age、gender三个参数。最后关闭了游标和连接。
注意,您需要替换host、user、password、db为您自己的数据库信息,并且根据您的实际需求修改插入数据的SQL语句。