handle missing credentials sdk2hub

This commit is contained in:
montezdesousa
2023-09-06 12:59:28 +01:00
parent b6d64b44b5
commit d9f41aafca

View File

@@ -231,13 +231,17 @@ class HubService:
@classmethod
def sdk2hub(cls, credentials: Credentials) -> HubUserSettings:
"""Convert SDK models to Hub user settings."""
def get_cred(cred: str) -> Optional[str]:
return getattr(credentials, cred, None)
features_keys = FeaturesKeys(
API_KEY_ALPHAVANTAGE=credentials.alpha_vantage_api_key,
API_FRED_KEY=credentials.fred_api_key,
API_KEY_FINANCIALMODELINGPREP=credentials.fmp_api_key,
API_INTRINIO_KEY=credentials.intrinio_api_key,
API_POLYGON_KEY=credentials.polygon_api_key,
API_KEY_QUANDL=credentials.quandl_api_key,
API_KEY_ALPHAVANTAGE=get_cred("alpha_vantage_api_key"),
API_FRED_KEY=get_cred("fred_api_key"),
API_KEY_FINANCIALMODELINGPREP=get_cred("fmp_api_key"),
API_INTRINIO_KEY=get_cred("intrinio_api_key"),
API_POLYGON_KEY=get_cred("polygon_api_key"),
API_KEY_QUANDL=get_cred("quandl_api_key"),
)
hub_user_settings = HubUserSettings(features_keys=features_keys)
return hub_user_settings