Selenium-iframe+pop window 小练习

from selenium import webdriver
from time import sleep

#启动浏览器
driver = webdriver.Firefox()
#打开网址
driver.get("http://www.ccb.com/cn/home/indexv3.html")
sleep(2)
#定位并点击登录按钮
login_button = driver.find_element_by_link_text("登录")
sleep(1)
login_button.click()
sleep(3)
#获取当前打开的所有窗口句柄
handles = driver.window_handles
#轮询所有句柄
for handle in handles:
#切换到当前窗口名柄
    driver.switch_to.window(handle)
    # print(driver.title)
#判断当前窗口标题是否与预期是否一样
    if driver.title == "这里是预期窗口标题":
#切换进入frame
        driver.switch_to.frame(0)
        sleep(2)
#定位用户名输入框并输入
        user_input = driver.find_element_by_id("USERID")
        sleep(2)
        user_input.send_keys("test")
        sleep(2)
#定位密码框并输入
        pass_input = driver.find_element_by_id("LOGPASS")
        sleep(2)
        pass_input.send_keys("123456")
        sleep(2)
#切回默认iframe
        driver.switch_to.default_content()
        sleep(2)
#定位顶部链接并输出指定链接文本
        links = driver.find_elements_by_class_name('top')
        # print(type(links))
        print(links[0].text)
#退出浏览器，即关闭所有打开的窗口
driver.quit()