Rooba
/
mitmproxy
mirror of https://github.com/mitmproxy/mitmproxy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update io-write-flow-file.py example with option (#6464) 

#### Description

Update example addon as described in
https://github.com/mitmproxy/mitmproxy/issues/6445

#### Checklist

 - [x] I have updated tests where applicable.

---------

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
Co-authored-by: Maximilian Hils <git@maximilianhils.com>
This commit is contained in:
Daniel Karandikar 2024-03-06 21:18:43 +00:00 committed by GitHub
parent 330244cabd
commit b4f49d0186
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
examples/addons/io-write-flow-file.py
View File

@ -8,8 +8,8 @@ flows should be saved and also allows you to rotate files or log
to multiple files in parallel.
"""
import os
import random
import sys
from typing import BinaryIO
from mitmproxy import http
@ -17,8 +17,11 @@ from mitmproxy import io
class Writer:
def __init__(self, path: str) -> None:
self.f: BinaryIO = open(path, "wb")
def __init__(self) -> None:
# We are using an environment variable to keep the example as simple as possible,
# consider implementing this as a mitmproxy option instead.
filename = os.getenv("MITMPROXY_OUTFILE", "out.mitm")
self.f: BinaryIO = open(filename, "wb")
self.w = io.FlowWriter(self.f)
def response(self, flow: http.HTTPFlow) -> None:
@ -29,4 +32,4 @@ class Writer:
self.f.close()
addons = [Writer(sys.argv[1])]
addons = [Writer()]