前两篇文章介绍了安装,此篇文章算是一个简单的进阶应用吧!它是在windows下通过selenium+python实现自动访问firefox和chrome并实现搜索截图的功能。
自动访问firefox
可以参照前文安装selenium环境,目前selenium这个用于web应用程序测试的工具支持的浏览器包括ie、mozilla firefox、mozilla
suite、chrome等。但是由于firefox是默认安装路径,webdriver可以正常访问它,而chrome和ie需要设置driver路径。
运行效果如下图所示,自动调用firefox浏览器搜索,同时输出断言错误:
assert "谷歌" in driver.title assertionerror

源码分析
from selenium import webdriver from selenium.webdriver.common.keys import keys import sys
首先导入selenium.webdriver模板,它提供了webdriver的实现方法,目前支持这些方法的有firefox、chrome、ie和remote。同时导入keys类,它提供了操作键盘的快捷键,如reture、f1、alt等。最后导入sys主要是设置编码方式。
reload(sys) sys.setdefaultencoding('gb18030')
由于汉语中可能会遇到错误:
unicodedecodeerror: 'ascii' codec can't decode byte 0xc4 in position 33
unicodedecodeerror: 'utf8' codec can't decode byte 0xb0 in position 35
所以此处转换成gb编码,该篇不重点介绍了。
driver = webdriver.firefox() driver.get("http://www.baidu.com")
创建firefoxwebdriver实例。其中firefox最简单,其他chrome还需要driver和配置路径。接下来通过driver.get()打开百度url网页,webdriver会等待网页元素加载完成之后才把控制权交回脚本。但是,如果要打开了页面在加载的过程中包含了很多ajax,webdriver可能无法准确判断页面何时加载完成。
assert "百度" in driver.title assert "谷歌" in driver.title
接下来使用断言判断文章的标题title是否包含“百度”和“谷歌”。对应的标题是“百度一下,你就知道”,所以其中“百度”包括,而“谷歌”会出现断言报错。
同时提交页面并获得返回结果,为了判断结果是否成功返回也可以使用断言。
elem = driver.find_element_by_name("wd")
webdriver提供了很多如find_element_by_*的方法来匹配要查找的元素。如利用name属性查找方法find_element_by_name来定位输入框,审查元素name=wd。
elem.send_keys("eastmount") elem.send_keys(keys.return)
send_keys方法可以用来模拟键盘操作,相当于是在搜索框中输入“eastmount”再按回车键搜索。但首先要从selenium.webdriver.common.keys导入keys类。
driver.save_screenshot('baidu.png') driver.close() driver.quit()
最后是调用save_screenshot进行截图,但是图片是过程中的,怎样获取最后加载的图片呢?同时,操作完成并关闭浏览器。当然,也可以调用quit()方法,两者的区别在于:quit()方法会退出浏览器,而close()方法只是关闭页面,但如果只有一个页面被打开,close()方法同样会退出浏览器。
自动访问chrome
代码如下:
需要放置chromedriver如下路径,同时可以通过代码设置。但是由于我的chrome可能bug一直未修复,总是打开错误。
driver = webdriver.chrome(executable_path="g:\chromedriver.exe")
参考资料:
最后希望该篇基础性文章对你有所帮助吧!如果有不足之处,还请海涵~