Update source.py

jhammarstedt · Oct 18, 2022 · 3d6d83c · 3d6d83c
1 parent 77b10a6
commit 3d6d83c
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/airbyte-integrations/connectors/source-lever-hiring/source_lever_hiring/source.py b/airbyte-integrations/connectors/source-lever-hiring/source_lever_hiring/source.py
@@ -12,16 +12,17 @@
 
 
 def _auth_from_config(config):
-    if config and config["credentials"] and 'api_key' in config["credentials"]:
+    if config["auth_type"] == 'Api Key':
         return BasicHttpAuthenticator(username=config["credentials"]["api_key"], password = None, auth_method="Basic")
-    else:
+    elif config["auth_type"] == 'Client':
         return Oauth2Authenticator(
             client_id=config["credentials"]["client_id"],
             client_secret=config["credentials"]["client_secret"],
             refresh_token=config["credentials"]["refresh_token"],
-            token_refresh_endpoint=f"{SourceLeverHiring.URL_MAP_ACCORDING_ENVIRONMENT[config['environment']]['login']}oauth/token",
-    )
-
+            token_refresh_endpoint=f"{SourceLeverHiring.URL_MAP_ACCORDING_ENVIRONMENT[config['environment']]['login']}oauth/token")
+    else:
+        print("Auth type was not configured properly")
+        return None
 
 class SourceLeverHiring(AbstractSource):
     URL_MAP_ACCORDING_ENVIRONMENT = {