pass http status code downstream in exceptions, related to #34

meeb · Jul 28, 2024 · 1d6abac · 1d6abac
1 parent 333448b
commit 1d6abac
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/whoisit/errors.py b/whoisit/errors.py
@@ -15,8 +15,9 @@ class QueryError(WhoisItError):
         Raised when there are any issues with queries.
     """
 
-    def __init__(self, message, response=''):
+    def __init__(self, message, status_code=0, response=''):
         super().__init__(message)
+        self.status_code = status_code
         self.response = response
 
 

diff --git a/whoisit/query.py b/whoisit/query.py
@@ -250,11 +250,14 @@ def _process_response(self, response):
         if response.status_code in status_code_map:
             error_class, error_message = status_code_map[response.status_code]
             error_content = self._decode_content(response)
-            raise error_class(error_message, response=error_content)
+            raise error_class(error_message,
+                              status_code=response.status_code,
+                              response=error_content)
         elif response.status_code != 200:
             error_content = self._decode_content(response)
             raise QueryError(f'RDAP {self.method} request to {self.url} returned a '
                              f'non-200 status code of {response.status_code}',
+                             status_code=response.status_code,
                              response=error_content)
         try:
             return response.json()