diff options
author | Wojtek Kosior <koszko@koszko.org> | 2022-10-19 18:55:08 +0200 |
---|---|---|
committer | Wojtek Kosior <koszko@koszko.org> | 2022-10-19 18:55:08 +0200 |
commit | 3beab050c339c51c484af9bcd8248ba8ebbbf4d4 (patch) | |
tree | 0771c5e3c3da18e2a1295954fc25ce3a938c542f /src/hydrilla/proxy/policies/rule.py | |
parent | 94e8af18b60eacfd397b288a7996fc114b853530 (diff) | |
download | haketilo-hydrilla-3beab050c339c51c484af9bcd8248ba8ebbbf4d4.tar.gz haketilo-hydrilla-3beab050c339c51c484af9bcd8248ba8ebbbf4d4.zip |
[proxy] pass all available flow information to relevant methods of Policy type
Diffstat (limited to 'src/hydrilla/proxy/policies/rule.py')
-rw-r--r-- | src/hydrilla/proxy/policies/rule.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/src/hydrilla/proxy/policies/rule.py b/src/hydrilla/proxy/policies/rule.py index b742a64..2e9443e 100644 --- a/src/hydrilla/proxy/policies/rule.py +++ b/src/hydrilla/proxy/policies/rule.py @@ -79,9 +79,11 @@ class BlockPolicy(base.Policy): yield b'Content-Security-Policy', extra_csp.encode() - def consume_response(self, response_info: http_messages.ResponseInfo) \ - -> http_messages.ProducedResponse: - """....""" + def consume_response( + self, + request_info: http_messages.RequestInfo, + response_info: http_messages.ResponseInfo + ) -> http_messages.ProducedResponse: new_response = response_info.make_produced_response() new_headers = self._modify_headers(response_info) |