PyMySQL 详解：Python 连接 MySQL 数据库实战

（三）插入数据

执行插入操作时，可以使用 execute() 或 executemany() 方法。

示例：

try:
    with connection.cursor() as cursor:
        # SQL 插入语句
        sql = "INSERT INTO users (name, age, email) VALUES (%s, %s, %s)"
        # 执行单条插入
        cursor.execute(sql, ('John Doe', 30, '[email protected]'))
        # 执行多条插入
        users = [
            ('Alice', 25, '[email protected]'),
            ('Bob', 28, '[email protected]')
        ]
        cursor.executemany(sql, users)
        # 提交更改
        connection.commit()
finally:
    connection.close()

（四）更新和删除数据

类似于插入数据，更新和删除数都是通过 SQL 语句和 execute() 函数来完成。

• 更新数据

示例：

try:
    with connection.cursor() as cursor:
        # SQL 更新语句
        sql = "UPDATE users SET email = %s WHERE name = %s"
        cursor.execute(sql, ('[email protected]', 'John Doe'))
        # 提交更改
        connection.commit()
finally:
    connection.close()

• 删除数据

示例：

try:
    with connection.cursor() as cursor:
        # SQL 删除语句
        sql = "DELETE FROM users WHERE name = %s"
        cursor.execute(sql, ('John Doe',))
        # 提交更改
        connection.commit()
finally:
    connection.close()

（五）事务管理

MySQL 数据库支持事务，PyMySQL 默认启用了自动提交。如果你需要手动控制事务，可以关闭自动提交并在合适的时机提交或回滚。

示例：

try:
    # 关闭自动提交
    connection.autocommit(False)
    with connection.cursor() as cursor:
        # SQL 插入语句
        sql = "INSERT INTO users (name, age, email) VALUES (%s, %s, %s)"
        cursor.execute(sql, ('Chris', 27, '[email protected]'))
        # 人为错误，测试回滚
        raise Exception("人为触发异常")
        # 提交事务
        connection.commit()
except Exception as e:
    print(f"出现错误：{e}")
    # 回滚事务
    connection.rollback()
finally:
    connection.close()

四、游标类型

PyMySQL 提供了多种游标类型，适用于不同的场景：

• 默认游标：返回元组格式的结果。
• 字典游标 (DictCursor)：返回字典格式的结果，字段名作为键。
• SSCursor：流式游标，用于处理大数据量时，避免一次性加载大量数据到内存。

使用不同游标类型可以通过 cursorclass 参数指定。例如：

# 字典游标
connection = pymysql.connect(
    host='localhost',
    user='your_username',
    password='your_password',
    database='your_dbname',
    charset='utf8mb4',
    cursorclass=pymysql.cursors.DictCursor
)

五、安全性

为了防止 SQL 注入攻击，务必使用参数化查询，而不是将参数直接拼接到 SQL 字符串中。

示例：

# 不安全的写法
sql = f"SELECT * FROM users WHERE name = '{name}'"
# 容易导致 SQL 注入

# 安全的写法
sql = "SELECT * FROM users WHERE name = %s"
cursor.execute(sql, (name,))

六、常见错误处理

PyMySQL 中常见的错误处理可以通过捕获异常来完成：

import pymysql

try:
    connection = pymysql.connect(host='localhost', user='root', database='test_db')
    cursor = connection.cursor()
except pymysql.MySQLError as e:
    print(f"数据库连接失败：{e}")

七、性能优化

• 使用批量操作：如插入数据时，使用 executemany() 批量插入，减少数据库的交互次数。
• 连接池：在大规模应用中，可以通过第三方库如 DBUtils 或 SQLAlchemy 提供的连接池来优化数据库连接的复用性。
• 流式查询：对于大规模查询，使用 SSCursor 流式游标，避免一次性加载所有数据到内存。

八、总结

PyMySQL 是一个轻量级、易于使用的 Python 库，适合 Python 程序员与 MySQL 数据库进行交互。通过它，你可以高效地执行 SQL 查询、插入、更新和删除数据操作，同时还可以利用其事务支持、游标控制等高级功能。