From a38d19576c387e505cc468b20ca5b8bcf2fa4759 Mon Sep 17 00:00:00 2001 From: Wojtek Kosior Date: Thu, 11 Aug 2022 15:47:13 +0200 Subject: move classes/protocols for representing HTTP requests and responses data into a separate file --- src/hydrilla/proxy/policies/rule.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) (limited to 'src/hydrilla/proxy/policies/rule.py') 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() -- cgit v1.2.3