notify-tls-client 0.1.6__py3-none-any.whl → 2.0.0__py3-none-any.whl

notify_tls_client/tls_client/exceptions.py

@@ -1,3 +1,3 @@
-
-class TLSClientExeption(IOError):
+
+class TLSClientExeption(IOError):
     """General error with the TLS client"""

notify_tls_client/tls_client/response.py

@@ -1,78 +1,84 @@
-from .cookies import cookiejar_from_dict, RequestsCookieJar
-from .structures import CaseInsensitiveDict
-
-from typing import Union
-import json
-
-
-class Response:
-    """object, which contains the response to an HTTP request."""
-
-    def __init__(self):
-
-        # Reference of URL the response is coming from (especially useful with redirects)
-        self.url = None
-
-        # Integer Code of responded HTTP Status, e.g. 404 or 200.
-        self.status_code = None
-
-        # String of responded HTTP Body.
-        self.text = None
-
-        # Case-insensitive Dictionary of Response Headers.
-        self.headers = CaseInsensitiveDict()
-
-        # A CookieJar of Cookies the server sent back.
-        self.cookies = cookiejar_from_dict({})
-        
-        self._content = False
-
-    def __enter__(self):
-        return self
-
-    def __repr__(self):
-        return f"<Response [{self.status_code}]>"
-
-    def json(self, **kwargs):
-        """parse response body to json (dict/list)"""
-        return json.loads(self.text, **kwargs)
-    
-    @property
-    def content(self):
-        """Content of the response, in bytes."""
-        
-        if self._content is False:
-            if self._content_consumed:
-                raise RuntimeError("The content for this response was already consumed")
-
-            if self.status_code == 0:
-                self._content = None
-            else:
-                self._content = b"".join(self.iter_content(10 * 1024)) or b""
-        self._content_consumed = True
-        return self._content
-
-
-def build_response(res: Union[dict, list], res_cookies: RequestsCookieJar) -> Response:
-    """Builds a Response object """
-    response = Response()
-    # Add target / url
-    response.url = res["target"]
-    # Add status code
-    response.status_code = res["status"]
-    # Add headers
-    response_headers = {}
-    if res["headers"] is not None:
-        for header_key, header_value in res["headers"].items():
-            if len(header_value) == 1:
-                response_headers[header_key] = header_value[0]
-            else:
-                response_headers[header_key] = header_value
-    response.headers = response_headers
-    # Add cookies
-    response.cookies = res_cookies
-    # Add response body
-    response.text = res["body"]
-    # Add response content (bytes)
-    response._content = res["body"].encode()
-    return response
+from orjson import orjson
+
+from .cookies import cookiejar_from_dict, RequestsCookieJar
+from .structures import CaseInsensitiveDict
+
+from typing import Union
+import json
+
+
+class Response:
+    """object, which contains the response to an HTTP request."""
+
+    def __init__(self):
+
+        # Reference of URL the response is coming from (especially useful with redirects)
+        self.elapsed = None
+        self.url = None
+
+        # Integer Code of responded HTTP Status, e.g. 404 or 200.
+        self.status_code = None
+
+        # String of responded HTTP Body.
+        self.text = None
+
+        # Case-insensitive Dictionary of Response Headers.
+        self.headers = CaseInsensitiveDict()
+
+        # A CookieJar of Cookies the server sent back.
+        self.cookies = cookiejar_from_dict({})
+
+        self._content = False
+        self._content_consumed = False
+
+    def __enter__(self):
+        return self
+
+    def __repr__(self):
+        return f"<Response [{self.status_code}]>"
+
+    def json(self, **kwargs):
+        """parse response body to json (dict/list)"""
+        return orjson.loads(self.text, **kwargs)
+    
+    @property
+    def content(self):
+        """Content of the response, in bytes."""
+        
+        if self._content is False:
+            if self._content_consumed:
+                raise RuntimeError("The content for this response was already consumed")
+
+            if self.status_code == 0:
+                self._content = None
+            else:
+                self._content = b"".join(self.iter_content(10 * 1024)) or b""
+        self._content_consumed = True
+        return self._content
+
+
+def build_response(res: Union[dict, list], res_cookies: RequestsCookieJar) -> Response:
+    """Builds a Response object """
+    response = Response()
+    # Add target / url
+    response.url = res.get("target", None)
+    # Add status code
+    response.status_code = res.get("status", 0)
+
+    response.elapsed = res.get("elapsed", None)
+    # Add headers
+    response_headers = {}
+    if res.get('headers', None) is not None:
+        for header_key, header_value in res["headers"].items():
+            if len(header_value) == 1:
+                response_headers[header_key] = header_value[0]
+            else:
+                response_headers[header_key] = header_value
+    response.headers = response_headers
+    # Add cookies
+    response.cookies = None
+    # Add response body
+    response.text = res["body"]
+    # Add response content (bytes)
+    response._content = res["body"].encode()
+    return response