"""Configuration for Throttler manager lambda.""" import os from logging import INFO from lambdacommon import util from secrets_manager.lambda_ext import LambdaSecretsManager from sentry_sdk.integrations.logging import LoggingIntegration ENVIRONMENT = os.environ.get('Environment', 'dev') # Logging LOGGER_LEVEL = os.environ.get('LOGGER_LEVEL', INFO) LOGGER_DSN = os.environ.get('LOGGER_DSN') # Name of script SCRIPT_NAME = 'lambda-vector-throttler-manager' INVOCATION_DELAY = float(os.environ.get('INVOCATION_DELAY', 0.2)) secrets_manager_client = LambdaSecretsManager( environment=ENVIRONMENT, service_name=SCRIPT_NAME) datadog_secrets_manager_client = LambdaSecretsManager( environment=ENVIRONMENT, service_name='datadog') sentry_logging = LoggingIntegration( level=LOGGER_LEVEL, event_level=None # Don't send error log as sentry event. ) util.init_sentry_for_lambda(sentry_logging) DD_MYSQL_USER = os.getenv('DD_MYSQL_USER') DD_MYSQL_PASSWORD = secrets_manager_client.get_cred('DD_MYSQL_PASSWORD') DD_MYSQL_HOST = os.getenv('DD_MYSQL_HOST') DD_MYSQL_PORT = int(os.getenv('DD_MYSQL_PORT') or 3306) DD_MYSQL_DATABASE = os.getenv('DD_MYSQL_DATABASE') DATADOG_API_KEY = datadog_secrets_manager_client.get_cred('DD_API_KEY') DATADOG_APP_KEY = datadog_secrets_manager_client.get_cred('DD_APP_KEY') DD_MYSQL_CONN_INFO = { 'user': DD_MYSQL_USER, 'password': DD_MYSQL_PASSWORD, 'host': DD_MYSQL_HOST, 'port': DD_MYSQL_PORT, 'database': DD_MYSQL_DATABASE } EXCLUDED_DMS_LIST = [669, 738, 739]