diff options
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], |