使用 pymysql 连接 MySQL 数据库

使用 pymysql 连接 MySQL 数据库

在现代应用程序开发中，与数据库的交互是一个常见的需求。本文将介绍如何使用 Python 的 pymysql 库连接到 MySQL 数据库，并执行基本的查询操作。我们将通过一个简单的例子来说明整个过程。

1. 安装 pymysql 库

在开始之前，需要确保你的开发环境中安装了 pymysql 库。你可以使用以下命令通过 pip 来安装它：

pip install pymysql

2. 连接到 MySQL 数据库

连接到 MySQL 数据库的第一步是建立与数据库服务器的连接。下面是一个简单的例子，展示了如何连接到 MySQL 服务器并选择一个数据库。

代码示例

import pymysql

def connect_to_database():
    # 建立与MySQL服务器的连接
    connection = pymysql.connect(
        host='localhost',  # MySQL服务器地址
        port=3306,         # MySQL服务器端口号，默认为3306
        user='root',       # MySQL用户名
        password='your_password',  # MySQL密码
        db='your_database'         # 要连接的数据库名
    )
    return connection

参数说明

• host：MySQL服务器的地址，通常是 localhost 或者是服务器的 IP 地址。
• port：MySQL服务器的端口号，默认是 3306。
• user：用于连接MySQL的用户名。
• password：用户名对应的密码。
• db：要连接的数据库名。

3. 执行 SQL 查询

连接成功后，就可以使用连接对象执行 SQL 查询。下面的示例展示了如何查询一个表中的所有数据并打印出来。

代码示例

import pymysql

def select_all_data(table_name):
    # 建立与MySQL服务器的连接
    conn = pymysql.connect(
        host='localhost',  # MySQL服务器地址
        port=3306,         # MySQL服务器端口号，默认为3306
        user='root',       # MySQL用户名
        password='your_password',  # MySQL密码
        db='your_database'         # 要连接的数据库名
    )

    try:
        # 创建一个游标对象，用于执行SQL语句
        with conn.cursor() as cursor:
            # 执行SQL查询语句，选择指定表的所有数据
            sql = f"SELECT * FROM `{table_name}`"
            cursor.execute(sql)
            # 获取所有查询结果
            results = cursor.fetchall()
            # 打印查询结果
            for row in results:
                print(row)
    finally:
        # 确保在完成数据库操作后关闭连接
        conn.close()

# 调用函数查询指定表的数据
select_all_data('your_table')

4. 查询结果的处理

在上面的示例中，我们使用 fetchall() 方法获取所有查询结果，并逐行打印。如果数据量较大，可以使用 fetchone() 方法逐行获取结果，或者使用 fetchmany(size) 方法一次获取多行结果。

代码示例

import pymysql

def select_with_fetchone(table_name):
    # 建立与MySQL服务器的连接
    conn = pymysql.connect(
        host='localhost',  # MySQL服务器地址
        port=3306,         # MySQL服务器端口号，默认为3306
        user='root',       # MySQL用户名
        password='your_password',  # MySQL密码
        db='your_database'         # 要连接的数据库名
    )

    try:
        # 创建一个游标对象，用于执行SQL语句
        with conn.cursor() as cursor:
            # 执行SQL查询语句，选择指定表的所有数据
            sql = f"SELECT * FROM `{table_name}`"
            cursor.execute(sql)
            # 获取一行查询结果
            row = cursor.fetchone()
            while row:
                print(row)
                row = cursor.fetchone()
    finally:
        # 确保在完成数据库操作后关闭连接
        conn.close()

# 调用函数查询指定表的数据
select_with_fetchone('your_table')

5. 错误处理

在实际应用中，连接数据库或执行查询时可能会发生错误。我们可以使用 try 和 except 语句来捕获这些错误，并进行相应的处理。

代码示例

import pymysql

def safe_select_all_data(table_name):
    try:
        # 建立与MySQL服务器的连接
        conn = pymysql.connect(
            host='localhost',  # MySQL服务器地址
            port=3306,         # MySQL服务器端口号，默认为3306
            user='root',       # MySQL用户名
            password='your_password',  # MySQL密码
            db='your_database'         # 要连接的数据库名
        )

        try:
            # 创建一个游标对象，用于执行SQL语句
            with conn.cursor() as cursor:
                # 执行SQL查询语句，选择指定表的所有数据
                sql = f"SELECT * FROM `{table_name}`"
                cursor.execute(sql)
                # 获取所有查询结果
                results = cursor.fetchall()
                # 打印查询结果
                for row in results:
                    print(row)
        finally:
            # 确保在完成数据库操作后关闭连接
            conn.close()
    except pymysql.MySQLError as e:
        print(f"Error connecting to MySQL Platform: {e}")

# 调用函数查询指定表的数据
safe_select_all_data('your_table')

总结

通过以上步骤，你已经学会了如何使用 pymysql 库连接到 MySQL 数据库并执行基本的查询操作。无论是简单的查询还是复杂的数据操作，这些基础知识都将为你后续的数据库操作打下坚实的基础。

希望这篇指南能对初学者有所帮助！