【Python】Seleniumを使ってコマンドラインから標準入力したURLのページのスクリーンショットを撮影する方法

selenium-stdin-screenshot.py

# coding: utf-8

from selenium import webdriver

while True:
  # 標準入力の値を取得
  url = raw_input()
  
  # FireFox起動
  driver = webdriver.Firefox()
  
  # 10秒のWait設定
  driver.implicitly_wait(10)
  
  # ページ表示
  driver.get(url)
  
  # スクリーンショット撮影
  driver.save_screenshot("%s.png" % driver.title)
  
  # 終了
  driver.quit()