问题描述
我是 Python 和 Selenium 的新手。我的目标是从 Google 图片搜索结果页面下载图片并将其作为文件保存在本地目录中,但我最初无法下载该图片。
我知道还有其他选项(使用请求等通过 url 检索图像),但我想知道是否可以使用图像的“src”属性来实现,例如“data:image /jpeg;base64,/9j/4AAQSkZJRgABAQAAAQABAAD/2wCEAAkGBxM..."
# This creates the folder to store the image in
if not os.path.exists(save_folder):
os.mkdir(save_folder)
driver = webdriver.Chrome(PATH)
# Goes to the given web page
driver.get("https://www.google.com/imghp?hl=en&ogbl")
# "q" is the name of the google search field input
search_bar = driver.find_element_by_name("q")
# Input the search term(s)
search_bar.send_keys("Ben Folds Songs for Silverman Album Cover")
# Returns the results (basically clicks "search")
search_bar.send_keys(Keys.RETURN)
# Wait 10 seconds for the images to load on the page before moving on to the next part of the script
try:
# Returns a list
search_results = webdriverwait(driver,10).until(
EC.presence_of_element_located((By.ID,"islrg"))
)
# print(search_results.text)
# Gets all of the images on the page (it should be a list)
images = search_results.find_elements_by_tag_name("img")
# I just want the first result
image = images[0].get_attribute('src')
### Need help here ###
except:
print("Error")
driver.quit()
# Closes the browser
driver.quit()
我试过了:
urllib.request.urlretrieve(image,"00001.jpg")
和
urllib3.request.urlretrieve(image,f"{save_folder}/captcha.png")
但我总是使用这些方法点击“except”块。在阅读了一篇有前途的帖子后,我也尝试了:
bufferedImage = imageio.read(image)
outputFile = f"{save_folder}/image.png"
imageio.write(bufferedImage,"png",outputFile)
有类似的结果,但我相信后者 example 在帖子中使用了 Java,我可能在将其翻译成 Python 时犯了错误。
我确定这很明显,但我做错了什么?感谢您的帮助。
解决方法
在这种情况下,您处理的 URL 是一个 Data URL,它是以 base64 编码的图像本身的数据。
从 Python 3.4+ 开始,您可以读取这些数据并使用 urllib.request.urlopen
将其解码为字节:
import urllib
data_url = "data:image/jpeg;base64,/9j/4AAQSkZJRgABAQAAAQABAAD/2wCEAAkGBxM..."
with urllib.request.urlopen(data_url) as response:
data = response.read()
with open("some_image.jpg",mode="wb") as f:
f.write(data)
或者,您可以使用 base64
自己解码数据 url 的 base64 编码部分:
import base64
data_url = "data:image/jpeg;base64,/9j/4AAQSkZJRgABAQAAAQABAAD/2wCEAAkGBxM..."
base64_image_data = data_url.split(",")[1]
data = base64.b64decode(base64_image_data)
with open("some_image.jpg",mode="wb") as f:
f.write(data)