pymysql 入门

原创已于 2024-12-17 14:00:06 修改 · 1k 阅读

15 ·

CC 4.0 BY-SA版权

文章标签：

#mysql #python

于 2024-12-17 13:56:47 首次发布

mysql 专栏收录该内容

4 篇文章

订阅专栏

发现宝藏

前些天发现了一个巨牛的人工智能学习网站，通俗易懂，风趣幽默，忍不住分享一下给大家。【宝藏入口】。

1. 什么是 PyMySQL？

PyMySQL 是一个纯 Python 编写的 MySQL 客户端库，可以通过它轻松地在 Python 中连接并操作 MySQL 数据库。它兼容 MySQL 5.x 和 MariaDB，并且与 MySQL 官方提供的客户端接口（C API）兼容。PyMySQL 支持数据库连接、执行 SQL 查询、以及提取查询结果等功能。

1.1 为什么选择 PyMySQL？

纯 Python 实现：没有 C 扩展，安装起来更加简单，兼容性更好。
易用性：与 MySQL 数据库的交互非常直观。
支持 Python3：在 Python3 环境中运行非常流畅。

2. 安装 PyMySQL

PyMySQL 可以通过 Python 的包管理工具 pip 进行安装。打开终端或命令提示符，输入以下命令：

pip install pymysql

安装完成后，你就可以在你的 Python 项目中使用 PyMySQL 了。

3. 连接 MySQL 数据库

要使用 PyMySQL 连接到 MySQL 数据库，你需要提供 MySQL 服务的主机地址、用户名、密码和数据库名等信息。以下是一个连接的简单示例：

import pymysql

# 连接到数据库
connection = pymysql.connect(
    host='localhost',         # MySQL 服务器地址，localhost 表示本地
    user='root',              # 数据库用户名
    password='password123',   # 数据库密码
    database='test_db',       # 要连接的数据库名称
    charset='utf8mb4'         # 字符集（推荐使用 utf8mb4）
)

# 确保连接成功
print("Connected to MySQL Database!")

3.1连接参数详解：

host：MySQL 服务器的地址，通常是 'localhost' 或 IP 地址。如果是远程服务器，则需要填写远程服务器的 IP。
user：连接 MySQL 数据库的用户名。
password：连接数据库的密码。
database：指定要操作的数据库。
charset：字符集，utf8mb4 是推荐的字符集，支持更多的 Unicode 字符。

4. 执行查询操作

4.1 SELECT 查询

查询数据库通常使用 SELECT 语句。下面是如何在 Python 中执行查询操作并处理返回结果的示例：

# 创建一个游标对象
cursor = connection.cursor()

# 执行 SELECT 查询
cursor.execute("SELECT * FROM users")

# 获取所有结果
results = cursor.fetchall()

# 遍历并打印每一行数据
for row in results:
    print(row)

# 关闭游标
cursor.close()

cursor.execute()：执行 SQL 语句。
cursor.fetchall()：获取查询结果的所有记录，返回一个列表，每个元素是数据库表中的一行。

如果只想获取单条记录，可以使用 fetchone() 方法：

cursor.execute("SELECT * FROM users LIMIT 1")
single_row = cursor.fetchone()
print(single_row)

4.2 INSERT 操作

插入数据时，我们使用 INSERT INTO 语句。例如：

cursor = connection.cursor()

# 插入一条数据
sql = "INSERT INTO users (name, age, email) VALUES (%s, %s, %s)"
values = ('Alice', 28, 'alice@example.com')

cursor.execute(sql, values)

# 提交事务
connection.commit()

print("Record inserted successfully!")

# 关闭游标
cursor.close()

cursor.execute(sql, values)：执行插入操作，其中 %s 是占位符，values 是一个元组或列表，包含实际的插入值。
connection.commit()：提交事务，确保数据写入数据库。

4.3 UPDATE 操作

更新数据时，我们使用 UPDATE 语句。例如：

cursor = connection.cursor()

# 更新一条数据
sql = "UPDATE users SET age = %s WHERE name = %s"
values = (30, 'Alice')

cursor.execute(sql, values)

# 提交事务
connection.commit()

print("Record updated successfully!")

# 关闭游标
cursor.close()

4.4 DELETE 操作

删除数据时，我们使用 DELETE FROM 语句。例如：

cursor = connection.cursor()

# 删除一条数据
sql = "DELETE FROM users WHERE name = %s"
value = ('Alice',)

cursor.execute(sql, value)

# 提交事务
connection.commit()

print("Record deleted successfully!")

# 关闭游标
cursor.close()

5. 处理数据库事务

数据库操作可以使用事务来管理。在 PyMySQL 中，默认情况下，数据库操作是自动提交的，但我们可以通过显式地调用 commit() 来提交事务，或者使用 rollback() 来回滚事务。

try:
    cursor = connection.cursor()

    # 执行一些数据库操作
    cursor.execute("UPDATE users SET age = 30 WHERE name = 'Alice'")

    # 如果没有错误，提交事务
    connection.commit()

except Exception as e:
    # 出现错误时回滚事务
    print(f"Error occurred: {e}")
    connection.rollback()

finally:
    cursor.close()

6. 使用游标(Cursor)和提取结果

游标是用于执行 SQL 语句并获取结果的对象。在 PyMySQL 中，我们通常使用游标来执行查询和数据操作。游标提供了如下几种常用方法：

fetchall()：返回所有查询结果。
fetchone()：返回查询结果中的第一条记录。
fetchmany(size)：返回查询结果中的多条记录，size 是返回的条数。

7. 如何使用连接池管理多个数据库连接

对于高并发的应用程序，管理数据库连接池是非常重要的。PyMySQL 本身不支持连接池，但你可以使用其他第三方库来实现连接池，如 DBUtils 或 SQLAlchemy。

例如，使用 DBUtils 创建连接池：

from dbutils.pooled_db import PooledDB
import pymysql

# 创建连接池
pool = PooledDB(
    creator=pymysql,          # 使用 pymysql 作为数据库接口
    maxconnections=5,         # 连接池允许的最大连接数
    host='localhost',
    user='root',
    password='password123',
    database='test_db',
    charset='utf8mb4'
)

# 从连接池获取连接
connection = pool.connection()

# 执行数据库操作
cursor = connection.cursor()
cursor.execute("SELECT * FROM users")
result = cursor.fetchall()
print(result)

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

8. 异常处理

在数据库操作中，捕获和处理异常非常重要。例如，尝试执行一个 SQL 语句时可能会出现数据库连接失败、SQL 错误等问题。你可以使用 try/except 来捕获这些异常。

try:
    connection = pymysql.connect(
        host='localhost',
        user='root',
        password='wrongpassword',
        database='test_db'
    )
except pymysql.MySQLError as e:
    print(f"Error connecting to MySQL: {e}")