Browse Source

Adding getUserId to return the full whoAmI json

Andrej 4 years ago
parent
commit
1a189be18b
1 changed files with 8 additions and 5 deletions
  1. 8 5
      labkeyInterface.py

+ 8 - 5
labkeyInterface.py

@@ -153,11 +153,14 @@ class labkeyInterface:
 
     def getUserId(self):
         url=self.GetLabkeyUrl()+'/login/whoAmI.view'
-        response=self.get(url)
-        self.cookie=response.getheader('Set-Cookie')
-        encoding=chardet.detect(response.data)['encoding']
-        jsonData=json.loads(response.data.decode(encoding))
-        return jsonData["id"]
+        try:
+            response=self.get(url)
+            self.cookie=response.getheader('Set-Cookie')
+            encoding=chardet.detect(response.data)['encoding']
+            jsonData=json.loads(response.data.decode(encoding))
+            return jsonData
+        except AttributeError:
+            return None