Seleniumとchromedriverでブラウザの自動操作を行い、スクリーンショットで画像を保存する

PythonでSeleniumという便利なライブラリを利用すれば、ブラウザーを自動操作することができます。

今回は自動操作でgoogleの検索から、検索結果をスクリーンショットするまでのコードを紹介します。

コンテンツ

環境

環境は以下になります。

環境

・macOS Big Sur

・python 3.9

・Chorme 87.0

到達目標

上の画像が今回のゴールになります。

pythonファイルからの自動操縦で、「django」の検索結果からそのままスクリーンショットを行いました。

通常だとMacの場合

shift + command + 3
shift + command + 4

あたりかと思いますが、全てpythonのコードで実行します。

インストール

必要なライブラリをインストールします。

ヒロヤンはpipenvを使用しているので以下のようにpipenvにインストールを実行します。

$ pipenv install chromedriver_binary  
$ pipenv install selenium

コードの実行

準備ができたのでコードを書いていきます。

sample.py

import chromedriver_binary
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.chrome.options import Options
from selenium import webdriver


options = Options()
driver = webdriver.Chrome(options=options)
driver.get('https://www.google.co.jp/')

# タイトルにgoogleが含まれていることを確認する
assert 'Google' in driver.title

# 検索語を入力して送信する
input_element = driver.find_element_by_name('q')
input_element.send_keys('django')
input_element.send_keys(Keys.RETURN)
# タイトルに'Python'が含まれていることを確認する
assert 'django' in driver.title

# スクリーンショットを撮る
driver.save_screenshot('django_result.png')

コードが具体的に何をしているかはコメントアウトに書いた通りです。