diff options
author | Wojtek Kosior <koszko@koszko.org> | 2022-10-21 18:29:43 +0200 |
---|---|---|
committer | Wojtek Kosior <koszko@koszko.org> | 2022-10-21 18:29:43 +0200 |
commit | 85a0cacb28b84005d5d034a53973153d49214723 (patch) | |
tree | 3839b70f196d28ef9db21add79be633f3847ffae /src/hydrilla/proxy/http_messages.py | |
parent | 3360fff8b25a60cc1f378b80692d2162e70dc142 (diff) | |
download | haketilo-hydrilla-85a0cacb28b84005d5d034a53973153d49214723.tar.gz haketilo-hydrilla-85a0cacb28b84005d5d034a53973153d49214723.zip |
[proxy] rework CSP manipulation
Diffstat (limited to 'src/hydrilla/proxy/http_messages.py')
-rw-r--r-- | src/hydrilla/proxy/http_messages.py | 10 |
1 files changed, 10 insertions, 0 deletions
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], |