Selenium是一款比较常见的web应用自动化测试系统,它支持多种浏览器,多用于在爬虫中解决JavaScript渲染问题。由于selenium原生代码似乎是不带post方式的,笔者会以headless firefox模式为例,简单谈一下在selenium下如何进行post数据。
第三方库
以seleniumrequests为例,这个库是一个selenium扩展,使得selenium下也可以使用requests的功能,我们可以采用pip安装:
pip install selenium-requests
或者也可以去gayhub上自行clone:
https://github.com/cryzed/Selenium-Requests
当然,这个库使用起来也是很简单的:
# selenium.webdriver from the seleniumrequests module
from seleniumrequests import Firefox
# Simple usage with built-in WebDrivers:
webdriver = Firefox()
response = webdriver.request('GET', 'https://www.google.com/')
print(response)
不过这个库也有它的缺点,不方便自定义一些驱动模式参数,无法设置headless状态(也许是我自己瞎几把搞没试出来)。有兴趣的朋友可以自行研究下,其支持如下:
>>> dir(seleniumrequests)
['Android', 'Chrome', 'Firefox', 'Ie', 'Opera', 'PhantomJS', 'Remote', 'RequestMixin', 'Safari', '__builtins__', '__doc__', '__file__', '__name__', '__package__', '__path__', '__warningregistry__', 'request']
HTML文件大法好
个人不太喜欢这种法子,不过好像有一部分人比较推崇。其原理是解析了原生的post请求后,将其传递的参数重构为form表单,最后再将这些新生成的代码存入html网页。
最后,程序会再借用selenium定位submit元素,触发事件提交表单。
窃以为这种方法不太妥帖,每开一个网页程序就得生成一个新的html文件。先不论程序是否一定具有写和删的权限,做法本身是显得比较繁琐的。
FB上貌似有类似介绍这种方法的文章,这里就不贴代码了。
Ajax代行天子令
Ajax模拟post发送请求,这是笔者自己采用过的办法。当然,效果一般般,我相信应该有更好的。
无论是原生JS的XMLHttpRequest,还是Jquery,都可以模拟生成ajax post请求,最后再借助selenium执行JS代码。
XMLHttpRequest示例片段:
brower = webdriver.Firefox(firefox_options=fireFoxOptions)
js = """var xmlhttp=new XMLHttpRequest();
xmlhttp.open("GET","http://127.0.0.1/get.php",false);
xmlhttp.setRequestHeader("Content-type","application/x-www-form-urlencoded");
xmlhttp.setRequestHeader("User-Agent","Mozilla/5.0");
xmlhttp.setRequestHeader("Cookie","");
xmlhttp.send("test=1");
return xmlhttp.responseText;
"""
brower.implicitly_wait(30)
#time.sleep(30)
resp = brower.execute_script(js)
Jquery示例片段:
jquery = open("jquery.min.js", "r").read()
driver = webdriver.Firefox(firefox_options=fireFoxOptions)
driver.execute_script(jquery)
ajax_query = '''
$.ajax('%s', {
type: %s,
data: %s,
headers: { "User-Agent": "Mozilla/5.0" },
crossDomain: true,
xhrFields: {
withCredentials: true
},
success: function(){}
});
''' % (url, request_type, data)
ajax_query = ajax_query.replace(" ", "").replace("\n", "")
resp = driver.execute_script("return " + ajax_query)
尾声
总而言之,selenium没有自带原生post方式是一个遗憾,而且其调用headless模式的浏览器,渲染和启动也显得太慢了些,难以适用于单机高并发。
还是那句话,由于selenium其本身的定位和特性。个人窃以为在资源有限的情况下,它不太适用于高并发的大规模测试,做低效精准的辅助也许尚可。
建议
可以用Selenium模拟登陆后 再把cookies保存下来,放到urllib.request里用。
不过selenium页面加载顺序 网速也会引起一系难控制的bug有的难受。
还可以去试试phantomjs 效率比selenium 高一些,不会再弹出个浏览器,是后台运行的。