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

Question

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

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)