PythonでSeleniumを使用してスクリーンショットを撮る

PythonでSeleniumを使用してスクリーンショットを撮ってみます。

スクリーンショットを撮るためには、Seleniumモジュールが必要となりますので、事前にインストールする必要があります。

■Python

Pythonバージョン

今回のPythonのバージョンは、「3.8.2」を使用しています。(Windows10)

■Seleniumを使用してスクリーンショットを撮る

では、早速Seleniumモジュールを使用してスクリーンショットを撮るスクリプトを書いていきます。

■コード

from selenium import webdriver
import chromedriver_binary

driver = webdriver.Chrome()

url = "https://laboratory.kazuuu.net/"

driver.get(url)

driver.save_screenshot("image_test.png")

インポートでSeleniumモジュールを呼び出して、webdriver.Chrome()関数で、Chromeを使用し、get()関数で、Chromeで指定したurlへ遷移します。今回は、当サイト(https://laboratory.kazuuu.net/)のスクリーンショットを撮ってみます。

スクリーンショットを撮るためには、save_screenshot()関数を使います。()内で撮ったスクリーンショットをpng形式が画像ファイルとして保存します。

■備考(注意)

driver = webdriver.Chrome()
File “C:\Users\user\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.8_qbz5n2kfra8p0\LocalCache\local-packages\Python38\site-packages\selenium\webdriver\chrome\webdriver.py”, line 73, in __init__
self.service.start()
File “C:\Users\user\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.8_qbz5n2kfra8p0\LocalCache\local-packages\Python38\site-packages\selenium\webdriver\common\service.py”, line 81, in start
raise WebDriverException(
selenium.common.exceptions.WebDriverException: Message: ‘chromedriver’ executable needs to be in PATH. Please see https://sites.google.com/a/chromium.org/chromedriver/home

プログラムを実行し、上記のようなエラーが発生する場合は、現在お使いのGoogleChromeのバージョンを確認し、それに応じたchromedriver-binaryをインストールします。

インストール後、スクリプト上でchromedriver-binaryをインポートで呼び出します。

■実行

今回のスクリプトを「website_screenshots.py」という名前で保存し、コマンドプロンプトから実行します。

実行してみると、Chromeが起動し、指定されたurlへ遷移します。

その後、カレントディレクトリを確認すると、png形式が画像ファイルが作成されます。この画像ファイルの中身を確認してみると、指定したurlのスクリーンショットが撮れていることが確認できました。

コメント

タイトルとURLをコピーしました