ライブラリ: selenium

2015/01/202016/05/26

Python の「 selenium 」というライブラリをご紹介します。

import selenium

selenium ライブラリは、同名の Selenium というブラウザ操作用のツールがあるのですが、その Python バインディング、いわゆる「 Selenium の Python 版」です。

selenium ライブラリを使うことで Firefox などのブラウザを Python からかんたんに操作できるようになります。

以下、使い方をかんたんに見ていきます。まずはインストールから。

インストール

pip が入っていれば、他のライブラリと同じように pip install でインストールすることができます。

pip install selenium

Mac 環境だとこの形でインストールできますが、 Windows 環境では少し異なるかもしれません。 Python 以外の情報も含む Selenium 全体の公式ダウンロードページはこちらになります。

Downloads | Selenium

使い方

基本的な使い方を見ていきます。

最小単位は次のような形になるでしょうか。

# ライブラリの読み込み
from selenium import webdriver

# ドライバの生成
# 問題なければFirefoxブラウザが自動的に起動する
driver = webdriver.Firefox()

# Firefox上でURLを指定してGETリクエスト
driver.get("https://www.google.co.jp/")

# ドライバを閉じる
# 問題なければFirefoxも終了する
driver.close()

ブラウザを開いて、特定のページを開いて、ブラウザを閉じる、という一連の流れがこれだけのコードで実現できます。

ちなみに、 selenium ではブラウザを操作するインスタンスを「ドライバ」と読んでいます。デフォルトで使えるのは Firefox ですが、Firefox のほかにも Chrome Safari PhantomJS IE など主要なブラウザを操作するドライバがサードパーティ製で公開されているので、適宜インストールすれば使うことができるようになります。

上の例ではページを開いて閉じただけであまりおもしろみがないので、キーボード操作をするような例についても見てみたいと思います。

こちらは公式ドキュメントの例にコメントをつけたものになります。

# ライブラリの読み込み
# Keys はキーボード操作に相当するモジュール
from selenium import webdriver
from selenium.webdriver.common.keys import Keys

# ドライバの生成
# Firefoxを起動させる
driver = webdriver.Firefox()

# FirefoxでGETリクエスト
driver.get("https://www.python.org")

# 期待どおりのページが開かれたことを確認するアサーション
assert "Python" in driver.title

# ページ内のHTML要素をname属性を使って取得
# name属性の値が q である要素が選択される
elem = driver.find_element_by_name("q")

# 選択した要素に対してキーボード入力
# selenium という文字を入力し、リターンキーを押す
elem.send_keys("selenium")
elem.send_keys(Keys.RETURN)

# 期待どおりのページが開かれたことを確認するアサーション
assert "Google" in driver.title

# ドライバを閉じる
# Firefoxも終了する
driver.close()