来自 Python 的 post 请求中的 JSON 查询并将结果保存为 csv 文件

Question

我正在尝试编写一个连接到指定 url 的程序：

http://api.scb.se/OV0104/v1/doris/sv/ssd/START/BO/BO0104/BostadsbestandK

并使用此查询：

{ "query": [ { "code": "Region", "selection": { "filter": "vs:RegionRiket99", "values": [ "00" ] } }, { "code": "Hustyp", "selection": { "filter": "item", "values": [ "FLERBO", "SMÅHUS" ] } }, { "code": "Tid", "selection": { "filter": "item", "values": [ "2012" ] } } ], "response": { "format": "csv" } }

然后我想将结果存储到 csv 文件中。

这是我目前的代码：

import urllib.request
import json

url = 'http://api.scb.se/OV0104/v1/doris/sv/ssd/START/BO/BO0104/BostadsbestandK'

data = '{ "query": [ { "code": "Region", "selection": { "filter": "vs:RegionRiket99", "values": [ "00" ] } }, { "code": "Hustyp", "selection": { "filter": "item", "values": [ "FLERBO", "SMÅHUS" ] } }, { "code": "Tid", "selection": { "filter": "item", "values": [ "2012" ] } } ], "response": { "format": "csv" } }'
data = json.dumps(data)

Answer 1

您应该不将查询定义为字符串；将其保留为 Python 对象：

data = { "query": [ { "code": "Region", "selection": { "filter": "vs:RegionRiket99", "values": [ "00" ] } }, { "code": "Hustyp", "selection": { "filter": "item", "values": [ "FLERBO", "SMÅHUS" ] } }, { "code": "Tid", "selection": { "filter": "item", "values": [ "2012" ] } } ], "response": { "format": "csv" } }
data = json.dumps(data)

您需要设置一个application/json 请求标头来指示内容是什么。您可以使用urllib.request.Request() 对象来执行此操作；首先将 JSON 数据编码为 UTF-8：

request = urllib.request.Request(url, data.encode('utf8'))
request.add_header('content-type', 'application/json')
response = urllib.request.urlopen(request)

现在您可以将 CSV 保存到文件中；使用shutil.copyfileobj() 会提高效率，即使对于大型响应：

import shutil

with open(csvfilename, 'wb') as outf:
    shutil.copyfileobj(response, outf)