mitmproxy/examples/flowwriter.py

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)


def start():
    if len(sys.argv) != 2:
        raise ValueError('Usage: -s "flowriter.py filename"')
    return Writer(sys.argv[1])
add flowwriter example 2015-04-12 01:47:58 +00:00			`import random`
			`import sys`
Move all tools into mitmproxy.tools, move models/* to top level The primary motivation here (and for all the other moving around) is to present a clean "front of house" to library users, and to migrate primary objects to the top of the module hierarchy. 2016-10-19 02:25:39 +00:00			`from mitmproxy import io`
spaces, not tabs 2015-04-12 01:50:14 +00:00
scripts: refactor some examples that keep global state We now have a better way to do this. 2016-07-22 23:57:31 +00:00
			`class Writer:`
			`def __init__(self, path):`
			`if path == "-":`
			`f = sys.stdout`
			`else:`
			`f = open(path, "wb")`
Remove flow module entirely, move contents to top level mitmproxy.flow.io -> mitmproxy.io mitmproxy.flow.export -> mitmproxy.export 2016-10-19 02:08:35 +00:00			`self.w = io.FlowWriter(f)`
scripts: refactor some examples that keep global state We now have a better way to do this. 2016-07-22 23:57:31 +00:00
			`def response(self, flow):`
			`if random.choice([True, False]):`
			`self.w.add(flow)`
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 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"')`
scripts: refactor some examples that keep global state We now have a better way to do this. 2016-07-22 23:57:31 +00:00			`return Writer(sys.argv[1])`