diff options
Diffstat (limited to 'src/hydrilla/proxy/policies/base.py')
-rw-r--r-- | src/hydrilla/proxy/policies/base.py | 18 |
1 files changed, 14 insertions, 4 deletions
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.' ) |