diff options
Diffstat (limited to 'examples/flowbasic')
| -rwxr-xr-x[-rw-r--r--] | examples/flowbasic | 19 |
1 files changed, 8 insertions, 11 deletions
diff --git a/examples/flowbasic b/examples/flowbasic index 4a87b86a..74af4e08 100644..100755 --- a/examples/flowbasic +++ b/examples/flowbasic @@ -8,7 +8,7 @@ Note that request and response messages are not automatically replied to, so we need to implement handlers to do this. """ -from mitmproxy import flow +from mitmproxy import flow, controller from mitmproxy.proxy import ProxyServer, ProxyConfig @@ -19,18 +19,15 @@ class MyMaster(flow.FlowMaster): except KeyboardInterrupt: self.shutdown() - def handle_request(self, f): - f = flow.FlowMaster.handle_request(self, f) - if f: - f.reply() - return f + @controller.handler + def request(self, f): + f = flow.FlowMaster.request(self, f) + print(f) - def handle_response(self, f): - f = flow.FlowMaster.handle_response(self, f) - if f: - f.reply() + @controller.handler + def response(self, f): + f = flow.FlowMaster.response(self, f) print(f) - return f config = ProxyConfig( |