mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-23 08:11:00 +00:00
e24b4cc1b6
* mypy checking pathod * initial commit , fixed errors * tox: mypy checking to pathod * Fixed mypy test failed * issue was with args in custom_contentview.py * tox: mypy checking to #2221 * follow-import=skip since we cant provide args to custom_contentview.py during mypy testing * Lint , Typo Fixed * code style: module import
28 lines
732 B
Python
28 lines
732 B
Python
"""
|
|
This script how to generate a mitmproxy dump file,
|
|
as it would also be generated by passing `-w` to mitmproxy.
|
|
In contrast to `-w`, this gives you full control over which
|
|
flows should be saved and also allows you to rotate files or log
|
|
to multiple files in parallel.
|
|
"""
|
|
import random
|
|
import sys
|
|
from mitmproxy import io, http
|
|
import typing # noqa
|
|
|
|
|
|
class Writer:
|
|
def __init__(self, path: str) -> None:
|
|
if path == "-":
|
|
f = sys.stdout # type: typing.IO[typing.Any]
|
|
else:
|
|
f = open(path, "wb")
|
|
self.w = io.FlowWriter(f)
|
|
|
|
def response(self, flow: http.HTTPFlow) -> None:
|
|
if random.choice([True, False]):
|
|
self.w.add(flow)
|
|
|
|
|
|
addons = [Writer(sys.argv[1])]
|