使用 PyMySQL 执行多个 SQL 脚本的指南
在实际的开发过程中，常常需要一次性执行多个 SQL 语句，比如新建表格、插入数据等等。如果你刚入行，遇到这样的需求，不用担心，下面将为你详细讲解如何使用 PyMySQL 来实现这一目标。
流程概述
在开始前，我们先了解一下整个过程的步骤，以下是执行多个 SQL 脚本的大致流程：
| 步骤 | 描述 |
|---|---|
| 1 | 安装并导入 PyMySQL 模块 |
| 2 | 创建数据库连接 |
| 3 | 编写 SQL 脚本 |
| 4 | 执行 SQL 脚本 |
| 5 | 关闭数据库连接 |
flowchart TD
A[安装并导入 PyMySQL] --> B[创建数据库连接]
B --> C[编写 SQL 脚本]
C --> D[执行 SQL 脚本]
D --> E[关闭数据库连接]
接下来，我们将详细讲解每一步以及相应的代码示例。
1. 安装并导入 PyMySQL 模块
首先，确保你已经安装了 PyMySQL。可以使用 pip 安装：
pip install pymysql
然后，在你的 Python 程序中导入该模块：
import pymysql # 导入 PyMySQL 模块
2. 创建数据库连接
接下来，你需要创建一个到 MySQL 数据库的连接。这需要提供数据库的主机地址、用户名、密码和数据库名：
# 创建数据库连接
connection = pymysql.connect(
host='localhost', # 数据库主机地址
user='your_username', # 替换为你的用户名
password='your_password', # 替换为你的密码
database='your_database' # 替换为你要连接的数据库名
)
3. 编写 SQL 脚本
你可以将多个 SQL 语句放在一个字符串中，也可以分开定义。以下是一个示例：
# 定义多个 SQL 语句
sql_scripts = [
"CREATE TABLE IF NOT EXISTS users (id INT AUTO_INCREMENT PRIMARY KEY, name VARCHAR(100))",
"INSERT INTO users (name) VALUES ('Alice')",
"INSERT INTO users (name) VALUES ('Bob')"
]
4. 执行 SQL 脚本
使用数据库连接来执行每一个 SQL 语句。我们这里使用一个循环来批量执行：
try:
with connection.cursor() as cursor: # 创建游标对象
for script in sql_scripts: # 遍历 SQL 脚本
cursor.execute(script) # 执行每条 SQL 语句
connection.commit() # 提交变更
except Exception as e:
print(f"执行 SQL 语句时出错: {e}") # 异常处理
5. 关闭数据库连接
完成所有操作后，记得关闭数据库连接，以释放资源：
connection.close() # 关闭数据库连接
完整代码示例
将上述步骤结合在一起，完整的代码如下：
import pymysql # 导入 PyMySQL 模块
# 创建数据库连接
connection = pymysql.connect(
host='localhost', # 数据库主机地址
user='your_username', # 替换为你的用户名
password='your_password', # 替换为你的密码
database='your_database' # 替换为你要连接的数据库名
)
# 定义多个 SQL 语句
sql_scripts = [
"CREATE TABLE IF NOT EXISTS users (id INT AUTO_INCREMENT PRIMARY KEY, name VARCHAR(100))",
"INSERT INTO users (name) VALUES ('Alice')",
"INSERT INTO users (name) VALUES ('Bob')"
]
try:
with connection.cursor() as cursor: # 创建游标对象
for script in sql_scripts: # 遍历 SQL 脚本
cursor.execute(script) # 执行每条 SQL 语句
connection.commit() # 提交变更
except Exception as e:
print(f"执行 SQL 语句时出错: {e}") # 异常处理
finally:
connection.close() # 关闭数据库连接
结尾
通过以上步骤，你已经学习了如何使用 PyMySQL 执行多个 SQL 脚本。这一过程不仅可以有效地管理数据库操作，还能通过异常处理提升你的代码健壮性。希望这篇教程能对你有所帮助，鼓励你继续深入学习更多的数据库操作和编程技能！
