def _ensure_environment()

in src/dubbo/configs.py [0:0]

• 16 lines of code
• 3 McCabe index (conditional complexity)

    def _ensure_environment(environment: Optional[str]) -> str:
        """
        Ensure the environment is valid.
        :param environment: The environment.
        :type environment: Optional[str]
        :return: The environment. If the environment is None, return the default environment.
        :rtype: str
        """
        if not environment:
            return config_constants.PRODUCTION_ENVIRONMENT

        # ignore case
        environment = environment.lower()

        allowed_environments = [
            config_constants.TEST_ENVIRONMENT,
            config_constants.DEVELOPMENT_ENVIRONMENT,
            config_constants.PRODUCTION_ENVIRONMENT,
        ]

        if environment not in allowed_environments:
            raise ValueError(
                f"Unsupported environment: {environment}, "
                f"only support {allowed_environments}, "
                f"default is {config_constants.PRODUCTION_ENVIRONMENT}."
            )

        return environment