diff options
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() +] |