"""Configuration for Renew lambda.""" import os from logging import INFO import sentry_sdk from sentry_sdk.integrations.aws_lambda import AwsLambdaIntegration from sentry_sdk.integrations.logging import LoggingIntegration from secrets_manager.lambda_ext import LambdaSecretsManager from vector_utils.aws_utils import sqs 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}-find-batch-to-close' SCRIPT_NAME = 'lambda-find-batch-to-close' 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 } DATADOG_SCRIPT_NAME = 'find_batch_to_close' DATADOG_ATTEMPT = 'attempt' DATADOG_DB_SUCCESS = 'db_update_success' DATADOG_DB_FAILURE = 'db_update_failure' DATADOG_APP_KEY = datadog_secrets_manager_client.get_cred('DD_APP_KEY') DATADOG_API_KEY = datadog_secrets_manager_client.get_cred('DD_API_KEY') CLOSE_BATCH_QUEUE_NAME = os.environ.get('CLOSE_BATCH_QUEUE_NAME', sqs.BATCH_TO_CLOSE_QUEUE_NAME_PATTERN)