selenium之多窗口切换

from selenium import webdriver
from selenium.webdriver.common.by import By
import time
'''
使用Switch_to.window()方法，进行多窗口切换
'''
driver = webdriver.Edge()
# 隐式等待10秒
driver.implicitly_wait(10)
# 打开指定网页
driver.get("http://www.baidu.com")

# 获得百度搜索窗口句柄
# 通过current_window_handle 获得当前窗口的网页名称，并赋值给变量search_windows
search_windows = driver.current_window_handle

# driver.find_element("登录").click()
# driver.find_element("立即注册").click()

# 打开登录弹窗，
driver.find_element(By.LINK_TEXT, '登录').click()
# 在登录弹窗上单击“立即注册”按钮，从而打开新的注册窗口
driver.find_element(By.LINK_TEXT, '立即注册').click()
# driver.find_element()
# driver.find_element_by_link_text('登录').click()
# driver.find_element_by_link_text('立即注册').click()

#获得所有打开的窗口的句柄
# 通过window_handles方法获取当前打开的所有窗口的句柄，并赋值给变量all_windows

all_windows = driver.window_handles

#进入注册窗口
# 通过第一个循环遍历all_windows
for handle in all_windows:
    # 如果handle不等于search_windows,那么一定是注册页面，因为脚本执行的过程中只打开了两个窗口
    if handle != search_windows:
        # 通过switch_to.window()切换到注册页进行注册操作
        driver.switch_to.window(handle)
        print("now register window!")
        driver.find_element(By.NAME, 'userName').send_keys('rongxiangbo123')
        time.sleep(5)
        # driver.find_element('userName').send_keys("rongxiangbo")
        driver.find_element(By.NAME, 'phone').send_keys('18132086962')
        time.sleep(5)
        # driver.find_element('phone').send_keys("18132086961")
        # driver.find_element(By.NAME, 'password').send_keys('123123')
        # driver.find_element('password').send_keys("123123")
        # time.sleep(2)

# 通过第二个循环遍历all_windows,查找百度搜索页面，并跳转到搜索页面
for handle in all_windows:
    if handle == search_windows:
        # 通过switch_to.window()跳转到百度搜索页面
        driver.switch_to.window(handle)
        print("now sreach window")
        # 通过ID元素定位,关闭[登录]弹出窗口
        driver.find_element(By.ID, 'TANGRAM__PSP_4__closeBtn').click()
        time.sleep(2)
        driver.find_element(By.ID, 'kw').send_keys('selenium')
        # driver.find_element('wd').send_keys('selenium')
        driver.find_element(By.ID, 'su').click()
        # driver.find_element('su').click()
        time.sleep(2)

time.sleep(5)

driver.quit()