python通过pymysql插入数据,如何判断插入数据成功
时间: 2023-12-08 20:39:45 浏览: 476
使用 `pymysql` 插入数据时,可以通过判断插入语句的返回值来确定插入是否成功。`execute()` 方法执行插入语句后,返回的是受影响的行数。如果返回的行数大于 0,则表示插入成功;如果返回的行数为 0,则表示插入失败。
以下是一个使用 `pymysql` 插入数据并判断插入是否成功的示例代码:
```python
import pymysql
# 连接数据库
connection = pymysql.connect(host='localhost', user='username', password='password', database='database_name')
try:
with connection.cursor() as cursor:
# 插入数据
sql = "INSERT INTO table_name (column1, column2) VALUES (%s, %s)"
values = ('value1', 'value2')
cursor.execute(sql, values)
# 判断是否插入成功
if cursor.rowcount > 0:
print("插入成功")
else:
print("插入失败")
# 提交事务
connection.commit()
finally:
# 关闭数据库连接
connection.close()
```
在上述示例中,首先通过 `pymysql.connect()` 方法连接到数据库。然后,使用 `with connection.cursor() as cursor` 创建游标对象,执行插入操作。通过 `cursor.rowcount` 可以获取受影响的行数,从而判断插入是否成功。
请根据实际情况替换示例中的 `host`、`user`、`password`、`database`、`table_name`、`column1`、`column2`、`value1`、`value2` 等参数,以及相应的 SQL 语句。
最后,记得通过 `connection.commit()` 提交事务,并使用 `connection.close()` 关闭数据库连接。
阅读全文