python+selenium多窗口切换

#-*- coding:utf -8 -*-
###多窗口切换

from selenium import webdriver
import time

first_url='http://www.baidu.com'
driver = webdriver.Chrome()
driver.implicitly_wait(10)
driver.get(first_url)
driver.maximize_window()

#获取百度搜索窗口句柄
handle_window = driver.current_window_handle
print(handle_window)

driver.find_element_by_link_text('登录').click()
driver.find_element_by_link_text('立即注册').click()

#获取当前所有打开的窗口的句柄
all_handles = driver.window_handles
print(all_handles)

#进入注册页面
for handle in all_handles:
    if handle != handle_window:
        driver.switch_to.window(handle)
        print('Now register window!')
        driver.find_element_by_name('userName').send_keys('XXXXXXXXX')
        driver.find_element_by_name('phone').send_keys('XXXXXXX')
        time.sleep(4)


#返回搜索窗口
for handle in all_handles:
    if handle == handle_window:
        driver.switch_to.window(handle)
        print('Now sreach window!')
        driver.find_element_by_id('TANGRAM__PSP_4__closeBtn').click()
        driver.find_element_by_id('kw').send_keys('Selenium2.0')
        driver.find_element_by_id('su').click()
        time.sleep(3)

driver.quit()