fixed bug in api key header routine

Author: khultman

gremlinapi/config.py

@@ -9,9 +9,23 @@
 
 
 class GremlinAPIConfig(object):
+
+    def __init__(self):
+        self._api_key = None
+        self._base_uri = None
+        self._bearer_timestamp = None
+        self._bearer_token = None
+        self._company_name = None
+        self._max_bearer_interval = None
+        self._password = None
+        self._team_guid = None
+        self._user = None
+        self._user_mfa_token_value = None
+
     @property
     def api_key(self):
-        """API Key for API authorization"""
+        if not self._api_key:
+            return None
         return self._api_key
 
     @api_key.setter

gremlinapi/http_clients.py

@@ -33,15 +33,20 @@ def api_call(cls, method, uri, *args, **kwargs):
 
     @classmethod
     def header(cls, *args, **kwargs):
-        api_key = kwargs.get('api_key', GremlinAPIConfig.api_key)
-        bearer_token = kwargs.get('bearer_token', GremlinAPIConfig.bearer_token)
+        api_key = kwargs.get('api_key', None)
+        bearer_token = kwargs.get('bearer_token', None)
         header = dict()
+        if not (api_key and bearer_token):
+            if GremlinAPIConfig.bearer_token:
+                bearer_token = GremlinAPIConfig.bearer_token
+            if GremlinAPIConfig.api_key:
+                api_key = GremlinAPIConfig.api_key
         if api_key and not bearer_token:
             if "Key" in api_key:
                 header['Authorization'] = api_key
             else:
                 header['Authorization'] = f'Key {api_key}'
-        if bearer_token:
+        elif bearer_token:
             if "Bearer" in bearer_token:
                 header['Authorization'] = bearer_token
             else: