diff options
author | Wojtek Kosior <koszko@koszko.org> | 2022-10-20 21:57:47 +0200 |
---|---|---|
committer | Wojtek Kosior <koszko@koszko.org> | 2022-10-20 21:57:47 +0200 |
commit | 55b95d70b24cfa1b4703ae442a3c6d1781cc95aa (patch) | |
tree | c6a33b401af373d7591e46e21dede0609313a140 /src/hydrilla/proxy/policies/base.py | |
parent | 3beab050c339c51c484af9bcd8248ba8ebbbf4d4 (diff) | |
download | haketilo-hydrilla-55b95d70b24cfa1b4703ae442a3c6d1781cc95aa.tar.gz haketilo-hydrilla-55b95d70b24cfa1b4703ae442a3c6d1781cc95aa.zip |
[proxy] rework internal HTTP headers representation
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', |