python的pymysql执行多条sql

原创

mob64ca12d0e5a4 2024-09-26 04:50:30 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12d0e5a4的原创作品，请联系作者获取转载授权，否则将追究法律责任

使用 PyMySQL 执行多条 SQL 语句的指南

在这篇文章中，我们将学习如何使用 Python 的 pymysql 库来执行多条 SQL 语句。对于刚入行的开发者来说，学习如何与数据库交互是非常重要的一步。

流程概述

在开始之前，让我们先了解一下执行多条 SQL 语句的基本流程。以下是一个简单的步骤表：

步骤	描述
1. 安装 PyMySQL	使用 pip 安装 PyMySQL 包。
2. 导入库	在 Python 脚本中导入 PyMySQL 库。
3. 连接数据库	使用 PyMySQL 的连接方法连接到数据库。
4. 创建游标	通过连接对象创建一个游标。
5. 编写 SQL 语句	定义要执行的多条 SQL 语句。
6. 执行 SQL 语句	使用游标执行 SQL 语句。
7. 提交事务	提交任何更改。
8. 关闭连接	关闭游标和数据库连接。

通过这个步骤表，我们可以清楚地看到每一步的目的和需要执行的操作。接下来，我们将详细讨论每一步的实现。

1. 安装 PyMySQL

在终端或命令行中，输入以下命令以安装 pymysql：

pip install pymysql

安装完成后，我们就可以在代码中使用这个库。

2. 导入库

在你的 Python 脚本中，首先需要导入 pymysql 库：

import pymysql  # 导入 pymysql 库

这条代码告诉Python要使用 PyMySQL 这个模块。

3. 连接数据库

使用 pymysql.connect() 方法连接到数据库。你需要提供数据库的主机地址、用户名、密码和数据库名称：

connection = pymysql.connect(  # 创建连接
    host='localhost',           # 数据库服务器地址
    user='your_username',       # 数据库用户名
    password='your_password',   # 数据库密码
    database='your_database'    # 具体使用的数据库名称
)

在这段代码中，我们创建了一个连接对象 connection，以便后续的数据库操作。

4. 创建游标

通过连接对象connection创建一个游标对象，这样我们就可以使用它来执行 SQL 语句：

cursor = connection.cursor()  # 创建游标

游标对象 cursor 是执行 SQL 的基础。

5. 编写 SQL 语句

现在你可以定义要执行的多条 SQL 语句了。比如，我们可以创建一个表并插入数据：

sql_commands = [  # 定义多条 SQL 语句
    """
    CREATE TABLE IF NOT EXISTS users (  # 创建 users 表
        id INT AUTO_INCREMENT PRIMARY KEY,
        name VARCHAR(255) NOT NULL,
        age INT NOT NULL
    )
    """,
    """
    INSERT INTO users (name, age) VALUES ('Alice', 30)  # 插入数据
    """,
    """
    INSERT INTO users (name, age) VALUES ('Bob', 25)    # 插入数据
    """
]

我们定义了一个包含三条 SQL 语句的列表 sql_commands。这三条语句分别用于创建一个表和插入两条记录。

6. 执行 SQL 语句

现在，我们遍历 SQL 语句列表并使用游标对象逐条执行 SQL 语句：

for sql in sql_commands:  # 遍历每条 SQL 语句
    cursor.execute(sql)  # 执行 SQL 语句

在这段代码中，我们循环遍历每条 SQL 语句，并通过 cursor.execute(sql) 方法执行它们。

7. 提交事务

在执行完所有 SQL 语句后，我们需要提交事务，以确保更改生效：

connection.commit()  # 提交事务

commit() 方法将刚刚执行的所有更改保存到数据库。

8. 关闭连接

操作完成后，记得关闭游标和数据库连接：

cursor.close()         # 关闭游标
connection.close()     # 关闭连接

关闭游标和连接非常重要，这样可以释放资源并避免潜在的内存泄露。

完整代码示例

下面是将所有步骤合并在一起的完整代码示例：

import pymysql  # 导入 pymysql 库

# 连接数据库
connection = pymysql.connect(
    host='localhost',           # 数据库服务器地址
    user='your_username',       # 数据库用户名
    password='your_password',   # 数据库密码
    database='your_database'    # 数据库名称
)

# 创建游标
cursor = connection.cursor()  

# 定义 SQL 语句
sql_commands = [
    """
    CREATE TABLE IF NOT EXISTS users (
        id INT AUTO_INCREMENT PRIMARY KEY,
        name VARCHAR(255) NOT NULL,
        age INT NOT NULL
    )
    """,
    """
    INSERT INTO users (name, age) VALUES ('Alice', 30)
    """,
    """
    INSERT INTO users (name, age) VALUES ('Bob', 25)
    """
]

# 执行 SQL 语句
for sql in sql_commands:
    cursor.execute(sql)

# 提交事务
connection.commit()

# 关闭游标和连接
cursor.close()
connection.close()