mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-27 02:24:18 +00:00
e6eeab6094
- Addons now nest, which means that addons can manage addons. This has a number of salutary effects - the scripts addon no longer has to poke into the global addons list, we no longer have to replace/remove/boot-outof parent addons when we load scripts, and this paves the way for making our top-level tools into addons themselves. - All addon calls are now wrapped in a safe execution environment where exceptions are caught, and output to stdout/stderr are intercepted and turned into logs. - We no longer support script arguments in sys.argv - creating an option properly is the only way to pass arguments. This means that all scripts are always directly controllable from interctive tooling, and that arguments are type-checked. For now, I've disabled testing of the har dump example - it needs to be moved to the new argument handling, and become a class addon. I'll address that in a separate patch.
27 lines
637 B
Python
27 lines
637 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
|
|
|
|
|
|
class Writer:
|
|
def __init__(self, path):
|
|
if path == "-":
|
|
f = sys.stdout
|
|
else:
|
|
f = open(path, "wb")
|
|
self.w = io.FlowWriter(f)
|
|
|
|
def response(self, flow):
|
|
if random.choice([True, False]):
|
|
self.w.add(flow)
|
|
|
|
|
|
addons = [Writer(sys.argv[1])]
|