feat/delete shared key if not valid

Author: atadj

src/sumo/wrapper/_decorators.py

@@ -1,14 +1,19 @@
 # For sphinx:
 from functools import wraps
+from sumo.wrapper._auth_provider import AuthProviderSumoToken
 
 
 def raise_for_status(func):
     @wraps(func)
-    def wrapper(*args, **kwargs):
+    def wrapper(self, *args, **kwargs):
         # FIXME: in newer versions of httpx, raise_for_status() is chainable,
         # so we could simply write
         # return func(*args, **kwargs).raise_for_status()
-        response = func(*args, **kwargs)
+        response = func(self, *args, **kwargs)
+        if response.status_code == 401 and isinstance(
+            self.auth, AuthProviderSumoToken
+        ):
+            self._handle_invalid_shared_key()
         response.raise_for_status()
         return response
 
@@ -17,11 +22,15 @@ def wrapper(*args, **kwargs):
 
 def raise_for_status_async(func):
     @wraps(func)
-    async def wrapper(*args, **kwargs):
+    async def wrapper(self, *args, **kwargs):
         # FIXME: in newer versions of httpx, raise_for_status() is chainable,
         # so we could simply write
         # return func(*args, **kwargs).raise_for_status()
-        response = await func(*args, **kwargs)
+        response = await func(self, *args, **kwargs)
+        if response.status_code == 401 and isinstance(
+            self.auth, AuthProviderSumoToken
+        ):
+            self._handle_invalid_shared_key()
         response.raise_for_status()
         return response
 

src/sumo/wrapper/sumo_client.py

@@ -208,6 +208,16 @@ def blob_client(self) -> BlobClient:
             self._retry_strategy,
         )
 
+    def _handle_invalid_shared_key(self):
+        """Handle the invalid shared key by deleting it."""
+        self.auth.delete_token()
+        logger.error(
+            "Invalid shared key detected and deleted, run again to reset automatically"
+        )
+        print(
+            "Invalid shared key detected and deleted, run again to reset automatically"
+        )
+
     @raise_for_status
     def get(self, path: str, params: Optional[Dict] = None) -> httpx.Response:
         """Performs a GET-request to the Sumo API.