Reset token to empty string

mujin · Dec 6, 2024 · a3b4d67 · a3b4d67
1 parent 73798be
commit a3b4d67
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/python/mujinwebstackclient/controllerwebclientraw.py b/python/mujinwebstackclient/controllerwebclientraw.py
@@ -258,5 +258,9 @@ def CallGraphAPI(self, query, variables=None, headers=None, timeout=5.0):
         return content['data']
 
     def UpdateJsonWebToken(self, jsonWebToken):
-        self._session.auth = None
-        self._headers['Authorization'] = 'Bearer ' + jsonWebToken
+        if jsonWebToken == '':
+            self._session.auth = None
+            self._headers['Authorization'] = 'Bearer ' + jsonWebToken
+        else:
+            self._session.auth = requests_auth.HTTPBasicAuth(self._username, self._password)
+            self._headers.pop('Authorization', None)
diff --git a/python/mujinwebstackclient/webstackclient.py b/python/mujinwebstackclient/webstackclient.py
@@ -206,10 +206,12 @@ def Login(self, timeout=5):
         """
         try:
             self.jsonWebToken = self.graphApi.Login(username=self.controllerusername, password=self.controllerpassword, fields={'jsonWebToken': None})['jsonWebToken']
-            self._webclient.UpdateJsonWebToken(self.jsonWebToken)
         except Exception as e:
+            self.jsonWebToken = ''
             log.debug('failed to login through graphql api, use basic HTTP authorization: %s', e)
             self.Ping(timeout=timeout)
+        finally:
+            self._webclient.UpdateJsonWebToken(self.jsonWebToken)
 
     def Ping(self, timeout=5):
         """Sends a dummy HEAD request to api endpoint