From 85a0cacb28b84005d5d034a53973153d49214723 Mon Sep 17 00:00:00 2001 From: Wojtek Kosior Date: Fri, 21 Oct 2022 18:29:43 +0200 Subject: [proxy] rework CSP manipulation --- src/hydrilla/proxy/http_messages.py | 10 ++++++++++ 1 file changed, 10 insertions(+) (limited to 'src/hydrilla/proxy/http_messages.py') diff --git a/src/hydrilla/proxy/http_messages.py b/src/hydrilla/proxy/http_messages.py index 1bed103..9aab510 100644 --- a/src/hydrilla/proxy/http_messages.py +++ b/src/hydrilla/proxy/http_messages.py @@ -195,6 +195,8 @@ class RequestInfo(HasHeadersMixin, _BaseRequestInfoFields): ) -> 'RequestInfo': return BodylessRequestInfo.make(url, method, headers).with_body(body) +AnyRequestInfo = t.Union[BodylessRequestInfo, RequestInfo] + @dc.dataclass(frozen=True) class _BaseResponseInfoFields: @@ -230,6 +232,14 @@ class ResponseInfo(HasHeadersMixin, _BaseResponseInfoFields): bl_info = BodylessResponseInfo.make(url, status_code, headers) return bl_info.with_body(body) +AnyResponseInfo = t.Union[BodylessResponseInfo, ResponseInfo] + + +@dc.dataclass(frozen=True) +class FullHTTPInfo: + request_info: RequestInfo + response_info: ResponseInfo + def is_likely_a_page( request_info: t.Union[BodylessRequestInfo, RequestInfo], -- cgit v1.2.3