selenium学习笔记（智能等待）

博主在尝试对百度首页用selenium完成自动登录的功能

反复多次尝试元素定位方法也未写错、最后发现问题原因：

脚本运行速度快于页面加载速度

如百度首页登录例子、脚本已经开始寻找登录弹窗

但是页面仍在加载、导致程序报错

博主就整理下智能等待的方法

首先是显示等待：即等待页面某个元素出现、超时则抛出错误

#!/usr/bin/env python
# -*- coding: utf_8 -*-

from learn_webdriver import Webdriver
from selenium import webdriver
from selenium.webdriver.support.ui import WebDriverWait
from time import sleep

browser_chrome = webdriver.Chrome(Webdriver.chrome())
browser_chrome.get("http://www.baidu.com")

browser_chrome.find_element_by_css_selector("html body div#wrapper div#head div.head_wrapper div#u1 a.lb").click()
# 定位百度首页中登录按钮元素
WebDriverWait(browser_chrome, 10).until(lambda the_driver:
                                        browser_chrome.find_element_by_class_name("tang-foreground").is_displayed())
browser_chrome.find_element_by_class_name("tang-foreground").find_element_by_name("userName").send_keys(u"baidu用户名")
# 定位到登录窗口后再定位到用户名输入框
browser_chrome.find_element_by_name("password").send_keys(u"baidu密码")
sleep(2)
print browser_chrome.title
browser_chrome.quit()

这里主要是使用 WebDriverWait() 类构建方法、直到找到登录弹窗元素为止

 

下面是隐形等待：即设置等待时间、等待页面完成新的操作后、若超时则抛出错误

#!/usr/bin/env python
# -*- coding: utf_8 -*-

from learn_webdriver import Webdriver
from selenium import webdriver
from selenium.webdriver.support.ui import WebDriverWait
from time import sleep

browser_chrome = webdriver.Chrome(Webdriver.chrome())
browser_chrome.get("http://www.baidu.com")

browser_chrome.find_element_by_css_selector("html body div#wrapper div#head div.head_wrapper div#u1 a.lb").click()
# 定位百度首页中登录按钮元素
browser_chrome.implicitly_wait(30)
# 添加智能等待时间
div_user = browser_chrome.find_element_by_class_name("tang-foreground").find_element_by_name("userName")
div_user.send_keys(u"baidu用户名")
div_pass = browser_chrome.find_element_by_name("password").send_keys(u"baidu密码")
sleep(2)
print browser_chrome.title
browser_chrome.quit()

 

 implicitly方法智能设置等待时间30S、超时则抛出异常