Merge branch 'master' of https://github.com/mitmproxy/mitmproxy

1 files changed, 20 insertions, 0 deletions

diff --git a/examples/flowwriter.py b/examples/flowwriter.py
new file mode 100644
index 00000000..8fb8cc60
--- /dev/null
+++ b/examples/flowwriter.py
@@ -0,0 +1,20 @@
+import random
+import sys
+
+from mitmproxy.flow import FlowWriter
+
+
+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)