소스 검색

Expanding on getCSRF to yield errors in Slicer

Andrej 4 년 전
부모
커밋
05c9b2a909
1개의 변경된 파일9개의 추가작업 그리고 3개의 파일을 삭제
  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):