使用 PyMySQL 插入数据并获取主键 ID
在使用 Python 操作 MySQL 数据库时，PyMySQL 是一个非常流行的轻量级库。本文将教你如何实现插入数据后获取主键 ID 的过程。我们将用一个简单的示例来演示整个流程，并展示相关的代码与注释。
流程步骤
下面的表格展示了整个过程的步骤：
| 步骤 | 描述 |
|---|---|
| 1 | 安装 PyMySQL 库 |
| 2 | 导入库并创建数据库连接 |
| 3 | 创建游标对象 |
| 4 | 执行 INSERT 语句 |
| 5 | 获取插入记录的 ID |
| 6 | 关闭游标和数据库连接 |
详细步骤及代码示例
1. 安装 PyMySQL 库
首先，我们需要确保你已经安装了 PyMySQL 库。如果尚未安装，可以使用以下命令：
pip install pymysql
2. 导入库并创建数据库连接
接下来，我们需要导入 PyMySQL 库，并创建一个数据库连接：
import pymysql
# 创建数据库连接
connection = pymysql.connect(
host='localhost', # 服务器地址
user='your_username', # 数据库用户名
password='your_password',# 数据库密码
database='your_database',# 选择的数据库
charset='utf8mb4', # 字符编码
cursorclass=pymysql.cursors.DictCursor # 设置游标类型为字典类型
)
在上面的代码中，记得将 host、user、password、database 替换为你的实际连接信息。
3. 创建游标对象
在执行任何 SQL 语句之前，我们需要创建一个游标对象：
with connection.cursor() as cursor:
# 下面可以执行 SQL 语句
4. 执行 INSERT 语句
使用游标对象，我们可以执行 INSERT 语句插入数据：
sql = "INSERT INTO your_table (column1, column2) VALUES (%s, %s)"
cursor.execute(sql, ('value1', 'value2')) # 插入的值
这里的 your_table 和 column1, column2 需要替换为实际的表名和字段。
5. 获取插入记录的 ID
插入数据后，我们可以通过 lastrowid 属性获取刚插入记录的主键 ID：
new_id = cursor.lastrowid # 获取主键 ID
print(f"Inserted record's ID: {new_id}") # 输出插入记录的 ID
6. 提交更改并关闭游标和连接
最后，别忘了提交事务并关闭游标和数据库连接：
connection.commit() # 提交事务
# 关闭数据库连接
cursor.close()
connection.close()
完整的示例代码如下：
import pymysql
# 建立数据库连接
connection = pymysql.connect(
host='localhost',
user='your_username',
password='your_password',
database='your_database',
charset='utf8mb4',
cursorclass=pymysql.cursors.DictCursor
)
with connection.cursor() as cursor:
# 执行插入
sql = "INSERT INTO your_table (column1, column2) VALUES (%s, %s)"
cursor.execute(sql, ('value1', 'value2'))
# 获取并打印插入记录的 ID
new_id = cursor.lastrowid
print(f"Inserted record's ID: {new_id}")
connection.commit() # 提交事务
# 关闭连接
cursor.close()
connection.close()
总结
通过上面的步骤和代码，我们成功地用 PyMySQL 在 MySQL 数据库中插入记录并获取了主键 ID。掌握这个过程不仅能够帮助你熟悉数据的插入操作，也能在以后的开发中增强你的数据库操作能力。
类图
classDiagram
class DBConnection {
+connect()
+cursor()
+commit()
+close()
}
class Cursor {
+execute()
+lastrowid
+close()
}
DBConnection --> Cursor : creates
旅行图
journey
title 从插入数据到获取 ID 的旅行
section 数据库连接
连接到数据库: 5: DBConnection
section 数据插入
执行 INSERT 语句: 3: Cursor
获取主键 ID: 2: Cursor
section 关闭连接
提交事务: 1: DBConnection
关闭游标: 1: DBConnection
希望这篇文章能帮助你更好地理解如何使用 PyMySQL 插入数据并获取主键 ID。祝你在开发中取得更大的进步！
