diff options
author | Aldo Cortesi <aldo@nullcube.com> | 2016-07-23 11:57:31 +1200 |
---|---|---|
committer | Aldo Cortesi <aldo@nullcube.com> | 2016-07-23 12:24:09 +1200 |
commit | dbafe9f87bb7b793ae2d84e01af5d39f034c78d4 (patch) | |
tree | f19954d23ce454dd381a62c29af86dcd9ed5cf77 /examples/flowwriter.py | |
parent | b5416895f5157d0191c3967850be576450ec2d7d (diff) | |
download | mitmproxy-dbafe9f87bb7b793ae2d84e01af5d39f034c78d4.tar.gz mitmproxy-dbafe9f87bb7b793ae2d84e01af5d39f034c78d4.tar.bz2 mitmproxy-dbafe9f87bb7b793ae2d84e01af5d39f034c78d4.zip |
scripts: refactor some examples that keep global state
We now have a better way to do this.
Diffstat (limited to 'examples/flowwriter.py')
-rw-r--r-- | examples/flowwriter.py | 25 |
1 files changed, 13 insertions, 12 deletions
diff --git a/examples/flowwriter.py b/examples/flowwriter.py index 07c7ca20..df2e5a40 100644 --- a/examples/flowwriter.py +++ b/examples/flowwriter.py @@ -3,20 +3,21 @@ import sys from mitmproxy.flow import FlowWriter -state = {} + +class Writer: + def __init__(self, path): + if path == "-": + f = sys.stdout + else: + f = open(path, "wb") + self.w = 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"') - - 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) + return Writer(sys.argv[1]) |