"""Config file for GFK lambda function.""" import os from secrets_manager.lambda_ext import LambdaSecretsManager APPLICATION_NAME = 'lambda-looker-gfk' # AWS region configuration DEFAULT_AWS_REGION = 'us-east-1' AWS_REGION = os.environ.get('AWS_REGION', DEFAULT_AWS_REGION) # Environment setup QA_ENVIRONMENT = 'qa' PROD_ENVIRONMENT = 'prod' ENVIRONMENT = os.environ.get('Environment', QA_ENVIRONMENT) if ENVIRONMENT in [PROD_ENVIRONMENT, QA_ENVIRONMENT]: secrets_manager_client = LambdaSecretsManager( environment=ENVIRONMENT, service_name=APPLICATION_NAME) def get_(secret, env=ENVIRONMENT): """Get secret by its name and env (optional).""" if env in [PROD_ENVIRONMENT, QA_ENVIRONMENT]: return secrets_manager_client.get_cred(secret) return None FTP_CREDENTIALS = dict( host=os.environ.get('GFK_FTP_HOST'), remote_ftp_dir=os.environ.get('GFK_FTP_PATH') )