获取插入数据的主键id
import pymysql
# 设置数据库连接信息
database = pymysql.connect(
host="127.0.0.1", port=3306, user="root", password="root", database="test"
)
# 开启数据库游标
cursor = database.cursor()
for i in range(5):
# 执行SQL
cursor.execute('insert into test (name) values ("test")')
print(database.insert_id())
# 提交至数据库
database.commit()
# 关闭游标
cursor.close()
# 断开数据库连接 释放资源
database.close()
通过 database.insert_id() 方法可以获取插入数据的主键id, 注意一定要在commit之前获取,否则返回0。