"""Configuration for Job Priority Rule Match 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 # required ENVIRONMENT = os.environ['Environment'] ENCODER_IDS = os.environ['ENCODER_IDS'] # Logger configuration. LOGGER_LEVEL = os.environ.get('LOGGER_LEVEL', INFO) LOGGER_DSN = os.environ.get('LOGGER_DSN') sentry_logging = LoggingIntegration( level=LOGGER_LEVEL, event_level=None # Don't send error log as sentry event. ) util.init_sentry_for_lambda(sentry_logging) # Name of script SCRIPT_NAME = 'job-priority-rule-match' # Name of lambda APPLICATION_NAME = 'lambda-vector' LAMBDA_NAME = f'{APPLICATION_NAME}-{SCRIPT_NAME}' REQUEST_RETRY = 5 INITIAL_REQUEST_WAIT_TIME = 1 RETRY_WAIT_EXPONENT = 2 # the number of records to fetch per query to direct delivery for jobs DD_MYSQL_LIMIT = int(os.environ.get('DD_MYSQL_LIMIT') or 0) # initialize the Secrets Manager client secrets_manager_client = LambdaSecretsManager( environment=ENVIRONMENT, service_name=LAMBDA_NAME) datadog_secrets_manager_client = LambdaSecretsManager( environment=ENVIRONMENT, service_name='datadog') DATADOG_APP_KEY = datadog_secrets_manager_client.get_cred('DD_APP_KEY') DATADOG_API_KEY = datadog_secrets_manager_client.get_cred('DD_API_KEY') 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') 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') AR_MYSQL_CONN_INFO = { 'user': AR_MYSQL_USER, 'password': AR_MYSQL_PASSWORD, 'host': AR_MYSQL_HOST, 'port': AR_MYSQL_PORT, 'database': AR_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 }