diff options
Diffstat (limited to 'examples/simple')
-rw-r--r-- | examples/simple/filter_flows.py | 18 |
1 files changed, 12 insertions, 6 deletions
diff --git a/examples/simple/filter_flows.py b/examples/simple/filter_flows.py index 896fa54a..fd49425a 100644 --- a/examples/simple/filter_flows.py +++ b/examples/simple/filter_flows.py @@ -1,15 +1,21 @@ """ This scripts demonstrates how to use mitmproxy's filter pattern in scripts. -Usage: - mitmdump -s "flowfilter.py FILTER" """ -import sys from mitmproxy import flowfilter +from mitmproxy import ctx class Filter: - def __init__(self, spec): - self.filter = flowfilter.parse(spec) + def __init__(self): + self.filter = None + + def configure(self, updated): + self.filter = flowfilter.parse(ctx.options.flowfilter) + + def load(self, l): + l.add_option( + "flowfilter", str, "", "Check that flow matches filter." + ) def response(self, flow): if flowfilter.match(self.filter, flow): @@ -17,4 +23,4 @@ class Filter: print(flow) -addons = [Filter(sys.argv[1])] +addons = [Filter()] |