config.py

# Copyright (C) 2022-2023 Indoc Systems
#
# Licensed under the GNU AFFERO GENERAL PUBLIC LICENSE, Version 3.0 (the "License") available at https://www.gnu.org/licenses/agpl-3.0.en.html.
# You may not use this file except in compliance with the License.

import os
from functools import lru_cache
from typing import Any
from typing import Dict
from typing import List

from common import VaultClient
from dotenv import load_dotenv
from pydantic import BaseSettings
from pydantic import Extra

# load env var from local env file for local test
load_dotenv()
SRV_NAMESPACE = os.environ.get('APP_NAME', 'service_cataloguing')
CONFIG_CENTER_ENABLED = os.environ.get('CONFIG_CENTER_ENABLED', 'false')


def load_vault_settings(settings: BaseSettings) -> Dict[str, Any]:
    if CONFIG_CENTER_ENABLED == 'false':
        return {}
    else:
        vc = VaultClient(os.getenv('VAULT_URL'), os.getenv('VAULT_CRT'), os.getenv('VAULT_TOKEN'))
        return vc.get_from_vault(SRV_NAMESPACE)


class Settings(BaseSettings):
    port: int = 5064
    host: str = '0.0.0.0'
    env: str = 'test'
    version: str = '0.1.0'
    namespace: str = ''
    opentelemetry_enabled: bool = False

    ROOT_PATH: str
    ATLAS_API: str
    UTILITY_SERVICE: str
    ATLAS_ADMIN: str
    ATLAS_PASSWD: str

    # the packaged modules
    api_modules: List[str] = ['atlas_api']

    def __init__(self):
        super().__init__()

        self.opentelemetry_enabled = True if self.OPEN_TELEMETRY_ENABLED == 'TRUE' else False
        self.ATLAS_API += '/'
        self.UTILITY_SERVICE += '/v1/'

    class Config:
        env_file = '.env'
        env_file_encoding = 'utf-8'
        extra = Extra.allow

        @classmethod
        def customise_sources(
            cls,
            init_settings,
            env_settings,
            file_secret_settings,
        ):
            return (
                init_settings,
                load_vault_settings,
                env_settings,
                file_secret_settings,
            )


@lru_cache(1)
def get_settings():
    settings = Settings()
    return settings


ConfigClass = Settings()