如何使用请求下载图像

Question

我试图下载和保存图像从网络使用python的请求模块。

以下是我使用的（工作）代码：

img = urllib2.urlopen(settings.STATICMAP_URL.format(**data))
with open(path, 'w') as f:
    f.write(img.read())

以下是使用请求的新（非工作）代码：

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)

您能否帮助我了解从请求中使用响应中的哪些属性？




Answer 1

这个怎么样，一个快速的解决方案。

import requests

url = "http://craphound.com/images/1006884_2adf8fc7.jpg"
response = requests.get(url)
if response.status_code == 200:
    with open("/Users/apple/Desktop/sample.jpg", 'wb') as f:
        f.write(response.content)

Answer 2

从请求中获取类似文件的对象，并将其复制到文件中。这样也可以避免一次将整件事情读入内存。

import shutil

import requests

url = 'http://example.com/img.png'
response = requests.get(url, stream=True)
with open('img.png', 'wb') as out_file:
    shutil.copyfileobj(response.raw, out_file)
del response

Answer 3

您可以使用response.raw文件对象，也可以迭代响应。

使用响应。默认情况下，raw类文件对象不会解码压缩响应（使用GZIP或deflate）。通过将decode\u content属性设置为True（requests将其设置为False以控制解码本身），您可以强制它为您解压。然后可以使用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字节块读取数据；如果您觉得另一个块大小效果更好，请使用响应。具有自定义块大小的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，这样请求不会首先将整个图像下载到内存中。