使用Python插入MySQL数据库后获取id

在Python中，我们经常需要将数据插入到MySQL数据库中，并且还需要获取插入数据的id值。本文将介绍如何使用Python插入MySQL数据库后获取id。

安装MySQL驱动

使用Python连接MySQL数据库，需要安装相关的MySQL驱动。常用的MySQL驱动有mysql-connector-python、PyMySQL和MySQLdb等。

本文以mysql-connector-python作为示例，你可以通过以下命令安装它：

pip install mysql-connector-python

连接MySQL数据库

在Python中连接MySQL数据库，需要使用mysql.connector模块。连接MySQL数据库的示例代码如下所示：

import mysql.connector

db = mysql.connector.connect(
  host="localhost",
  user="root",
  password="123456",
  database="test"
)

# 获取数据库连接的游标
cursor = db.cursor()

插入数据并获取id

在Python中插入数据到MySQL数据库中，需要使用execute()方法。但是，execute()方法并不会返回插入数据的id值，因此我们需要借助MySQL的LAST_INSERT_ID()函数获取id值。

在插入数据后，我们使用cursor.lastrowid获取最后插入数据的id值。示例代码如下所示：

# 插入数据到表中
sql = "INSERT INTO user (username, age) VALUES (%s, %s)"
val = ("Tom", 18)
cursor.execute(sql, val)

# 获取最后插入数据的id值
id = cursor.lastrowid
print("插入数据的id为:", id)

关闭数据库连接

在使用完MySQL数据库后，需要关闭数据库连接，释放资源。关闭数据库连接的示例代码如下所示：

# 提交数据并关闭游标和数据库连接
db.commit()
cursor.close()
db.close()

到此，本文介绍的内容已全部介绍完毕。通过上述示例代码，我们可以方便地在Python中插入数据到MySQL数据库中，并获取插入数据的id值。