我正在尝试使用python的requests模块从网络下载并保存图像。
python
requests
这是我使用的(工作)代码:
img = urllib2.urlopen(settings.STATICMAP_URL.format(**data)) with open(path, 'w') as f: f.write(img.read()) 这是使用requests以下代码的新代码(无效): r = requests.get(settings.STATICMAP_URL.format(**data)) if r.status_code == 200: img = r.raw.read() with open(path, 'w') as f: f.write(img)
你能帮助我从响应中使用什么属性requests吗?
你可以使用response.rawfile对象,也可以遍历响应。
response.rawfile
response.raw默认情况下,使用类似文件的对象不会解码压缩的响应(使用GZIP或deflate)。你可以通过将decode_content属性设置为True(requests将其设置False为控制自身解码)来强制为你解压缩。然后,你可以使用shutil.copyfileobj()Python将数据流式传输到文件对象:
response.raw
decode_content
shutil.copyfileobj()Python
import requests import shutil r = requests.get(settings.STATICMAP_URL.format(**data), stream=True) if r.status_code == 200: with open(path, 'wb') as f: r.raw.decode_content = True shutil.copyfileobj(r.raw, f)
要遍历响应,请使用循环;这样迭代可确保在此阶段解压缩数据:
r = requests.get(settings.STATICMAP_URL.format(**data), stream=True) if r.status_code == 200: with open(path, 'wb') as f: for chunk in r: f.write(chunk)
这将以128字节的块读取数据;如果你觉得其他块大小更好,请使用具有自定义块大小的Response.iter_content()方法:
Response.iter_content()
r = requests.get(settings.STATICMAP_URL.format(**data), stream=True) if r.status_code == 200: with open(path, 'wb') as f: for chunk in r.iter_content(1024): f.write(chunk)
请注意,你需要以二进制模式打开目标文件,以确保python不会尝试为你翻译换行符。我们还设置stream=True为requests不先将整个图像下载到内存中。
stream=True