Python--使用Python从MySQL数据库导出数据到JSON文件

1. 概述

这段代码演示了如何使用Python从MySQL数据库查询数据，并将查询结果以JSON格式导出到文件。

2. 代码分析

• 首先，代码从data_define_mysql模块导入Record类，该类用于封装从数据库查询到的每条记录。
• 接着，使用pymysql库建立与MySQL数据库的连接，并创建游标对象。
• 选择数据库py_sql，执行SQL查询语句SELECT * FROM orders，获取所有订单数据。
• 对查询结果进行遍历，将每条记录转换为Record对象，然后调用to_json方法转换为JSON格式的字符串。
• 将JSON格式的字符串写入到文件d:/output.json中，每条记录后面添加换行符。
• 最后，关闭数据库连接和文件。

3. 关键点

• 数据库连接：使用pymysql.Connection创建数据库连接对象。
• 游标操作：通过游标执行SQL语句并获取结果。
• 数据转换：将查询结果封装为Record对象，并转换为JSON格式。
• 文件操作：使用文件的写入模式w，指定编码为UTF-8，确保中文能正确写入文件。

4. 扩展应用

• 异常处理：在实际应用中，应对数据库连接和操作进行异常处理，确保程序健壮性。
• 数据封装：Record类可以根据实际需求进行扩展，以支持更复杂的数据结构。
• 批量插入：如果需要将大量数据写入文件，可以考虑使用writelines方法批量写入，提高效率。

5. 代码示例扩展

以下是一些代码的扩展示例：

from data_define_mysql import Record
from pymysql import Connection, Error

try:
    # 构建MySQL连接
    conn = Connection(
        host="localhost",
        port=3306,
        user="root",
        password="123456",
        autocommit=True
    )
    # 获得游标对象
    cursor = conn.cursor()
    # 选择数据库
    conn.select_db("py_sql")
    # 执行查询
    cursor.execute("SELECT * FROM orders")
    result = cursor.fetchall()

    # 打开文件准备写入
    with open("d:/output.json", "w", encoding="UTF-8") as f:
        for r in result:
            record = Record(r[0], r[1], r[2], r[3])
            f.write(record.to_json() + "\n")

except Error as e:
    print("数据库查询失败:", e)
finally:
    # 确保游标和连接被正确关闭
    if conn:
        conn.close()

6. 编程规范

• 异常处理：使用try...except结构来捕获和处理可能发生的异常。
• 资源管理：使用with语句确保文件在操作完成后正确关闭。
• 代码可读性：保持代码的清晰和组织性，使用适当的变量命名和注释。

通过这些示例和扩展，可以更深入地理解如何将数据库中的数据导出为JSON格式，并学习到一些编写健壮、可读代码的技巧。