[파이썬] 셀레니움을 통한 화면 캡처

앞에서 설명한 셀레니움이 제대로 설치 되었다면 아래와 같은 작업을 진행 할 수 있게 된다.

 

from selenium import webdriver

url = "http://www.naver.com/"

#PhantomJS 드라이버 추출
driver = webdriver.PhantomJS('C:/Program Files/phantomjs/phantomjs-2.1.1-windows/bin/phantomjs')

driver.implicitly_wait(3) #드라이버를 초기화 될 때까지 3초간 대기

driver.get(url)
driver.save_screenshot("naver.png")
driver.quit()

다음과 같은 코드에서 driver 설치 경로는 아래의 설치 경로와 동일하다.

 

Phantomjs를 c드라이브에 설치했으므로 그 경로에 맞춰 설정한 다음에

phantomjs까지 넣어줘야 실행 되는것을 확인 할 수 있다.

 

또한, 만약 경로를 복사해서 붙여 넣을 경우에 \으로 표시되므로 / 으로 변경하는 것을 주의해야 한다.

 

 

성공적으로 진행되었을 경우, 쥬피터 홈에서 naver.png가 생성된 것을 확인 할 수 있습니다.

 

 

naver 화면 전체를 정상적으로 캡처한 것을 확인 할 수 있게 됩니다.