Merge pull request #52 from drscholly/fix/authv2

Add authv2 token handling
datagalaxy-lab · Oct 28, 2024 · 4aad491 · 4aad491
2 parents bb2a2ce + 0c406e8
commit 4aad491
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/toolbox/api/datagalaxy_api.py b/toolbox/api/datagalaxy_api.py
@@ -20,7 +20,15 @@ def get_client_space_id(self) -> str:
             options={"verify_signature": False},
             algorithms=["HS256"]
         )
-        client_space_id = decode_token["cid"]
+        # old auth
+        if "cid" in decode_token:
+            client_space_id = decode_token["cid"]
+        # authv2
+        elif "dg_client_id" in decode_token:
+            client_space_id = decode_token["dg_client_id"]
+        else:
+            raise Exception('Error while decoding the token')
+
         return client_space_id