파이썬으로 이미지 다운로드 하기

파이썬으로 이미지를 다운받으면 반복적인 이미지 다운로드 작업을 자동화 할 수 있다.

이 코드는 단순히 다운로드 하는 작업만 진행하지만 selenium이나 requests를 잘 섞어서 사용하면 자동으로 특정한 페이지의 이미지들을 다운 받을 수 있도록 만들 수 있다.

## 필요한 모듈을 불러온다.
import requests # to get image from the web
import shutil # to save it locally

## 다운로드할 이미지 URL
image_url = "http://~~~~.jpg"
filename = image_url.split("/")[-1]

# 스트림으로 이미지 얻어옴
r = requests.get(image_url, stream = True)

# 이미지가 잘 받아졌는지 확인
if r.status_code == 200:
    # 디코드 컨텐츠를 True로 세팅, 그렇지 않으면 파일 사이즈가 0
    r.raw.decode_content = True
    
    # 바이너리로 쓰기 권한으로 저장
    with open(filename,'wb') as f:
        shutil.copyfileobj(r.raw, f)
        
    print('Image sucessfully Downloaded: ',filename)
else:
    print('Image Couldn\'t be retreived')