fixed compute

Author: eduardomourey

gbmapi/auth.py

@@ -92,7 +92,6 @@ def login(self):
                 "codeChallenge": code_challenge,
                 "codeChallengeMethod": "SHA256",
             },
-            check_success=False
         )['challengeInfo']
 
         # Create a TOTP object
@@ -114,9 +113,9 @@ def login(self):
                 "responseType": "code",
                 "codeChallenge": code_challenge,
                 "codeChallengeMethod": "SHA256"
-            },
-            check_success=True
+            }
         )
+        assert resp['code'] == 0
 
         resp = self._request(
             path="/api/v1/session/token",
@@ -125,9 +124,10 @@ def login(self):
                 "clientId": self.client_id,
                 "codeVerifier": code_verifier,
                 "code": resp['authorizationCode']
-            },
-            check_success=True
+            }
         )
+        assert resp['code'] == 0
+
         return resp
 
     def refresh(self):

gbmapi/base.py

@@ -11,7 +11,7 @@ class GBMApiBase:
     def __init__(self, auth):
         self.auth = auth
 
-    def _request(self, path, method=None, headers=None, json=None, check_success=False, authenticate=True):
+    def _request(self, path, method=None, headers=None, json=None, authenticate=True):
         resp = requests.request(
             method if method else ("GET" if json is None else "POST"),
             self.API_URL + path,
@@ -28,8 +28,4 @@ def _request(self, path, method=None, headers=None, json=None, check_success=Fal
             raise ResponseError(resp)
 
         resp = resp.json()
-        if check_success:
-            assert resp['code'] == 0, resp
-            assert resp['id'] == "Success", resp
-            assert resp['message'] == "Exitoso", resp
         return resp