sh_cpl/src/sh_edraft/configuration/configuration.py

import json
import os
import sys

from sh_edraft.configuration.base.configuration_model_base import ConfigurationModelBase
from sh_edraft.configuration.base.configuration_base import ConfigurationBase
from sh_edraft.configuration.model.configuration_variable_name import ConfigurationVariableName
from sh_edraft.environment.base.environment_base import EnvironmentBase
from sh_edraft.environment.hosting_environment import HostingEnvironment
from sh_edraft.environment.model.environment_name import EnvironmentName
from sh_edraft.console.console import Console
from sh_edraft.console.model import ForegroundColor


class Configuration(ConfigurationBase):

    def __init__(self):
        ConfigurationBase.__init__(self)

        self._hosting_environment = HostingEnvironment()
        self._config: dict[type, ConfigurationModelBase] = {}

    @property
    def environment(self) -> EnvironmentBase:
        return self._hosting_environment

    @staticmethod
    def _print_info(name: str, message: str):
        Console.set_foreground_color(ForegroundColor.green)
        Console.write_line(f'[{name}] {message}')
        Console.set_foreground_color(ForegroundColor.default)

    @staticmethod
    def _print_warn(name: str, message: str):
        Console.set_foreground_color(ForegroundColor.yellow)
        Console.write_line(f'[{name}] {message}')
        Console.set_foreground_color(ForegroundColor.default)

    @staticmethod
    def _print_error(name: str, message: str):
        Console.set_foreground_color(ForegroundColor.red)
        Console.write_line(f'[{name}] {message}')
        Console.set_foreground_color(ForegroundColor.default)

    def _set_variable(self, name: str, value: str):
        if name == ConfigurationVariableName.environment.value:
            self._hosting_environment.environment_name = EnvironmentName(value)

        elif name == ConfigurationVariableName.name.value:
            self._hosting_environment.application_name = value

        elif name == ConfigurationVariableName.customer.value:
            self._hosting_environment.customer = value

    def add_environment_variables(self, prefix: str):
        for variable in ConfigurationVariableName.to_list():
            var_name = f'{prefix}{variable}'
            if var_name in [key.upper() for key in os.environ.keys()]:
                self._set_variable(variable, os.environ[var_name])

    def add_argument_variables(self):
        for arg in sys.argv[1:]:
            try:
                argument = arg.split('--')[1].split('=')[0].upper()
                value = arg.split('=')[1]

                if argument not in ConfigurationVariableName.to_list():
                    raise Exception(f'Invalid argument name: {argument}')

                self._set_variable(argument, value)
            except Exception as e:
                self._print_error(__name__, f'Invalid argument: {arg} -> {e}')
                exit()

    def add_json_file(self, name: str, optional: bool = None):
        if self._hosting_environment.content_root_path.endswith('/') and not name.startswith('/'):
            file_path = f'{self._hosting_environment.content_root_path}{name}'
        else:
            file_path = f'{self._hosting_environment.content_root_path}/{name}'

        if not os.path.isfile(file_path):
            if not optional:
                self._print_error(__name__, f'File not found: {file_path}')
                exit()

            self._print_warn(__name__, f'Not Loaded config file: {file_path}')
            return None

        config_from_file = self._load_json_file(file_path)
        for sub in ConfigurationModelBase.__subclasses__():
            for key, value in config_from_file.items():
                if sub.__name__ == key:
                    configuration = sub()
                    configuration.from_dict(value)
                    self.add_configuration(sub, configuration)

    def _load_json_file(self, file: str) -> dict:
        try:
            # open config file, create if not exists
            with open(file, encoding='utf-8') as cfg:
                # load json
                json_cfg = json.load(cfg)
                self._print_info(__name__, f'Loaded config file: {file}')
                return json_cfg
        except Exception as e:
            self._print_error(__name__, f'Cannot load config file: {file}! -> {e}')
            return {}

    def add_configuration(self, key_type: type, value: ConfigurationModelBase):
        self._config[key_type] = value

    def get_configuration(self, search_type: type) -> ConfigurationModelBase:
        if search_type not in self._config:
            raise Exception(f'Config model by type {search_type} not found')

        for config_model in self._config:
            if config_model == search_type:
                return self._config[config_model]

    def create(self):
        pass