Python 操作 MySQL 数据库：使用 PyMySQL 库

在Python的数据库操作中，PyMySQL是一个常用的MySQL数据库驱动。本篇博客将详细介绍如何使用PyMySQL连接、查询、插入和更新MySQL数据库，并附有实际的代码示例。

1. 安装 PyMySQL

首先，确保你已经安装了PyMySQL。如果没有安装，可以使用以下命令：

pip install pymysql

2. 连接到 MySQL 数据库

在使用PyMySQL之前，需要先连接到MySQL数据库。以下是一个连接到本地MySQL服务器的示例：

import pymysql

# 建立连接
connection = pymysql.connect(
    host='localhost',
    user='your_username',
    password='your_password',
    database='your_database',
    port=3306,
    charset='utf8mb4',
    cursorclass=pymysql.cursors.DictCursor
)

# 获取游标
cursor = connection.cursor()

# 执行SQL查询
cursor.execute("SELECT VERSION()")

# 获取查询结果
data = cursor.fetchone()
print("MySQL版本:", data["VERSION()"])

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

3. 查询数据

下面是一个查询数据的简单示例：

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

# 获取查询结果
data = cursor.fetchall()

# 打印查询结果
for row in data:
    print(row)

4. 插入数据

插入数据使用 INSERT INTO 语句。以下是一个插入数据的例子：

# 执行SQL插入
cursor.execute("INSERT INTO your_table (column1, column2) VALUES (%s, %s)", ("value1", "value2"))

# 提交事务
connection.commit()

5. 更新数据

更新数据使用 UPDATE 语句。以下是一个更新数据的示例：

# 执行SQL更新
cursor.execute("UPDATE your_table SET column1=%s WHERE column2=%s", ("new_value", "old_value"))

# 提交事务
connection.commit()

6. 使用事务

在执行插入、更新等涉及到数据一致性的操作时，可以使用事务。以下是一个使用事务的示例：

try:
    # 开始事务
    connection.begin()

    # 执行SQL插入
    cursor.execute("INSERT INTO your_table (column1, column2) VALUES (%s, %s)", ("value1", "value2"))

    # 执行SQL更新
    cursor.execute("UPDATE your_table SET column1=%s WHERE column2=%s", ("new_value", "old_value"))

    # 提交事务
    connection.commit()

except Exception as e:
    # 发生异常时回滚事务
    connection.rollback()
    print("Error:", e)

finally:
    # 关闭连接
    cursor.close()
    connection.close()

7. 使用上下文管理器

为了更好地管理数据库连接，可以使用Python的 with 语句，确保在使用完毕后正确关闭连接：

import pymysql

# 使用上下文管理器
with pymysql.connect(
    host='localhost',
    user='your_username',
    password='your_password',
    database='your_database',
    port=3306,
    charset='utf8mb4',
    cursorclass=pymysql.cursors.DictCursor
) as connection:
    with connection.cursor() as cursor:
        # 在此处执行数据库操作

结语

PyMySQL提供了一个方便而强大的工具，用于在Python中连接和操作MySQL数据库。通过上述示例，你可以了解如何连接数据库、执行查询、插入和更新数据，以及如何使用事务和上下文管理器来更好地管理数据库连接。希望这篇博客对于使用PyMySQL进行MySQL数据库操作的学习和实践有所帮助。