"""Configuration for close-batch lambda.""" import os from logging import INFO from secrets_manager.lambda_ext import LambdaSecretsManager import sentry_sdk from sentry_sdk.integrations.aws_lambda import AwsLambdaIntegration from sentry_sdk.integrations.logging import LoggingIntegration ENVIRONMENT = os.environ.get('Environment', 'dev') # Logging LOGGER_LEVEL = os.environ.get('LOGGER_LEVEL', INFO) # Sentry Config SENTRY_DSN = os.environ.get('SENTRY_DSN') or None if SENTRY_DSN: sentry_sdk.init( dsn=SENTRY_DSN, environment=ENVIRONMENT, integrations=[ LoggingIntegration( level=LOGGER_LEVEL, event_level=None # Don't send error log as sentry event. ), AwsLambdaIntegration() ]) # Name of script APPLICATION_NAME = 'lambda-vector' LAMBDA_NAME = f'{APPLICATION_NAME}-close-batch' SCRIPT_NAME = 'lambda-close-batch' secrets_manager_client = LambdaSecretsManager( environment=ENVIRONMENT, service_name=LAMBDA_NAME) datadog_secrets_manager_client = LambdaSecretsManager( environment=ENVIRONMENT, service_name='datadog') DD_MYSQL_USER = secrets_manager_client.get_cred('DD_MYSQL_USER') DD_MYSQL_PASSWORD = secrets_manager_client.get_cred('DD_MYSQL_PASSWORD') DD_MYSQL_HOST = os.environ.get('DD_MYSQL_HOST') DD_MYSQL_PORT = int(os.environ.get('DD_MYSQL_PORT', 3306)) DD_MYSQL_DATABASE = os.environ.get('DD_MYSQL_DATABASE') DD_MYSQL_CONN_INFO = { 'user': DD_MYSQL_USER, 'password': DD_MYSQL_PASSWORD, 'host': DD_MYSQL_HOST, 'port': DD_MYSQL_PORT, 'database': DD_MYSQL_DATABASE, 'connect_timeout': 15 } AR_MYSQL_USER = secrets_manager_client.get_cred('AR_MYSQL_USER') AR_MYSQL_PASSWORD = secrets_manager_client.get_cred('AR_MYSQL_PASSWORD') AR_MYSQL_HOST = os.environ.get('AR_MYSQL_HOST') AR_MYSQL_PORT = int(os.environ.get('AR_MYSQL_PORT', 3306)) AR_MYSQL_DATABASE = os.environ.get('AR_MYSQL_DATABASE') AR_MYSQL_CONN_INFO = { 'user': AR_MYSQL_USER, 'password': AR_MYSQL_PASSWORD, 'host': AR_MYSQL_HOST, 'port': AR_MYSQL_PORT, 'database': AR_MYSQL_DATABASE, 'connect_timeout': 15 } DATADOG_SCRIPT_NAME = 'close_batch' DATADOG_ATTEMPT = 'attempt' DATADOG_TRANSFER_START = 'transfer_start' DATADOG_TRANSFER_SUCCESS = 'transfer_success' DATADOG_TRANSFER_FAIL = 'transfer_fail' DATADOG_APP_KEY = datadog_secrets_manager_client.get_cred('DD_APP_KEY') DATADOG_API_KEY = datadog_secrets_manager_client.get_cred('DD_API_KEY') vector_secrets_manager_client = LambdaSecretsManager( environment=ENVIRONMENT, service_name='direct_delivery') STORES_WITH_DISABLED_ALGORITHMS = os.environ.get( 'STORES_WITH_DISABLED_ALGORITHMS', '569,453').split(',') DISABLE_PUBLISHING_METRICS = os.environ.get('DISABLE_PUBLISHING_METRICS', 'false').lower() == 'true'