【Python】標準入力でヤフー検索をして検索結果をスクリーンショットを撮影する方法

stdin-search-screenshot.py

from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from datetime import datetime

# 標準入力の値を取得
keyword = raw_input()

driver = webdriver.Firefox()
driver.implicitly_wait(10)

# ヤフーにアクセス
driver.get("http://www.yahoo.co.jp/")
elem = driver.find_element_by_name("p")
# キーワード入力
elem.send_keys(keyword)
# 検索
elem.send_keys(Keys.RETURN)

# スクリーンショットを撮影
driver.save_screenshot("%s.png" % datetime.now())

driver.close()