diff options
Diffstat (limited to 'src/hydrilla/proxy/policies/rule.py')
| -rw-r--r-- | src/hydrilla/proxy/policies/rule.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/src/hydrilla/proxy/policies/rule.py b/src/hydrilla/proxy/policies/rule.py index eb70147..bcb110e 100644 --- a/src/hydrilla/proxy/policies/rule.py +++ b/src/hydrilla/proxy/policies/rule.py @@ -37,6 +37,7 @@ import typing as t from ...url_patterns import ParsedPattern from .. import csp from .. import state +from ..import http_messages from . import base @@ -50,7 +51,7 @@ class BlockPolicy(base.Policy): priority: t.ClassVar[base.PolicyPriority] = base.PolicyPriority._TWO - def _modify_headers(self, response_info: base.ResponseInfo) \ + def _modify_headers(self, response_info: http_messages.ResponseInfo) \ -> t.Iterable[tuple[bytes, bytes]]: """....""" csp_policies = csp.extract(response_info.headers) @@ -80,8 +81,8 @@ class BlockPolicy(base.Policy): yield b'Content-Security-Policy', extra_csp.encode() - def consume_response(self, response_info: base.ResponseInfo) \ - -> base.ProducedResponse: + def consume_response(self, response_info: http_messages.ResponseInfo) \ + -> http_messages.ProducedResponse: """....""" new_response = response_info.make_produced_response() |