网页截图+cookie处理+自动化验证码处理
网页截图
案例：打开百度页面，截图
#/usr/bin python #-*- coding:UTF-8 -*- #####################截图##################### from selenium import webdriver from time import sleep driver = webdriver.Chrome() driver.get("https://www.baidu.com/") ##截图 driver.get_screenshot_as_file(r"E:\study\python\baidu.jpg") sleep(5) driver.quit()
cookie的处理
Cookie是储存在用户本地终端上的数据，实际上是一小段的文本信息。
帮助 Web 站点保存有关访问者的信息，方便用户的访问。如记住用户名密码实现自动登录。
在自动化UI测试过程中，也可以用来避开登录
driver.get(url) driver.delete_all_cookies() for cookie in cookies: driver.add_cookie(cookie) driver.refresh()
# -*- coding: utf-8 -*- # @Time : 2020/7/17 15:06 # @Author : 饭盆里 # @File : test_cookie.py # @Software: PyCharm # @desc : import shelve from time import sleep import pytest from selenium import webdriver from selenium.webdriver.chrome.options import Options from selenium.webdriver.common.by import By class Test(): def setup(self): options = Options() # 和浏览器打开的调试端口进行通信 # 浏览器要使用 --remote-debugging-port=9222 开启调试， # 由于我环境变量设置了变量，alias driver_debugging=\'/Applications/Google\ Chrome.app/Contents/MacOS/Google\ Chrome --remote-debugging-port=9222\' # 所以可以直接：driver_debugging options.debugger_address = \'127.0.0.1:9222\' self.driver = webdriver.Chrome(options=options) # self.driver = webdriver.Chrome() self.driver.implicitly_wait(3) def teardown(self): self.driver.quit() @pytest.mark.skip def test_baidu(self): self.driver.get(\'https://www.baidu.com/\') self.driver.find_element(By.ID,\'kw\').send_keys(\'hi\') self.driver.find_element(By.XPATH,\'//*[@id="su"]\').click() sleep(5) def test_wework(self): self.driver.get("https://work.weixin.qq.com/") # 获取 cookies # print(self.driver.get_cookies()) cookies = [{\'domain\': \'.work.weixin.qq.com\', \'httpOnly\': False, \'name\': \'wwrtx.vid\', \'path\': \'/\', \'secure\': False, \'value\': \'1688850531850742\'}, {\'domain\': \'.work.weixin.qq.com\', \'httpOnly\': True, \'name\': \'wwrtx.vst\', \'path\': \'/\', \'secure\': False, \'value\': \'PI81GkMu8osptnJ46fIu-UWJvR3o8_ViBwX5USscimOQRrHRyNdnruNfbjNLbSU2p3AvksJoKJc6tACT8EJXxe89v6xc0h_5FZUIikMzapf27LnrSjJkfUxRoHkefiaUAoTRgR0a_QBWoj8mEOELID9xl2Etr5gBTKAqCPdJzz3aBZC5b7iKufwcWtgb54qvKyXIK4J5zMnAy8DBzvrRb2pUE_hYYbkptzwyfk5rR7xiK01F3fqXjJwA2QIuI5R_e35ZDA10FFYBxpMtVyvbJA\'}, {\'domain\': \'.work.weixin.qq.com\', \'httpOnly\': False, \'name\': \'wxpay.vid\', \'path\': \'/\', \'secure\': False, \'value\': \'1688850531850742\'}, {\'domain\': \'.work.weixin.qq.com\', \'httpOnly\': False, \'name\': \'wxpay.corpid\', \'path\': \'/\', \'secure\': False, \'value\': \'1970325138147142\'}, {\'domain\': \'.work.weixin.qq.com\', \'expiry\': 1626742501, \'httpOnly\': False, \'name\': \'Hm_lvt_9364e629af24cb52acc78b43e8c9f77d\', \'path\': \'/\', \'secure\': False, \'value\': \'1594975399,1594979354,1594979612,1595206501\'}, {\'domain\': \'.qq.com\', \'expiry\': 2147385600, \'httpOnly\': False, \'name\': \'pgv_pvid\', \'path\': \'/\', \'secure\': False, \'value\': \'8752271993\'}, {\'domain\': \'.work.weixin.qq.com\', \'httpOnly\': True, \'name\': \'wwrtx.ref\', \'path\': \'/\', \'secure\': False, \'value\': \'direct\'}, {\'domain\': \'.work.weixin.qq.com\', \'httpOnly\': True, \'name\': \'wwrtx.ltype\', \'path\': \'/\', \'secure\': False, \'value\': \'1\'}, {\'domain\': \'.work.weixin.qq.com\', \'httpOnly\': False, \'name\': \'wwrtx.d2st\', \'path\': \'/\', \'secure\': False, \'value\': \'a3834163\'}, {\'domain\': \'work.weixin.qq.com\', \'expiry\': 1595238009, \'httpOnly\': True, \'name\': \'ww_rtkey\', \'path\': \'/\', \'secure\': False, \'value\': \'oj731e\'}, {\'domain\': \'.work.weixin.qq.com\', \'httpOnly\': False, \'name\': \'Hm_lpvt_9364e629af24cb52acc78b43e8c9f77d\', \'path\': \'/\', \'secure\': False, \'value\': \'1595206501\'}, {\'domain\': \'.work.weixin.qq.com\', \'httpOnly\': True, \'name\': \'wwrtx.refid\', \'path\': \'/\', \'secure\': False, \'value\': \'26225621643276726\'}, {\'domain\': \'.work.weixin.qq.com\', \'httpOnly\': True, \'name\': \'wwrtx.sid\', \'path\': \'/\', \'secure\': False, \'value\': \'lvWc6ImOzsx44Nw-2SegyMqIl7_H2EiVGhYcto58RPhmIx18vxT56MuBeZCy7aHo\'}, {\'domain\': \'.work.weixin.qq.com\', \'expiry\': 1595238009, \'httpOnly\': False, \'name\': \'wwrtx.c_gdpr\', \'path\': \'/\', \'secure\': False, \'value\': \'0\'}, {\'domain\': \'.qq.com\', \'expiry\': 1658051906, \'httpOnly\': False, \'name\': \'_ga\', \'path\': \'/\', \'secure\': False, \'value\': \'GA1.2.1583622381.1594901080\'}, {\'domain\': \'.work.weixin.qq.com\', \'expiry\': 1597798500, \'httpOnly\': False, \'name\': \'wwrtx.i18n_lan\', \'path\': \'/\', \'secure\': False, \'value\': \'zh-cn\'}] # # 创建或者打开一个数据库 db = shelve.open("cookies") # # 将数据存储到 shelve 中 db["cookies"] = cookies # # # 取出数据 cookies = db["cookies"] #把cookie 中的时间戳去除"expiry",然后加入到浏览器的cookies中 for cookie in cookies: if "expiry" in cookie.keys(): cookie.pop("expiry") # 把字典加入到 driver 的 cookie 中 self.driver.add_cookie(cookie) sleep(2) self.driver.get("https://work.weixin.qq.com/wework_admin/frame#contacts") # sleep(4) self.driver.find_element(By.XPATH, "//*[@id=\'menu_contacts\']").click() # db.close()
- 手动在代理浏览器页面进行扫码登陆企业微信
- 用代码获取此时的cookie：
self.driver.get_cookies()
将此时的cookie打印后，放入一个变量中，以备后续使用
#/usr/bin python #-*- coding:UTF-8 -*- #####################cookie##################### from selenium import webdriver from time import sleep driver = webdriver.Chrome() driver.get("https://www.baidu.com/") ##获取cookie cookie = driver.get_cookies() ##打印全部cookie print(cookie) ##打印第一组cookie print(cookie[0]) #添加cookie 字典形式 driver.add_cookie({\'name\':\'fanpl\',\'value\':\'fanfan\'}) for cookie in driver.get_cookies(): print("%s --- %s" %(cookie[\'name\'],cookie[\'value\'])) driver.quit()
自动化测试验证码问题
验证码作用
不少网站在用户登录、用户提交信息等登录和输入的页面上使用了验证码技术。验证码技术可以有效防止恶意用户对网站的滥用，使得网站可以有效避免用户信息失窃、保证网站稳定安全性。
但是验证码给自动化测试带来一些不便，使脚本无法正常运行覆盖功能模块。
解决办法
1.去掉验证码
这是最简单的方法，对于开发人员来说，只是把验证码的相关代码注释掉即可，如果是在测试环境，这样做可省去了测试人员不少麻烦，如果自动化脚本是要在正式环境跑，这样就给系统带来了一定的风险。
2.设置万能码
去掉验证码的主要是安全问题，为了应对在线系统的安全性威胁，可以在修改程序时不取消验证码，而是程序中留一个“后门”—设置一个“万能验证码”，只要用户输入这个“万能验证码”，程序就认为验证通过，否则按照原先的验证方式进行验证。
3.验证码识别技术（OCR）
例如可以通过Python-tesseract 来识别图片验证码，Python-tesseract是光学字符识别Tesseract OCR引擎的Python封装类。能够读取任何常规的图片文件(JPG, GIF ,PNG , TIFF等)。不过，目前市面上的验证码形式繁多，目前任何一种验证码识别技术，识别率都不是100% 。
4.记录cookie
通过向浏览器中添加cookie 可以绕过登录的验证码。
基于Cookie绕过验证码自动登录
案例：使用Cookie绕过百度验证码自动登录账户。（需要用到抓包工具）
from selenium import webdriver from time import sleep driver=webdriver.Firefox() driver.get("http://www.baidu.com/") #手动添加cookie driver.add_cookie({\'name\':\'BAIDUID\',\'value\':\'9E4BF1D44014…(根据实际获取值填写)\'}) driver.add_cookie({\'name\':\'BDUSS\',\'value\':\'根据实际抓包获取值填写\'}) sleep(2) driver.refresh() sleep(3) driver.quit()
版权声明：本文为fanpl原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。