저장 시 일부 이미지가 안 보이는 문제에 대하여

강의 상에서는 눈에 띄지 않았으나, 일괄적으로 png 포맷으로 저장하면 원본 파일의 포맷이 다른 경우 이미지가 보이지 않더라고요...

해결책을 찾은 것 같지만 이와 별개로 강의는 약간 수정이 되어야 할 것 같아서 써 봅니다. (이미지 저장 부분)

images = driver.find_elements_by_css_selector('._image._listImage')
index = 1
for image in images:
    image_source = image.get_attribute('src')
    if '.PNG' in image_source:
        urlretrieve(image_source, f'{search_word}/{index}.png')
        index += 1
    elif '.jpg' in image_source:
        urlretrieve(image_source, f'{search_word}/{index}.jpg')
        index += 1
    if index > 500:
        break

elif를 더 쓰면 gif도 포함할 수 있습니다.

* enumerate를 쓰면 저장 안 하고 넘기는 파일에도 인덱스가 증가해서 중간에 숫자가 비어버리기 때문에 안 썼습니다.