#coding = utf-8
'''
多視窗控制代碼
獲取
切換
'''
from selenium import webdriver
import time
driver = webdriver.Chrome()
driver.get("http://home.zhiyoo.com/")
driver.implicitly_wait(3)

#獲取當前視窗控制代碼
h = driver.current_window_handle
print(h)  #列印首頁控制代碼

driver.find_element_by_link_text(u'智友論壇').click()
driver.implicitly_wait(3)

all_h = driver.window_handles
print(all_h) #列印所有的控制代碼

#切換控制代碼
#方法一：
for i in all_h:
    if i != h:
        driver.switch_to_window(i)
        print(driver.title)
#方法二：
#driver.switch_to_window(all_h[1])
#print(driver.title)

#關閉新視窗
driver.close()
#切換到首頁控制代碼
driver.switch_to_window(h)
#列印當前的title
print(driver.title)