diff options
author | ColdFire <9128903+fenilgandhi@users.noreply.github.com> | 2018-02-23 00:27:10 +0530 |
---|---|---|
committer | GitHub <noreply@github.com> | 2018-02-23 00:27:10 +0530 |
commit | 777cb98a86f3e8883383935661e2ed20ed9b6bd4 (patch) | |
tree | 67d3e1259dccb83866496f45fe20ac3bd76961ce /examples/addons/options-simple.py | |
parent | 6bf250aa3e8ede84d16170d95728c6488444a5a8 (diff) | |
parent | 58ccad7576e5fd33e937aba58df2f9edc389e52e (diff) | |
download | mitmproxy-777cb98a86f3e8883383935661e2ed20ed9b6bd4.tar.gz mitmproxy-777cb98a86f3e8883383935661e2ed20ed9b6bd4.tar.bz2 mitmproxy-777cb98a86f3e8883383935661e2ed20ed9b6bd4.zip |
Merge branch 'master' into issue-2872
Diffstat (limited to 'examples/addons/options-simple.py')
-rw-r--r-- | examples/addons/options-simple.py | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/examples/addons/options-simple.py b/examples/addons/options-simple.py new file mode 100644 index 00000000..0acefb3f --- /dev/null +++ b/examples/addons/options-simple.py @@ -0,0 +1,24 @@ +from mitmproxy import ctx + + +class AddHeader: + def __init__(self): + self.num = 0 + + def load(self, loader): + loader.add_option( + name = "addheader", + typespec = bool, + default = False, + help = "Add a count header to responses", + ) + + def response(self, flow): + if ctx.options.addheader: + self.num = self.num + 1 + flow.response.headers["count"] = str(self.num) + + +addons = [ + AddHeader() +] |