파이썬 Selenium 스크린샷 저장 전체 화면과 요소 캡처 완벽 가이드

🖼️ Selenium 전체 화면 스크린샷 저장하기

Selenium을 활용하면 브라우저에 표시된 전체 페이지 화면을 그대로 이미지로 저장할 수 있습니다.
이 기능은 자동화 테스트 중 에러가 발생했을 때 상태를 기록하거나, 결과 보고서를 작성할 때 매우 유용합니다.
특히 페이지가 정상적으로 로딩되었는지 검증하거나, 특정 UI 요소가 올바르게 보이는지 확인하는 데도 활용할 수 있습니다.

전체 화면 스크린샷을 저장하는 방법은 크게 두 가지가 있습니다.
첫 번째는 driver.save_screenshot(“파일명.png”) 메서드를 사용하는 방식이고,
두 번째는 driver.get_screenshot_as_file(“파일명.png”) 메서드를 활용하는 것입니다.
두 방법 모두 동일한 결과를 반환하며, 지정한 경로에 PNG 이미지로 저장됩니다.

from selenium import webdriver

driver = webdriver.Chrome()
driver.get("https://www.python.org")

# 전체 화면 스크린샷 저장
driver.save_screenshot("screenshot.png")

# 동일한 기능 (다른 방식)
driver.get_screenshot_as_file("screenshot2.png")

driver.quit()

위 코드처럼 간단히 파일명을 지정하면 프로젝트 실행 폴더에 이미지가 저장됩니다.
만약 특정 경로에 저장하고 싶다면, “C:/screenshots/test.png” 처럼 절대 경로를 지정할 수도 있습니다.

🎯 특정 요소만 캡처하는 element.screenshot 활용법

전체 화면 스크린샷이 필요할 때도 있지만, 어떤 경우에는 특정 요소만 잘라서 저장해야 할 때가 있습니다.
예를 들어 버튼, 배너 이미지, 특정 텍스트 영역 등 필요한 부분만 저장하면 불필요한 영역이 포함되지 않아 훨씬 깔끔하게 결과를 관리할 수 있습니다.
Selenium은 element.screenshot(“파일명.png”) 메서드를 제공하여 이런 작업을 간단하게 처리할 수 있습니다.

이 기능은 find_element() 또는 find_element_by_*() 계열 메서드로 특정 요소를 먼저 찾은 다음, 해당 요소 객체에서 직접 호출하면 됩니다.
즉, element.screenshot()은 요소 단위 스크린샷을 손쉽게 저장할 수 있는 방법입니다.

from selenium import webdriver
from selenium.webdriver.common.by import By

driver = webdriver.Chrome()
driver.get("https://www.python.org")

# 특정 요소 찾기 (예: 로고)
logo = driver.find_element(By.XPATH, '//*[@id="touchnav-wrapper"]/header/div/div[1]/a')

# 해당 요소만 캡처
logo.screenshot("logo.png")

driver.quit()

위 예제에서는 Python 공식 홈페이지에서 로고 이미지를 찾아서 별도로 저장했습니다.
이렇게 캡처하면 불필요한 배경이나 다른 요소 없이 해당 부분만 이미지로 보관할 수 있습니다.
특히 UI 자동화 테스트에서 요소별 상태를 개별적으로 확인할 때 매우 유용합니다.

⚙️ 스크린샷 저장 시 파일 경로와 형식 지정하기

스크린샷을 저장할 때 가장 많이 하는 실수 중 하나는 파일 경로나 확장자를 잘못 지정하는 것입니다.
Selenium은 주로 .png 형식으로 저장되며, 파일 경로는 기본적으로 현재 실행 중인 프로젝트 폴더를 기준으로 합니다.
따라서 원하는 위치에 저장하려면 절대 경로나 상대 경로를 올바르게 지정해야 합니다.

만약 운영체제가 Windows라면 C:/screenshots/test.png 같은 경로를 사용할 수 있고,
Mac이나 Linux 환경에서는 /Users/username/screenshots/test.png 와 같은 경로를 사용해야 합니다.
중요한 점은 해당 폴더가 실제로 존재해야 한다는 것입니다. 폴더가 없으면 에러가 발생하므로 미리 디렉토리를 만들어 두는 것이 좋습니다.

import os
from selenium import webdriver

# 저장 폴더 생성
save_dir = "C:/screenshots"
os.makedirs(save_dir, exist_ok=True)

driver = webdriver.Chrome()
driver.get("https://www.python.org")

# 경로와 파일명 지정하여 저장
filepath = os.path.join(save_dir, "python_home.png")
driver.save_screenshot(filepath)

driver.quit()

이처럼 경로를 지정해두면 스크린샷 파일을 한곳에 모아 관리할 수 있어 편리합니다.
특히 자동화 테스트를 여러 번 수행할 경우, 날짜나 시간 정보를 포함한 파일명 규칙을 적용하면 파일이 덮어씌워지는 문제를 방지할 수 있습니다.

• 🛠️폴더가 존재하지 않으면 os.makedirs()로 생성
• ⚙️파일명에 날짜·시간을 포함해 덮어쓰기 방지
• 📂운영체제별 경로 구분자를 주의

📂 자동화 테스트 보고서에 스크린샷 활용하기

스크린샷은 단순히 화면을 저장하는 용도를 넘어, 자동화 테스트 보고서에 포함시켜 결과를 시각적으로 확인할 수 있는 강력한 도구가 됩니다.
예를 들어 기능 테스트나 UI 테스트를 수행한 뒤 스크린샷을 자동으로 수집하면, 실패한 케이스를 빠르게 파악하고 원인을 분석하는 데 큰 도움이 됩니다.
또한 협업 과정에서 개발자뿐만 아니라 기획자, 디자이너도 동일한 결과를 확인할 수 있어 커뮤니케이션 효율성이 높아집니다.

파이썬에서는 스크린샷 파일을 HTML 리포트, 엑셀, 또는 PDF 문서에 포함시킬 수 있습니다.
특히 Pytest와 같은 테스트 프레임워크를 사용할 경우, 실패한 테스트마다 스크린샷을 자동으로 첨부하도록 설정할 수 있습니다.
이 방식은 테스트 자동화에서 발생할 수 있는 버그를 빠르게 재현하고 공유하는 데 매우 유용합니다.

import pytest
from selenium import webdriver

@pytest.fixture
def driver():
    driver = webdriver.Chrome()
    yield driver
    driver.quit()

def test_example(driver):
    driver.get("https://www.python.org")
    try:
        assert "Python" in driver.title
    except AssertionError:
        driver.save_screenshot("fail_screenshot.png")
        raise

위 예제에서는 테스트가 실패했을 때 자동으로 스크린샷을 저장하도록 설정했습니다.
이렇게 저장된 이미지는 나중에 테스트 리포트와 함께 첨부할 수 있어, 어떤 화면에서 오류가 발생했는지 명확하게 파악할 수 있습니다.

💬 스크린샷은 단순한 캡처 이상의 역할을 합니다.
자동화 보고서에 포함하면 팀원 간의 협업과 버그 추적 속도를 크게 향상시킬 수 있습니다.

💡 스크린샷 저장 시 자주 하는 실수와 해결법

Selenium에서 스크린샷을 저장할 때는 의외로 사소한 실수로 인해 원하는 결과를 얻지 못하는 경우가 많습니다.
예를 들어 파일이 저장되지 않거나, 이미지가 잘리지 않고 공백 화면만 저장되는 문제가 발생할 수 있습니다.
이러한 문제는 대부분 설정이나 코드 작성 방식에서 비롯되므로 몇 가지 핵심 포인트만 기억하면 쉽게 예방할 수 있습니다.

⚠️ 저장 경로 오류

저장 경로가 존재하지 않으면 스크린샷은 생성되지 않습니다.
따라서 미리 디렉토리를 만들어 두거나, os.makedirs()를 사용해 자동으로 생성하는 것이 안전합니다.

🖥️ 화면 크기 문제

브라우저 창이 너무 작으면 원하는 요소가 화면에 다 나타나지 않아 캡처가 잘리지 않는 경우가 있습니다.
이럴 때는 driver.set_window_size(1920, 1080) 등을 활용해 해상도를 미리 조정하는 것이 좋습니다.

⏳ 로딩 지연

웹 페이지가 아직 로딩되지 않았는데 캡처를 시도하면 빈 화면만 저장될 수 있습니다.
이 문제를 해결하려면 WebDriverWait을 사용해 요소가 표시될 때까지 기다린 후 스크린샷을 찍는 것이 안전합니다.

결국 스크린샷을 제대로 저장하기 위해서는 경로 관리, 창 크기 설정, 로딩 대기 이 세 가지를 꼭 점검해야 합니다.
이 원칙만 지켜도 대부분의 오류를 쉽게 예방할 수 있습니다.

❓ 자주 묻는 질문 (FAQ)