目录
前言:
一.简介
二.用法
1.安装selenium-wire库
2.获取请求信息
3.获取响应信息
4.实例
前言:
Selenium Wire 是一个 Python 库,它扩展了 Selenium WebDriver 的功能,使其能够轻松地拦截、修改和检查浏览器与服务器之间的网络请求。它提供了更高级的网络交互能力,能够方便地进行网络级别的测试和验证。
一.简介
以下来自chatGPT回答:
selenium-wire是一个基于selenium的Python库,它扩展了selenium的功能,使得我们可以在自动化测试中直接访问和修改浏览器的网络请求和响应。selenium-wire可以拦截和修改HTTP请求和响应,从而可以在测试过程中模拟 网络环境、调试和分析网络请求以及实现自定义的网络请求和响应处理逻辑。与selenium自带的webdriver不同,selenium-wire使用了第三方库mitmproxy来实现网络请求的拦截和修改。因此,使用selenium-wire需要先安装mitmproxy。
二.用法
1.安装selenium-wire库
pip install selenium-wire
2.获取请求信息
1)获取所有的请求信息
get_list = driver.requests # 返回的是个数组
当调用 driver.requests时,返回的是当前页面所有已经请求并响应过了的接口数据。如果某个请求还没有完成或者被阻塞,那么这个请求对应的数据不会出现在 requests
列表中。
2)获取请求行/头/体
for i in get_list:if 'https://www.baidu.com/sugrec' in i.url:print(i.url) # 请求地址print(i.date) # 请求时间print(i.method) # 请求方式print(i.headers) # 请求头 or i.headers['Content-Type']print(i.params) # 请求参数print(i.host) # 请求域名
driver.requests获取的是当前页面所有的请求,因此我们在使用时需过滤下自己所要的接口信息。
3)create_response()方法
for i in get_list:if 'text/html' not in i.response.headers['Content-Type']:# create_response(status_code, headers=(), body=b'')i.create_response(200, [('Content-Type', 'text/plain')], b'["Hello","world"]') # mock接口响应
create_response()为mock接口响应信息,返回我们需要的信息。
4)abort()方法
for i in get_list:if 'text/html' not in i.response.headers['Content-Type']:i.abort(error_code=403) # 中断请求,并返回状态码403print(i.response.status_code)
这里需要注意的是3,4方法所可以改变响应数据,但服务端记录的数据还是实际请求的值。
3.获取响应信息
这里直接贴代码吧,备注很详细。
for i in get_list:if 'https://www.baidu.com/sugrec' in i.url:print(i.response.date) # 当前响应时间print(i.response.reason) # 响应状态 ok or Not foundprint(i.response.headers['Content-Type']) # 响应的数据类型print(i.response.status_code) # 响应状态码# 获取响应的body并转为json类型输出from seleniumwire.utils import decodebody = decode(i.response.body, i.response.headers.get('Content-Encoding', 'identity'))decoded_response = body.decode('utf-8') # 将二进制字节串解码为 UTF-8 编码的字符串json_response = json.loads(decoded_response) # 将 JSON 字符串转换为 Python 对象print(json_response)
4.实例
import json
import time
from selenium.webdriver.common.by import By
from seleniumwire import webdriverdriver = webdriver.Chrome()
driver.implicitly_wait(10)
driver.get("https://www.baidu.com")
driver.find_element(By.ID, value='kw').send_keys('猪')
driver.find_element(By.ID, value='su').click()
time.sleep(5) # 等待5s,让所有接口请求完
get_list = driver.requests # 获取当前所有的请求信息for i in get_list:if 'https://www.baidu.com/sugrec' in i.url:print(i.response.date) # 当前响应时间print(i.response.reason) # 响应状态 ok or Not foundprint(i.response.headers['Content-Type']) # 响应的数据类型print(i.response.status_code) # 响应状态码# 获取响应的body并转为json类型输出from seleniumwire.utils import decodebody = decode(i.response.body, i.response.headers.get('Content-Encoding', 'identity'))decoded_response = body.decode('utf-8') # 将二进制字节串解码为 UTF-8 编码的字符串json_response = json.loads(decoded_response) # 将 JSON 字符串转换为 Python 对象print(json_response)driver.quit()
作为一位过来人也是希望大家少走一些弯路,希望能对你带来帮助。(WEB自动化测试、app自动化测试、接口自动化测试、持续集成、自动化测试开发、大厂面试真题、简历模板等等),相信能使你更好的进步!
留【自动化测试】即可【自动化测试交流】:574737577(备注ccc)http://qm.qq.com/cgi-bin/qm/qr?_wv=1027&k=mDKyfdjYNAeufGp8xdITLYX82oszfcAW&authKey=WrZ1YMYKla5sOgXugyqBWmcAJglmktIVxAfVRtY1Hwt1Y7JTcse%2BT6u5PiF1D9aw&noverify=0&group_code=574737577