fixed compute

Author: eduardomourey

gbmapi/api.py

@@ -1,7 +1,7 @@
 from enum import StrEnum
 from urllib.parse import quote
 
-from .base import GBMBase
+from .base import GBMApiBase
 
 GBM_APP = 'https://app.gbm.com'
 
@@ -16,7 +16,7 @@ class CommodityType(StrEnum):
     MATERIA_PRIMA = "f3851585-1456-4fcd-a6cb-23ef31dd351b"
 
 
-class GBMApi(GBMBase):
+class GBMApi(GBMApiBase):
     API_URL = "https://api.gbm.com"
     ORIGIN = "https://homebroker.gbm.com"
 

gbmapi/auth.py

@@ -52,7 +52,7 @@ def access_token(self):
 
     def identity_token(self):
         cred = self.check_credentials()
-        return cred.get('idToken') or cred['identityToken']
+        return cred['identityToken']
 
     def _request(self, method, path, headers=None, json=None, check_success=True):
         resp = requests.request(

gbmapi/base.py

@@ -5,7 +5,7 @@
 from .utils import STANDARD_HEADERS
 
 
-class GBMBase:
+class GBMApiBase:
     API_URL = None  # "https://api.gbm.com"
     ORIGIN = None  # "https://homebroker.gbm.com"
 

gbmapi/homebroker_api.py

@@ -4,7 +4,7 @@
 from time import time
 
 from . import GBMAuth
-from .base import GBMBase
+from .base import GBMApiBase
 from .utils import format_date, format_date_end
 
 
@@ -34,7 +34,7 @@ class ProcessStatus(IntEnum):
 APP_ID = 16
 
 
-class GBMHomebrokerApi(GBMBase):
+class GBMHomebrokerApi(GBMApiBase):
     API_URL = "https://homebroker-api.gbm.com"
     ORIGIN = "https://homebroker.gbm.com"