[Selenium] 스크린샷

이전에 작성했던 스크립트에 스크린샷 부분만 추가를 해보았다.

import pyscreenshot

t = driver.switch_to.alert.text
time.sleep(1)
if t == "제목을 입력하세요":
    print("PASS")
else:
    print("FAIL")
    image = pyscreenshot.grab()
    image.save("..\\scrs\\제목_얼럿_오류.png")

추가한 부분은 이 부분으로, 이슈 발생 시 FAIL을 출력하고 스크린샷을 저장하는 형식이다.

 

실행하면 FAIL이 나올 경우 scrs라는 폴더에 이미지 파일이 저장된다.

 

왜 FAIL이 떴냐면?

alert 내 텍스트가 "제목을 입력 하세요."였기 때문..

 

* 웹 한정 캡처가 아닌 PC 화면 전체 캡처(Prt Sc)이기 때문에 메인 모니터에 자동화 실행한 웹을 띄워놔야 한다.

* 주소창, 작업 표시줄까지 모두 캡처됨!

 

파이썬 기반으로는 alert 캡처 정보가 많이 안 나오길래 다른 방법을 찾아보다가 처음으로 'pyscreenshot'을 사용해보았다.

처음에 'import pyscreenshot'만 추가해서 캡처를 뜨려고 하면 실행하자마자 'PIL...' 오류가 떴는데 'Pillow'라는 패키지를 설치해주니 해결됐다.

얼럿이 아닌 웹 자체 캡처는 다른 방법으로도 가능할 듯하다.