셀레니움 - 이미지(img) 다운로드 (No urllib.urlretrieve)

URL 요청 방식인 urllib.urlretrieve은 src 주소로 요청을 하는 방식이기 때문에 캡챠(자동등록방지 문자)처럼 매번 요청할 때마다 이미지가 바뀌는 경우가 있어 무용지물이 된다. 그러므로 이미 브라우저에 로드가 된 이미지를 가져오는 방법을 사용해야 한다. 

 

방법 1. screenshot_as_png 사용 (추천)

from selenium import webdriver

driver = webdriver.Firefox()
driver.get('https://www.webpagetest.org/')

with open('filename.png', 'wb') as file:
	file.write(driver.find_element_by_xpath('/html/body/div[1]/div[5]/div[2]/table[1]/tbody/tr/td[1]/a/div').screenshot_as_png)

 

 

방법 2. 자바스크립트 사용 (PIL 모듈의 Image 함수로 저장)

from io import BytesIO
from PIL import Image
from base64 import b64decode

driver.get(url)

# Create a canvas, set it's width and height equal to image's
# Write image to canvas, translate to base64
# Remove metadata prefix
b64img = driver.execute_script(r'''
var img = document.getElementsByTagName("img")[0];
var canvas = document.createElement("canvas");
canvas.width = img.width;
canvas.height = img.height;
var ctx = canvas.getContext("2d");
ctx.drawImage(img, 0, 0);
var dataURL = canvas.toDataURL("image/png");
return dataURL.replace(/^data:image\/(png|jpg);base64,/, "");
''')

# Decode from base64, translate to bytes and write to PIL image
img = Image.open(BytesIO(b64decode(b64img)))
img.save("./filename.png")

 

방법 3. 자바스크립트 사용 (파일 입출력으로 저장)

from io import BytesIO
from base64 import b64decode

driver.get(url)

# Create a canvas, set it's width and height equal to image's
# Write image to canvas, translate to base64
# Remove metadata prefix
b64img = driver.execute_script(r'''
var img = document.getElementsByTagName("img")[0];
var canvas = document.createElement("canvas");
canvas.width = img.width;
canvas.height = img.height;
var ctx = canvas.getContext("2d");
ctx.drawImage(img, 0, 0);
var dataURL = canvas.toDataURL("image/png");
return dataURL.replace(/^data:image\/(png|jpg);base64,/, "");
''')

imgdata = base64.b64decode(b64img)
filename = 'filename.png'  # I assume you have a way of picking unique filenames
with open(filename, 'wb') as f:
    f.write(imgdata)

 

 

출처: 

https://stackoverflow.com/questions/17361742/download-image-with-selenium-python

Download image with selenium python

 

https://stackoverflow.com/questions/16214190/how-to-convert-base64-string-to-image

How to convert base64 string to image?