mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-23 00:01:36 +00:00
17 lines
393 B
Python
17 lines
393 B
Python
|
from libmproxy.flow import FlowWriter
|
||
|
|
import random
|
||
|
|
import sys
|
||
|
|
|
||
|
|
def start(context, argv):
|
||
|
|
if len(argv) != 2:
|
||
|
|
raise ValueError('Usage: -s "flowriter.py filename"')
|
||
|
|
|
||
|
|
if argv[1] == "-":
|
||
|
|
f = sys.stdout
|
||
|
|
else:
|
||
|
|
f = open(argv[1], "wb")
|
||
|
|
context.flow_writer = FlowWriter(f)
|
||
|
|
|
||
|
|
def response(context, flow):
|
||
|
|
if random.choice([True, False]):
|
||
|
|
context.flow_writer.add(flow)
|