Selenium自动化测试实现窗口切换

import time
from selenium import webdriver
from selenium.webdriver.common.by import By
driver = webdriver.Chrome()
driver.implicitly_wait(10)

driver.get("https://www.baidu.com")
search_windows = driver.current_window_handle

driver.find_element(By.LINK_TEXT,'登录').click()
driver.find_element(By.LINK_TEXT,'立即注册').click()

all_handles = driver.window_handles

for handle in all_handles:
    if handle != search_windows:
        driver.switch_to.window(handle)
        print(driver.title)
        driver.find_element(By.NAME,'userName').send_keys('userName')
        driver.find_element(By.NAME,'phone').send_keys('138xxxxxxx')
        time.sleep(2)

        driver.close()

driver.switch_to.window(search_windows)
print(driver.title)
driver.quit()