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/payload.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/payload.py')
-rw-r--r-- | src/hydrilla/proxy/policies/payload.py | 17 |
1 files changed, 11 insertions, 6 deletions
diff --git a/src/hydrilla/proxy/policies/payload.py b/src/hydrilla/proxy/policies/payload.py index d3e8e30..5b71af7 100644 --- a/src/hydrilla/proxy/policies/payload.py +++ b/src/hydrilla/proxy/policies/payload.py @@ -257,9 +257,11 @@ class PayloadInjectPolicy(PayloadAwarePolicy): return dc.replace(new_response, body=new_data) - 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: try: return self._consume_response_unsafe(response_info) except Exception as e: @@ -286,15 +288,18 @@ class AutoPayloadInjectPolicy(PayloadInjectPolicy): """....""" priority: t.ClassVar[base.PolicyPriority] = base.PolicyPriority._ONE - 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: try: if self.payload_data.ref.has_problems(): raise _PayloadHasProblemsError() self.payload_data.ref.ensure_items_installed() - return super().consume_response(response_info) + return super().consume_response(request_info, response_info) except (state.RepoCommunicationError, state.FileInstallationError, _PayloadHasProblemsError) as ex: extra_params: dict[str, str] = { |