python selenium2 窗口切換實例

遍歷hao123中某一區域的所有鏈接，點擊每個鏈接時，會打開新的窗口，獲取新窗口的title后關閉窗口，切換到初始窗口繼續打開下一個鏈接

代碼如下：

#coding=utf-8
from selenium import webdriver
import time

base_url = "http://www.hao123.com"
driver = webdriver.Firefox()
driver.get(base_url)
handle_hao123 = driver.current_window_handle
s = driver.find_elements_by_css_selector("ul.js_bd.site-bd.site-hd0>li>a")
for i in s:
    i.click()
    all_handles = driver.window_handles

    #切換到新打開的窗口
    for handle in all_handles:
        if handle != handle_hao123:
            driver.switch_to.window(handle)

    print driver.title
    #關閉新窗口
    driver.close()
    #切換到初始窗口
    driver.switch_to.window(handle_hao123)

#為了表示最后回到初始窗口再加一個打印
print driver.title,driver.current_url
driver.close()

不妥之處請指正