"""Config file for sme-analytics-ingestion-flow-control script.""" from logging import INFO import os import uuid from owslogger import logger PROD_ENVIRONMENT = 'prod' DEV_ENVIRONMENT = 'dev' QA_ENVIRONMENT = 'qa' TEST_ENVIRONMENT = 'test' ENVIRONMENT = os.environ.get('ENVIRONMENT', DEV_ENVIRONMENT).lower() MAX_STATE_MACHINES = os.environ.get('MAX_STATE_MACHINES') SFN_START_DELAY = os.environ.get('SFN_START_DELAY') SFN_BATCH_SIZE = os.environ.get('SFN_BATCH_SIZE') # Application Metadata SCRIPT_NAME = 'grps-ingester-ingestion-flow-controller' SERVICE_NAME = 'grps-ingester' LAMBDA_NAME = f'grps-ingester-flow-control-{ENVIRONMENT}' #noqa APP_VERSION = os.environ.get('APP_VERSION', '1.0') # Logger DSN ---------------------------------------------------------------- LOGGER_LEVEL = os.environ.get('LOGGER_LEVEL', INFO) or None # Global logger app_logger = logger.setup( ENVIRONMENT, SCRIPT_NAME, LOGGER_LEVEL, SCRIPT_NAME, APP_VERSION) # Logger with def get_current_logger(correlation_id=None): """Get logger with a correlation_id attached. Args: correlation_id: correlation_id that will be sent along with log record Returns: logger.OwsLoggingAdapter: instance of OwsLoggingAdapter with correlation_id attached. """ return logger.OwsLoggingAdapter( app_logger, {'correlation_id': correlation_id or str(uuid.uuid4())})