Procházet zdrojové kódy

Expanding on getCSRF to yield errors in Slicer

Andrej před 4 roky
rodič
revize
05c9b2a909
1 změnil soubory, kde provedl 9 přidání a 3 odebrání
  1. 9 3
      labkeyInterface.py

+ 9 - 3
labkeyInterface.py

@@ -139,10 +139,16 @@ class labkeyInterface:
             
     def getCSRF(self):
         url=self.GetLabkeyUrl()+'/login/whoAmI.view'
-        response=self.get(url)
+        try:
+            response=self.get(url)
+            encoding=chardet.detect(response.data)['encoding']
+            jsonData=json.loads(response.data.decode(encoding))
+        except AttributeError:
+            print('Failed')
+            return None
+
         self.cookie=response.getheader('Set-Cookie')
-        encoding=chardet.detect(response.data)['encoding']
-        jsonData=json.loads(response.data.decode(encoding))
+        print('CSRF: {}'.format(jsonData['CSRF']))
         return jsonData["CSRF"]
 
     def getUserId(self):