diff options
Diffstat (limited to 'examples/flowbasic')
| -rwxr-xr-x[-rw-r--r--] | examples/flowbasic | 15 |
1 files changed, 6 insertions, 9 deletions
diff --git a/examples/flowbasic b/examples/flowbasic index 4a87b86a..02a65b59 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): + @controller.handler + def request(self, f): f = flow.FlowMaster.handle_request(self, f) - if f: - f.reply() - return f + print(f) - def handle_response(self, f): + @controller.handler + def response(self, f): f = flow.FlowMaster.handle_response(self, f) - if f: - f.reply() print(f) - return f config = ProxyConfig( |