python - 使用 Python Mechanize 下载图像

Question

我正在尝试编写一个 Python 脚本来下载图像并将其设置为我的墙纸。不幸的是，Mechanize 文档非常糟糕。我的脚本正确地跟随链接，但我很难将图像实际保存在我的计算机上。根据我的研究，.retrieve() 方法应该可以完成工作，但是如何指定文件应该下载到的路径？这是我的...

def followLink(browser, fixedLink):
    browser.open(fixedLink)

if browser.find_link(url_regex = r'1600x1200'):

    browser.follow_link(url_regex = r'1600x1200')

elif browser.find_link(url_regex = r'1400x1050'):

    browser.follow_link(url_regex = r'1400x1050')

elif browser.find_link(url_regex = r'1280x960'):

    browser.follow_link(url_regex = r'1280x960')

 return

Answer 1

import mechanize, os
from BeautifulSoup import BeautifulSoup

browser = mechanize.Browser()
html = browser.open(url)
soup = BeautifulSoup(html)
image_tags = soup.findAll('img')
for image in image_tags:
    filename = image['src'].lstrip('http://')
    filename = os.path.join(dir, filename.replace('/', '_'))
    data = browser.open(image['src']).read()
    browser.back()
    save = open(filename, 'wb')
    save.write(data)
    save.close()

这可以帮助您从网页下载所有图像。至于解析html你最好使用BeautifulSoup或lxml。而下载只是读取数据然后写入本地文件。您应该将自己的值分配给 dir。这是您的图像存在的地方。