selenium.save_screenshot으로 자동화 스크린샷 저장하기

Selenium을 활용한 자동화 스크린샷 저장하기: selenium.save_screenshot

자동화 도구를 사용하여 웹 작업을 수행할 때, 웹 페이지의 상태를 캡처하는 것이 중요합니다. Selenium은 이러한 스크린샷을 손쉽게 저장할 수 있는 기능을 제공합니다. 오늘은 selenium.save_screenshot을 통해 자동화된 스크린샷을 저장하는 방법에 대해 소개하겠습니다.

selenium.save_screenshot 함수 소개

selenium.save_screenshot 함수는 현재 브라우저 화면을 PNG 형식으로 캡처하여 지정된 파일 경로에 저장합니다. 웹 페이지의 특정 상태를 기록하여 나중에 참고할 수 있는 유용한 기법입니다.

함수 시그니처

driver.save_screenshot(filename)

매개변수:

• filename: 저장할 스크린샷 파일의 경로와 이름입니다. 예: 'screenshot.png'

반환 값:

• 스크린샷이 성공적으로 저장되면 True를 반환합니다. 저장 실패 시 False를 반환할 수 있습니다.

사용 예제

기본 예제

다음은 selenium.save_screenshot 함수를 사용하여 자동으로 스크린샷을 저장하는 기본 예제입니다.

from selenium import webdriver
import time

# Chrome 웹 드라이버 초기화
driver = webdriver.Chrome()

# 웹페이지 열기
driver.get('https://www.example.com')

# 웹 페이지 로딩 후 잠시 대기
time.sleep(2)

# 스크린샷 저장
screenshot_path = 'screenshot.png'
driver.save_screenshot(screenshot_path)

print(f'Screenshot saved at: {screenshot_path}')

# 드라이버 종료
driver.quit()

다양한 페이지 스크린샷 예제

다음은 여러 페이지에서 스크린샷을 찍고 각각 다른 파일 이름으로 저장하는 예제입니다.

from selenium import webdriver
import time

# Chrome 웹 드라이버 초기화
driver = webdriver.Chrome()

# 첫 번째 웹페이지 열기
driver.get('https://www.example.com')
time.sleep(2)
driver.save_screenshot('example_screenshot.png')

# 두 번째 웹페이지 열기
driver.get('https://www.google.com')
time.sleep(2)
driver.save_screenshot('google_screenshot.png')

print('Screenshots saved for both pages.')

# 드라이버 종료
driver.quit()

결론

selenium.save_screenshot 함수를 활용하면 웹 페이지의 현재 상태를 쉽게 저장하고 기록할 수 있습니다. 이는 자동화 스크립트를 테스트할 때 유용하며, 문제 발생 시 해당 상태를 분석하는 데 도움을 줄 수 있습니다.

• webdriver를 통해 스크린샷을 기록하고, 성공적인 자동화를 위해 활용해 보세요!
• 지금 바로 selenium.save_screenshot으로 간편하게 웹 페이지의 모습을 저장해 보세요!