From 3beab050c339c51c484af9bcd8248ba8ebbbf4d4 Mon Sep 17 00:00:00 2001 From: Wojtek Kosior Date: Wed, 19 Oct 2022 18:55:08 +0200 Subject: [proxy] pass all available flow information to relevant methods of Policy type --- src/hydrilla/proxy/policies/base.py | 18 ++++++++++++++---- 1 file changed, 14 insertions(+), 4 deletions(-) (limited to 'src/hydrilla/proxy/policies/base.py') diff --git a/src/hydrilla/proxy/policies/base.py b/src/hydrilla/proxy/policies/base.py index b3d3172..fcdbf9d 100644 --- a/src/hydrilla/proxy/policies/base.py +++ b/src/hydrilla/proxy/policies/base.py @@ -61,10 +61,17 @@ class Policy(ABC): priority: t.ClassVar[PolicyPriority] - def should_process_request(self, parsed_url: ParsedUrl) -> bool: + def should_process_request( + self, + request_info: http_messages.BodylessRequestInfo + ) -> bool: return self._process_request - def should_process_response(self, parsed_url: ParsedUrl) -> bool: + def should_process_response( + self, + request_info: http_messages.RequestInfo, + response_info: http_messages.BodylessResponseInfo + ) -> bool: return self._process_response def consume_request(self, request_info: http_messages.RequestInfo) \ @@ -73,8 +80,11 @@ class Policy(ABC): 'This kind of policy does not consume requests.' ) - def consume_response(self, response_info: http_messages.ResponseInfo) \ - -> t.Optional[http_messages.ProducedResponse]: + def consume_response( + self, + request_info: http_messages.RequestInfo, + response_info: http_messages.ResponseInfo + ) -> t.Optional[http_messages.ProducedResponse]: raise NotImplementedError( 'This kind of policy does not consume responses.' ) -- cgit v1.2.3