python - 如何下载一个网页的所有图片并以原名保存？

Question

我编写了一个小的 Python 脚本，使用 selenium 从网站下载图片:

from selenium import webdriver
import urllib.request
class FirefoxTest:
    def firefoxTest(self):
        self.driver=webdriver.Firefox()
        self.driver.get("http://www.sitew.com")
        self.r=self.driver.find_element_by_tag_name('img')
        self.uri=self.r.get_attribute("src")
        self.g=urllib.request.urlopen(self.uri)
        with open("begueradj.png",'b+w') as self.f:
            self.f.write(self.g.read())             
if __name__=='__main__':
    FT=FirefoxTest()
    FT.firefoxTest()

如何修改我的代码以便:

1. 下载网页上的所有图片 ?
   
2. 不要为我下载的图像命名，而是保留它们的默认名称？

Answer 1

你需要切换到find_elements_by_tag_name .对于下载文件，我会使用 urllib.urlretrieve() - 它会为您从 url 中提取文件名:

images = self.driver.find_elements_by_tag_name('img')
for image in images:
    src = image.get_attribute("src")
    if src:
        urllib.urlretrieve(src)