Python SQL查询并生成json文件操作示例

下面我为您详细讲解一下如何在 Python 中进行 SQL 查询并将结果生成为 JSON 文件的操作示例。

环境准备

首先，我们需要安装一些必要的模块，如 pymysql 和 json。可通过运行以下命令进行安装：

pip install pymysql

pip install json

连接数据库

连接数据库通常需要填写数据库地址、用户名、密码和库名等信息，可通过以下代码创建数据库连接：

import pymysql

db = pymysql.connect(host='localhost',
                     user='root',
                     password='password',
                     db='database_name',
                     charset='utf8')
cursor = db.cursor()

这里需要根据实际情况填写数据库连接信息。

执行SQL查询

执行SQL查询，可以通过 execute() 方法向数据库发送 SQL 命令以获取查询结果：

sql = "SELECT * FROM table_name"
cursor.execute(sql)

这里需要替换 sql 为具体的 SQL 语句和 table_name 为具体的表名。

获取查询结果

获取查询结果需要使用 fetchall() 方法，返回所有满足查询条件的记录集。代码示例如下：

result = cursor.fetchall()

获取到查询结果后，可以将结果转为 JSON 格式并写入文件：

将结果写入JSON文件

import json

json_data = json.dumps(result, ensure_ascii=False)
with open('result.json', 'w') as f:
    f.write(json_data)

这里需要将 result 替换为查询结果，将 result.json 替换为具体的文件名和路径。

示例说明

假设我们有一张名为 students 的学生信息表，其中包含学生的姓名、年龄和性别等信息。我们想要查询所有男性学生的姓名和年龄，并将结果以 JSON 格式保存到文件 system/user/result.json 中。具体操作步骤如下：

1. 连接数据库

```python
import pymysql

db = pymysql.connect(host='localhost',
user='root',
password='password',
db='db_students',
charset='utf8')
cursor = db.cursor()
```

1. 执行SQL查询

python
sql = "SELECT name, age FROM students WHERE gender='Male'"
cursor.execute(sql)

1. 获取查询结果

python
result = cursor.fetchall()

1. 将结果写入JSON文件

```python
import json

json_data = json.dumps(result, ensure_ascii=False)
with open('system/user/result.json', 'w') as f:
f.write(json_data)
```

以上就是完整的 Python SQL 查询并生成 JSON 文件的操作示例。