Python操作MySQL获取插入数据的主键id

276 阅读1分钟
获取插入数据的主键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。