python - 使用 urllib.urlretrieve() 保存图像

Question

我正在尝试访问以下link通过我的脚本并下载出现的图表。

我正在使用已接受的响应 here 来实现它但是当我尝试打开该文件时，出现错误:文件“test.png”无法打开，因为它是空的。

这是我的代码片段:

import urllib
image_element = driver.find_element_by_id('chartImg')
src = image_element.get_attribute("src")
if src:
    urllib.urlretrieve(str(src), "test.png")

接下来我尝试进一步调试并将代码更改为

if src:
    a, b = urllib.urlretrieve(str(src), "test.png")
    print a, b.items() 

这给了我以下输出:

test.png
[('date', 'Sat, 19 Nov 2016 01:19:20 GMT'), ('connection', 'Keep-Alive'), ('content-length', '0'), ('server', 'BigIP')]

有谁知道为什么'content-length'是'0'？我认为这就是下载的文件为空的原因。

Answer 1

我认为这是因为您正在抓取的图像不包含扩展名。例如，如果您运行此代码:

src = "http://i.imgur.com/2C7Csq6.png"
urllib.urlretrieve(src, "test.png")

PNG 文件有效，并且是完全相同的图像。我尝试寻找方法来做到这一点，而无需上传到提供扩展的图像共享服务，但没有找到任何东西。我还尝试将 .png 添加到原始 src 字符串中，但这也不起作用。我的猜测是这是一个特定于网站的问题。希望您能找到解决方法，祝您好运!