mitmproxy/examples/flowwriter.py

import random
import sys

from mitmproxy.flow import FlowWriter

state = {}


def start():
    if len(sys.argv) != 2:
        raise ValueError('Usage: -s "flowriter.py filename"')

    if sys.argv[1] == "-":
        f = sys.stdout
    else:
        f = open(sys.argv[1], "wb")
    state["flow_writer"] = FlowWriter(f)


def response(flow):
    if random.choice([True, False]):
        state["flow_writer"].add(flow)
add flowwriter example 2015-04-12 01:47:58 +00:00			`import random`
			`import sys`

fix all libmproxy->mitmproxy references 2016-02-16 19:49:10 +00:00			`from mitmproxy.flow import FlowWriter`
spaces, not tabs 2015-04-12 01:50:14 +00:00
remove context from all scripts 2016-07-08 01:37:33 +00:00			`state = {}`
spaces, not tabs 2015-04-12 01:50:14 +00:00
remove context from all scripts 2016-07-08 01:37:33 +00:00
Add .start for addons and scripts Also improve error messages, fix various unit tests 2016-07-15 01:22:20 +00:00			`def start():`
update examples, tests, docs 2016-06-14 01:17:09 +00:00			`if len(sys.argv) != 2:`
add flowwriter example 2015-04-12 01:47:58 +00:00			`raise ValueError('Usage: -s "flowriter.py filename"')`

update examples, tests, docs 2016-06-14 01:17:09 +00:00			`if sys.argv[1] == "-":`
spaces, not tabs 2015-04-12 01:50:14 +00:00			`f = sys.stdout`
add flowwriter example 2015-04-12 01:47:58 +00:00			`else:`
update examples, tests, docs 2016-06-14 01:17:09 +00:00			`f = open(sys.argv[1], "wb")`
remove context from all scripts 2016-07-08 01:37:33 +00:00			`state["flow_writer"] = FlowWriter(f)`
add flowwriter example 2015-04-12 01:47:58 +00:00
spaces, not tabs 2015-04-12 01:50:14 +00:00
remove context from all scripts 2016-07-08 01:37:33 +00:00			`def response(flow):`
spaces, not tabs 2015-04-12 01:50:14 +00:00			`if random.choice([True, False]):`
remove context from all scripts 2016-07-08 01:37:33 +00:00			`state["flow_writer"].add(flow)`