플그래밍/파이써언

[파이썬] Selenium - 웹사이트 스크롤풀샷 캡쳐 법

훗티v 2022. 12. 14. 20:35
728x90
반응형

기능

파이썬 셀레니움을 활용하여 웹사이트의 풀 스크린샷(스크롤 끝까지)을 캡쳐하는 방법입니다.

코드

import time
from selenium import webdriver
from selenium.webdriver.chrome.options import Options

def full_screenshot(driver, url, output_path):
    driver.get(url)
    time.sleep(1)  # Give the page some time to load

    total_height = driver.execute_script("return document.body.parentNode.scrollHeight")

    driver.execute_script("window.scrollTo(0, document.body.parentNode.scrollHeight);")

    time.sleep(1)

    driver.set_window_size("1920", total_height)

    time.sleep(1)
    driver.save_screenshot(output_path)

chrome_options = Options()
chrome_options.add_argument('--headless')
driver = webdriver.Chrome(options=chrome_options)

url = "https://www.daum.net"
output_path = "screenshot.png"
full_screenshot(driver, url, output_path)
driver.quit()

예제

 

 

 

 

 

 

반응형