feat: Detect unauthorized requests

Bisonai · May 18, 2022 · 8f87e44 · 8f87e44
1 parent 0745184
commit 8f87e44
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/pykorbit/exception.py b/pykorbit/exception.py
@@ -8,3 +8,7 @@ class KorbitWebsocketMessageReceiveFailed(Exception):
 
 class KorbitUnexpectedResponse(Exception):
     pass
+
+
+class KorbitUnauthorized(Exception):
+    pass
diff --git a/pykorbit/utils.py b/pykorbit/utils.py
@@ -5,6 +5,8 @@
 from requests.adapters import HTTPAdapter
 from urllib3.util.retry import Retry
 
+from .exception import KorbitUnauthorized
+
 
 def requests_retry_session(
     retries=5,
@@ -33,11 +35,19 @@ def send_post_request(
     headers=None,
     data=None,
 ) -> Union[Dict, List]:
+    """
+    Raises:
+      KorbitUnauthorized
+    """
     resp = requests_retry_session().post(
         url,
         headers=headers,
         data=data,
     )
+
+    if resp.status_code == 401:
+        raise KorbitUnauthorized(resp.text)
+
     return resp.json()
 
 
@@ -46,11 +56,19 @@ def send_get_request(
     headers=None,
     params=None,
 ) -> Union[Dict, List]:
+    """
+    Raises:
+      KorbitUnauthorized
+    """
     resp = requests_retry_session().get(
         url,
         headers=headers,
         params=params,
     )
+
+    if resp.status_code == 401:
+        raise KorbitUnauthorized(resp.text)
+
     return resp.json()