chromeを操作するツールであるchrome driverを使用して、webサイトのスクリーンショットを撮影します。

chrome driverのダウンロード

chrome driverのversionは、PCにインストールされているGoogle Chromeと同じにする必要があります。

コード

from selenium import webdriver

options = webdriver.ChromeOptions()
# headlessモードで起動します
options.add_argument('--headless')
# ダウンロードしたchrome driverのパスを指定します
driver = webdriver.Chrome(options=options, executable_path="chromedriver_win32/chromedriver.exe")
# スクショのサイズ
driver.set_window_size(1400, 2000)
driver.get("https://www.yahoo.co.jp/")
driver.save_screenshot('screenshot.png')
driver.quit()

補足

headlessモードにしない場合は、chromeが立ち上がります。

おすすめ本