import base64 import logging import os from typing import List from urllib import parse from pydantic import BaseModel from starlette.config import Config from starlette.datastructures import CommaSeparatedStrings log = logging.getLogger(__name__) class BaseConfigurationModel(BaseModel): """Base configuration model used by all config options.""" pass def get_env_tags(tag_list: List[str]) -> dict: """Create dictionary of available env tags.""" tags = {} for t in tag_list: tag_key, env_key = t.split(":") env_value = os.environ.get(env_key) if env_value: tags.update({tag_key: env_value}) return tags def get_config(): try: # Try to load from .env file first config = Config(".env") except FileNotFoundError: # If .env doesn't exist, use environment variables config = Config(environ=os.environ) return config config = get_config() LOG_LEVEL = config("LOG_LEVEL", default="INFO") ENV = config("ENV") PORT = config("PORT", cast=int) HOST = config("HOST") # database DATABASE_HOSTNAME = config("DATABASE_HOSTNAME") _DATABASE_CREDENTIAL_USER = config("DATABASE_CREDENTIAL_USER") _DATABASE_CREDENTIAL_PASSWORD = config("DATABASE_CREDENTIAL_PASSWORD") _QUOTED_DATABASE_PASSWORD = parse.quote(str(_DATABASE_CREDENTIAL_PASSWORD)) DATABASE_NAME = config("DATABASE_NAME") DATABASE_PORT = config("DATABASE_PORT") DATABASE_ENGINE_POOL_SIZE = config("DATABASE_ENGINE_POOL_SIZE", cast=int, default=20) DATABASE_ENGINE_MAX_OVERFLOW = config( "DATABASE_ENGINE_MAX_OVERFLOW", cast=int, default=0 ) COLLECTOR_HOSTNAME = config("COLLECTOR_HOSTNAME") COLLECTOR_PORT = config("COLLECTOR_PORT", cast=int) COLLECTOR_CREDENTIAL_USER = config("COLLECTOR_CREDENTIAL_USER") COLLECTOR_CREDENTIAL_PASSWORD = config("COLLECTOR_CREDENTIAL_PASSWORD") QUOTED_COLLECTOR_CREDENTIAL_PASSWORD = parse.quote(str(COLLECTOR_CREDENTIAL_PASSWORD)) COLLECTOR_NAME = config("COLLECTOR_NAME") # Deal w SQLALCHEMY_DATABASE_URI = f"postgresql+asyncpg://{_DATABASE_CREDENTIAL_USER}:{_QUOTED_DATABASE_PASSWORD}@{DATABASE_HOSTNAME}:{DATABASE_PORT}/{DATABASE_NAME}" SQLALCHEMY_COLLECTOR_URI = f"postgresql+asyncpg://{COLLECTOR_CREDENTIAL_USER}:{QUOTED_COLLECTOR_CREDENTIAL_PASSWORD}@{COLLECTOR_HOSTNAME}:{COLLECTOR_PORT}/{COLLECTOR_NAME}" TIMEZONE = config("TIMEZONE") MAXIMO_BASE_URL = config("MAXIMO_BASE_URL") MAXIMO_API_KEY = config("MAXIMO_API_KEY") AUTH_SERVICE_API = config("AUTH_SERVICE_API") REALIBILITY_SERVICE_API = config("REALIBILITY_SERVICE_API") RBD_SERVICE_API = config("RBD_SERVICE_API") TEMPORAL_URL = config("TEMPORAL_URL") BASE_URL = config("BASE_URL") API_KEY = config("API_KEY") TR_RBD_ID = config("TR_RBD_ID") TC_RBD_ID = config("TC_RBD_ID") DEFAULT_TC_ID = config("DEFAULT_TC_ID") RATELIMIT_STORAGE_URI = f"redis://{config('REDIS_HOST')}:{config('REDIS_PORT')}"