"""Config file for lambda warmer.""" import os import logging from lambdacommon import util from secrets_manager.lambda_ext import LambdaSecretsManager util.init_sentry_for_lambda() # AWS region configuration DEFAULT_AWS_REGION = 'us-east-1' AWS_REGION = os.environ.get('AWS_REGION', DEFAULT_AWS_REGION) # Environment setup PROD_ENVIRONMENT = 'prod' QA_ENVIRONMENT = 'qa' DEV_ENVIRONMENT = 'dev' ENVIRONMENT = os.environ.get('Environment', DEV_ENVIRONMENT) # Errors and loggers LOGGER_DSN = os.environ.get('LOGGER_DSN') LOGGER_LEVEL = logging.INFO # Name of script SCRIPT_NAME = 'lambda-support-copy-assets' secrets_manager_client = LambdaSecretsManager( environment=ENVIRONMENT, service_name=SCRIPT_NAME) SFTP_USERNAME = os.environ.get('sftp_username') SFTP_PASSWORD = secrets_manager_client.get_cred('SFTP_PASSWORD') AVL_FOLDER_NAME = 'avl' SFTP_USERNAME_FOR_NYC = os.environ.get('sftp_username_for_nyc') SFTP_PASSWORD_FOR_NYC = secrets_manager_client.get_cred( 'SFTP_PASSWORD_FOR_NYC') NYC_FOLDER_NAME = 'nyc' DD_DB_HOST = os.environ.get('dd_db_host') DD_DB_USER = os.environ.get('dd_db_user') DD_DB_PASSWORD = secrets_manager_client.get_cred('DD_DB_PASSWORD') DD_DB = os.environ.get('dd_db', 'direct_delivery') DD_DB_PORT = int(os.environ.get('DD_DB_PORT', 3306)) # SFTP Server SFTP_PORT = int(os.environ.get('SFTP_PORT', 22)) CHUNK_SIZE = int(os.environ.get('CHUNK_SIZE', 6291456)) REMOTE_STORAGE_IP = os.environ.get('remote_storage_ip') REMOTE_STORAGE_IP_FOR_NYC = os.environ.get('remote_storage_ip_for_nyc') PHYSICAL_LOCATION_ID_AVL = 1 PHYSICAL_LOCATION_ID_NYC = 2 DEFAULT_PHYSICAL_LOCATION_ID = int(os.environ.get( 'physical_location_id', PHYSICAL_LOCATION_ID_AVL)) # aws s3 S3 = 's3' REGION_NAME = 'us-east-1' RAW_BUCKET = os.environ.get( 'RAW_BUCKET', '{}-product-support-assets-from-avl'.format(ENVIRONMENT) ) DD_MYSQL_CONN_INFO = { 'user': DD_DB_USER, 'password': DD_DB_PASSWORD, 'host': DD_DB_HOST, 'port': DD_DB_PORT, 'database': DD_DB } AVL_SFTP_INFO = { 'server_ip': REMOTE_STORAGE_IP, 'sftp_username': SFTP_USERNAME, 'sftp_password': SFTP_PASSWORD, 'folder_name': AVL_FOLDER_NAME } NYC_SFTP_INFO = { 'server_ip': REMOTE_STORAGE_IP_FOR_NYC, 'sftp_username': SFTP_USERNAME_FOR_NYC, 'sftp_password': SFTP_PASSWORD_FOR_NYC, 'folder_name': NYC_FOLDER_NAME } # Error reporting. SENTRY_DSN = os.environ.get('SENTRY_DSN')