基于python的selenium实现多窗口切换

#coding=UTF-8
'''
Created on 2016-7-7

@author: sun
'''
#coding=UTF-8
from selenium import webdriver
import time

driver=webdriver.Firefox()
driver.implicitly_wait(10)
driver.get("http://www.baidu.com")
#获得百度搜索窗口句柄
sreach_windows=driver.current_window_handle

driver.find_element_by_link_text(u'登录').click()
driver.find_element_by_link_text(u"立即注册").click()
#获得当前所有打开的窗口的句柄
all_handles=driver.window_handles
#进入注册窗口
for handle in all_handles:
    if handle !=sreach_windows:
        driver.switch_to_window(handle)
        print ('now register window!')
        driver.find_element_by_name("phone").send_keys('15143049892')
        driver.find_element_by_name("password").send_keys('password')
#回到搜索窗口        
for handle in all_handles:
    if handle==sreach_windows:
        driver.switch_to_window(handle)
        print ('no sreach window!')
        driver.find_element_by_id('TANGRAM__PSP_2__closeBtn').click()
        driver.find_element_by_id("kw").send_keys("selenium")
        driver.find_element_by_id("su").click()
        time.sleep(2)
        
driver.quit()