【Python】Seleniumを使って標準入力したURLからRSSフィードのURLを探す方法

find-feed-url.py

# coding: utf-8

from selenium import webdriver
import re, urllib

# Firefox起動
driver = webdriver.Firefox()

while True:
    # 標準入力の値を取得
    url = raw_input()

    # urlをパース
    type, opaquestring = urllib.splittype(url)

    # urlのスキームをチェック
    if type not in ("http", "https"):
        driver.close()

    # ページ表示
    driver.get(url)

    # linkタグを探す
    for link in driver.find_elements_by_tag_name("link"):
        # linkタグのtypeにrssが含まれているかチェック
        if re.search('rss', link.get_attribute("type")) is not None:
            # hrefを取得
            print link.get_attribute('href')

driver.close()