Python代码如何连接MySQL数据库 Python代码使用PyMySQL驱动的连接方法

pip install PyMySQL

import pymysql
import pymysql.cursors # 引入游标类型，方便后续使用DictCursor

# 数据库连接参数，这些通常不应该硬编码在代码里，后面会提到更好的管理方式
DB_CONFIG = {
    'host': 'localhost',  # 数据库服务器地址，如果是远程服务器，请填写对应IP或域名
    'user': 'your_mysql_user',  # 你的MySQL用户名
    'password': 'your_mysql_password',  # 你的MySQL密码
    'database': 'your_database_name',  # 要连接的数据库名称
    'port': 3306,  # MySQL默认端口
    'charset': 'utf8mb4',  # 字符集，非常重要，避免中文乱码问题
    'cursorclass': pymysql.cursors.DictCursor # 默认返回元组，DictCursor返回字典，更方便处理
}

conn = None # 初始化连接对象
cursor = None # 初始化游标对象

try:
    # 建立数据库连接
    # 这里的**DB_CONFIG是将字典解包成关键字参数，很Pythonic的写法
    conn = pymysql.connect(**DB_CONFIG)
    print("数据库连接成功！")

    # 创建游标对象
    # 如果没有指定cursorclass，默认是pymysql.cursors.Cursor，返回元组
    cursor = conn.cursor()

    # --- 示例1：创建一张表 (DDL操作) ---
    # 实际项目中，表结构通常在数据库初始化时就确定了，这里只是演示
    create_table_sql = """
    CREATE TABLE IF NOT EXISTS users (
        id INT AUTO_INCREMENT PRIMARY KEY,
        name VARCHAR(100) NOT NULL,
        email VARCHAR(100) UNIQUE,
        age INT,
        created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
    ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;
    """
    cursor.execute(create_table_sql)
    # DDL操作通常是自动提交的，但显式提交也无妨
    conn.commit()
    print("表 'users' 检查或创建成功。")

    # --- 示例2：插入数据 (DML操作) ---
    insert_sql = "INSERT INTO users (name, email, age) VALUES (%s, %s, %s)"
    user_data = ('张三', 'zhangsan@example.com', 30)
    cursor.execute(insert_sql, user_data) # 参数化查询，防止SQL注入
    print(f"插入了 {cursor.rowcount} 条数据。")

    # 批量插入数据
    users_to_insert = [
        ('李四', 'lisi@example.com', 25),
        ('王五', 'wangwu@example.com', 35),
        ('赵六', 'zhaoliu@example.com', 28)
    ]
    cursor.executemany(insert_sql, users_to_insert)
    print(f"批量插入了 {cursor.rowcount} 条数据。")

    # 提交事务，保存更改到数据库
    conn.commit()
    print("数据插入并提交成功。")

    # --- 示例3：查询数据 (DQL操作) ---
    select_sql = "SELECT id, name, email, age, created_at FROM users WHERE age > %s"
    cursor.execute(select_sql, (25,)) # 参数必须是元组，即使只有一个参数也要加逗号

    # 获取所有查询结果
    results = cursor.fetchall()
    print("\n查询结果 (年龄大于25的用户):")
    for row in results:
        print(row) # DictCursor会打印字典，方便通过键访问字段

    # 获取单条查询结果
    select_one_sql = "SELECT name FROM users WHERE id = %s"
    cursor.execute(select_one_sql, (1,))
    one_result = cursor.fetchone()
    if one_result:
        print(f"\nID为1的用户姓名: {one_result['name']}")

except pymysql.Error as e:
    # 捕获PyMySQL特有的异常
    print(f"数据库操作发生错误: {e}")
    if conn:
        conn.rollback() # 发生错误时回滚事务，撤销未提交的更改
        print("事务已回滚。")

except Exception as e:
    # 捕获其他通用异常
    print(f"发生未知错误: {e}")
    if conn:
        conn.rollback()
        print("事务已回滚。")

finally:
    # 确保在任何情况下都关闭游标和连接
    if cursor:
        cursor.close()
        print("游标已关闭。")
    if conn:
        conn.close()
        print("数据库连接已关闭。")

在实际操作中，我个人觉得`charset='utf8mb4'`和`cursorclass=pymysql.cursors.DictCursor`这两个配置特别重要。前者能有效避免各种字符编码问题，尤其是处理中文时，而后者则让查询结果以字典形式返回，比元组更直观，通过字段名就能访问数据，代码可读性会好很多。至于参数化查询，那更是防止SQL注入的基石，任何时候都应该坚持使用。

### PyMySQL相比其他Python数据库驱动（如mysql-connector-python）有何独特之处？

在我看来，选择数据库驱动，就像是选择一款工具，各有各的哲学和适用场景。PyMySQL和`mysql-connector-python`都是Python连接MySQL的常用选择，但它们确实有各自的特点。

PyMySQL最显著的特点，也是我个人偏爱它的原因之一，就是它是**纯Python实现**。这意味着它不依赖任何C语言扩展或者外部编译库。对于开发者而言，这意味着安装和部署过程通常更简单、更少遇到兼容性问题，尤其是在一些对环境有严格要求的部署场景下，这简直是福音。你不需要担心系统上有没有安装特定的开发工具链，`pip install PyMySQL`往往就能一劳永逸。

相比之下，`mysql-connector-python`是Oracle官方提供的MySQL连接器。它的优势在于“官方”二字，通常意味着对MySQL新特性和协议的更新支持会更及时、更全面。但它的缺点也可能在于，有时它会依赖一些C语言库，这在某些环境下可能会导致安装过程略显复杂，或者出现一些编译问题。

另一个值得一提的是，PyMySQL在性能上往往表现不俗，而且它的API设计非常Pythonic，用起来很顺手。特别是它提供的`DictCursor`，能够直接将查询结果封装成字典，这在处理数据时简直不要太方便，直接通过字段名访问数据，代码清晰度提升不止一个档次。`mysql-connector-python`也有类似的功能，但PyMySQL的纯Python特性让它在轻量级应用和快速原型开发中更具吸引力。

总的来说，如果你追求**部署的简便性、纯Python的优雅，以及对Python 3的良好兼容性**，PyMySQL会是一个非常棒的选择。而如果你需要**严格遵循官方标准、追求最新MySQL特性的无缝支持，或者在特定企业级环境中对官方支持有硬性要求**，那么`mysql-connector-python`可能更适合你。在我日常的开发中，PyMySQL往往是我的首选。

### 在实际项目中，如何高效且安全地管理PyMySQL的数据库连接？

在真实的项目开发中，数据库连接的管理可不仅仅是写几行代码连接关闭那么简单。它涉及到安全性和效率两个核心维度，这两者往往需要精心的设计和权衡。

**安全性方面：绝不硬编码凭证**

这是首要原则，也是我看到很多新手容易犯的错误。将数据库的用户名、密码、主机等敏感信息直接写在代码里，简直是把“请来攻击我”写在了脸上。正确的做法应该是：

1.  **环境变量：** 这是最简单也最常用的方法之一。将数据库凭证配置为操作系统的环境变量。在Python代码中，通过`os.getenv()`来获取。这样，你的代码可以保持通用，而部署环境则负责提供正确的凭证。
    ```python
    import os
    # ...
    DB_CONFIG = {
        'host': os.getenv('DB_HOST', 'localhost'), # 提供一个默认值，方便本地开发
        'user': os.getenv('DB_USER'),
        'password': os.getenv('DB_PASSWORD'),
        'database': os.getenv('DB_NAME'),
        'port': int(os.getenv('DB_PORT', 3306)),
        'charset': 'utf8mb4',
        'cursorclass': pymysql.cursors.DictCursor
    }

# 示例：使用DBUtils的PooledDB
from DBUtils.PooledDB import PooledDB
import os

# 同样从环境变量获取配置
DB_CONFIG_POOL = {
    'host': os.getenv('DB_HOST', 'localhost'),
    'user': os.getenv('DB_USER'),
    'password': os.getenv('DB_PASSWORD'),
    'database': os.getenv('DB_NAME'),
    'port': int(os.getenv('DB_PORT', 3306)),
    'charset': 'utf8mb4',
    'cursorclass': pymysql.cursors.DictCursor
}

# 初始化连接池
# mincached: 启动时创建的最小连接数
# maxcached: 池中最多可存储的空闲连接数
# maxconnections: 最大允许的连接数，包括正在使用的和空闲的
# blocking: 当连接池满时，是否阻塞等待，True表示阻塞
POOL = PooledDB(
    creator=pymysql, # 指定连接模块
    mincached=5,
    maxcached=10,
    maxconnections=20,
    blocking=True,
    **DB_CONFIG_POOL
)

def get_db_connection():
    """从连接池获取一个连接"""
    return POOL.connection()

# 示例使用
if __name__ == '__main__':
    try:
        conn_from_pool = get_db_connection()
        cursor_from_pool = conn_from_pool.cursor()

        cursor_from_pool.execute("SELECT VERSION()")
        version = cursor_from_pool.fetchone()
        print(f"从连接池获取的连接成功查询到MySQL版本: {version['VERSION()']}")

        cursor_from_pool.close() # 关闭游标
        conn_from_pool.close() # 将连接归还到池中，而不是真正关闭
        print("连接已归还到连接池。")

    except Exception as e:
        print(f"使用连接池时发生错误: {e}")