不需要将request内容保存为多个文件,保存到一个txt中即可
时间: 2024-03-09 16:50:33 浏览: 21
好的,你可以尝试以下代码,它会将所有的request内容保存到一个名为`requests.txt`的文件中:
```python
import requests
class AddHeader:
def __init__(self):
self.num = 0
self.file = open("requests.txt", "w")
def request(self, flow: http.HTTPFlow) -> None:
self.num += 1
flow.request.headers["My-Header"] = f"Request {self.num}"
print("Request Headers: ")
for name, value in flow.request.headers.items():
print(f"{name}: {value}")
self.save_request(flow)
def response(self, flow: http.HTTPFlow) -> None:
self.num += 1
flow.response.headers["My-Header"] = f"Response {self.num}"
print("Response Headers: ")
for name, value in flow.response.headers.items():
print(f"{name}: {value}")
def save_request(self, flow: http.HTTPFlow):
self.file.write(f"{flow.request.method} {flow.request.url}\n")
for name, value in flow.request.headers.items():
self.file.write(f"{name}: {value}\n")
self.file.write("\n")
if flow.request.method == "POST":
self.file.write(flow.request.content.decode("utf-8"))
self.file.write("\n\n")
addons = [
AddHeader()
]
if __name__ == "__main__":
mitmdump(['-p', '8081', '-s', __file__])
```
在以上代码中,`AddHeader`类的构造函数会打开一个名为`requests.txt`的文件,并将其保存在`self.file`属性中。`save_request`方法则会将抓取到的request内容以文本形式保存到该文件中。当程序结束时,该文件会自动关闭。
注意:如果你已经运行过之前的代码,需要先删除已经生成的`request_*.txt`文件,否则会出现文件名冲突。
相关推荐
![txt](https://img-home.csdnimg.cn/images/20210720083642.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)