python selenium打開新窗口，多窗口切換

# coding=utf-8

from selenium import webdriver

browser=webdriver.Firefox()
browser.maximize_window()  # 窗口最大化

browser.get('https://www.baidu.com')  # 在當前瀏覽器中訪問百度

# 新開一個窗口，通過執行js來新開一個窗口
js='window.open("https://www.sogou.com");'
browser.execute_script(js)

print (browser.current_window_handle)  # 輸出當前窗口句柄（百度）
handles = (browser.window_handles)  # 獲取當前窗口句柄集合（列表類型）
print (handles)  # 輸出句柄集合

for handle in handles:# 切換窗口（切換到搜狗）
    if handle!=browser.current_window_handle:
        print ('switch to ',handle)
        browser.switch_to_window(handle)
        print (browser.current_window_handle)  # 輸出當前窗口句柄（搜狗）
        break

browser.close()  # 關閉當前窗口（搜狗）
browser.switch_to_window(handles[0])  # 切換回百度窗口
import time
time.sleep(10)
browser.quit()