"""Application configuration.""" import logging import os from os.path import abspath, dirname, join, pardir from dotenv import load_dotenv from secrets_manager.flask_ext import FlaskSecretsManager # Load environment variables from a .env file if present dotenv_path = abspath(join(dirname(__file__), pardir, '.env')) load_dotenv(dotenv_path) class Config: """Base application configuration. Subclass and override for specific environments. """ # Service information SERVICE_NAME = os.environ.get('SERVICE_NAME') or 'ows-abacus-event' SERVICE_VERSION = '1.0.0' # Production environment PROD_ENVIRONMENT = 'prod' DEV_ENVIRONMENT = 'dev' QA_ENVIRONMENT = 'qa' UAT_ENVIRONMENT = 'uat' ENVIRONMENT = os.environ.get('Environment') or DEV_ENVIRONMENT # Errors and loggers SENTRY = os.environ.get('SENTRY_DSN') or None LOGGER_LEVEL = logging.INFO LOGGER_NAME = 'ows1' ONLY_LOG_ACCESS_ERRORS = False # Generic handlers HEALTH_CHECK = '/hello/' # Swagger documentation path SWAGGER_FILE_PATH = 'spec/ows_abacus_event-1.0.0.yaml' # Database vendor DB_VENDOR = 'mysql' # Mysql database connection MYSQL_DB_USER = os.environ.get('MYSQL_DB_USER') MYSQL_DB_PASS = os.environ.get('MYSQL_DB_PASS') MYSQL_DB_HOST = os.environ.get('MYSQL_DB_HOST') MYSQL_DB_NAME = os.environ.get('MYSQL_DB_NAME', 'royalty_accounting') MYSQL_DB_PORT = os.environ.get('MYSQL_DB_PORT', '3306') MYSQL_DB_PORT = int(MYSQL_DB_PORT) # AWS Airflow AWS_AIRFLOW_ENVIRONMENT_NAME = os.environ.get( 'AWS_AIRFLOW_ENVIRONMENT_NAME', f'{ENVIRONMENT}-abacus-airflow' ) AWS_REGION = os.environ.get('AWS_REGION', 'us-east-1') force_remote = False if Config.ENVIRONMENT == Config.DEV_ENVIRONMENT: force_remote = True secrets_manager_client = FlaskSecretsManager( application_context=False, environment=Config.ENVIRONMENT, service_name=Config.SERVICE_NAME, force_remote=force_remote, ) def get_cred_value(cred_name): """Return secrets value for given cred_name.""" return secrets_manager_client.get_cred(cred_name) if Config.ENVIRONMENT in ( Config.PROD_ENVIRONMENT, Config.QA_ENVIRONMENT, Config.UAT_ENVIRONMENT, ): Config.MYSQL_DB_PASS = get_cred_value('MYSQL_DB_PASS') Config.SENTRY = get_cred_value('SENTRY_DSN')