sh_cpl/tools/set_pip_urls/application.py

import json
import os
import sys
from typing import Optional

from cpl_core.environment import EnvironmentNameEnum

from cpl_core.application import ApplicationABC
from cpl_core.configuration import ConfigurationABC
from cpl_core.console import Console
from cpl_core.dependency_injection import ServiceProviderABC
from set_pip_urls.pip_settings import PIPSettings


class Application(ApplicationABC):

    def __init__(self, config: ConfigurationABC, services: ServiceProviderABC):
        ApplicationABC.__init__(self, config, services)

        self._pip_settings: Optional[PIPSettings] = config.get_configuration(PIPSettings)

    def configure(self):
        self._configuration.parse_console_arguments(self._services)

    def main(self):
        if self._pip_settings is None:
            Console.error('appsettings.json not found')
            sys.exit()

        url = None
        match self._environment.environment_name:
            case EnvironmentNameEnum.production.value:
                url = self._pip_settings.production
            case EnvironmentNameEnum.staging.value:
                url = self._pip_settings.staging
            case EnvironmentNameEnum.development.value:
                url = self._pip_settings.development

        cli_json = {
            "CLI": {
                "PipPath": url
            }
        }
        file = os.path.abspath(os.path.join(self._environment.working_directory, '../../src/cpl_cli', 'appsettings.json'))
        Console.write_line(f'Writing PipPath: {url} to {file}')
        with open(file, 'w') as f:
            f.write(json.dumps(cli_json, indent=2))
            f.close()