move mitmproxy

1 files changed, 20 insertions, 0 deletions

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