diff options
Diffstat (limited to 'src/hydrilla/proxy/policies/base.py')
| -rw-r--r-- | src/hydrilla/proxy/policies/base.py | 15 |
1 files changed, 10 insertions, 5 deletions
diff --git a/src/hydrilla/proxy/policies/base.py b/src/hydrilla/proxy/policies/base.py index fcdbf9d..8ea792f 100644 --- a/src/hydrilla/proxy/policies/base.py +++ b/src/hydrilla/proxy/policies/base.py @@ -48,9 +48,9 @@ class PolicyPriority(int, enum.Enum): _TWO = 2 _THREE = 3 -ProducedMessage = t.Union[ - http_messages.ProducedRequest, - http_messages.ProducedResponse +MessageInfo = t.Union[ + http_messages.RequestInfo, + http_messages.ResponseInfo ] class Policy(ABC): @@ -75,7 +75,7 @@ class Policy(ABC): return self._process_response def consume_request(self, request_info: http_messages.RequestInfo) \ - -> t.Optional[ProducedMessage]: + -> t.Optional[MessageInfo]: raise NotImplementedError( 'This kind of policy does not consume requests.' ) @@ -84,7 +84,7 @@ class Policy(ABC): self, request_info: http_messages.RequestInfo, response_info: http_messages.ResponseInfo - ) -> t.Optional[http_messages.ProducedResponse]: + ) -> t.Optional[http_messages.ResponseInfo]: raise NotImplementedError( 'This kind of policy does not consume responses.' ) @@ -109,6 +109,11 @@ class PolicyFactory(ABC): sorting_keys.get(other.__class__.__name__, 999) sorting_order = ( + 'WebUIMainPolicyFactory', + 'WebUILandingPolicyFactory', + + 'MitmItPagePolicyFactory', + 'PayloadResourcePolicyFactory', 'PayloadPolicyFactory', |