tux.utils.env ¶

def get(self, key: str, default: T | None = None, required: bool = False) -> T | None:
    """
    Get environment variable with type conversion.

    Parameters
    ----------
    key : str
        Environment variable name
    default : Optional[T]
        Default value if not found
    required : bool
        Whether this variable is required

    Returns
    -------
    Optional[T]
        The value of the environment variable

    Raises
    ------
    ConfigurationError
        If variable is required but not found
    """
    value = os.environ.get(key)

    if value is None:
        if required:
            error_msg = f"Required environment variable {key} is not set"
            raise ConfigurationError(error_msg)
        return default

    # If default is provided, attempt to cast to the same type
    if default is not None:
        try:
            if isinstance(default, bool):
                return value.lower() in ("true", "yes", "1", "y")  # type: ignore
            return type(default)(value)  # type: ignore
        except ValueError as e:
            if required:
                error_msg = f"Environment variable {key} is not a valid {type(default).__name__}"
                raise ConfigurationError(error_msg) from e
            return default

    return value  # type: ignore

def set(self, key: str, value: Any, persist: bool = False) -> None:
    """
    Set environment variable.

    Parameters
    ----------
    key : str
        Environment variable name
    value : Any
        Value to set
    persist : bool
        Whether to persist to .env file
    """
    os.environ[key] = str(value)

    if persist and self.dotenv_path.exists():
        set_key(self.dotenv_path, key, str(value))

def _get_env_specific_value(self, env: Environment, dev_key: str, prod_key: str, value_name: str) -> str:
    """
    Get environment-specific configuration value.

    Parameters
    ----------
    env : Environment
        The environment to get value for
    dev_key : str
        Environment variable key for development
    prod_key : str
        Environment variable key for production
    value_name : str
        Human-readable name for error messages

    Returns
    -------
    str
        Configuration value

    Raises
    ------
    ConfigurationError
        If value is not configured for environment
    """
    key = dev_key if env.is_dev else prod_key
    value = self.get(key)  # Don't provide a default value

    if value is None:
        error_msg = f"No {value_name} found for the {env.value.upper()} environment."
        raise ConfigurationError(error_msg)

    return value

def get_database_url(self, env: Environment) -> str:
    """
    Get database URL for specified environment.

    Parameters
    ----------
    env : Environment
        The environment to get URL for

    Returns
    -------
    str
        Database URL

    Raises
    ------
    ConfigurationError
        If database URL is not configured for environment
    """
    return self._get_env_specific_value(env, "DEV_DATABASE_URL", "PROD_DATABASE_URL", "database URL")

def get_bot_token(self, env: Environment) -> str:
    """
    Get bot token for specified environment.

    Parameters
    ----------
    env : Environment
        The environment to get token for

    Returns
    -------
    str
        Bot token

    Raises
    ------
    ConfigurationError
        If bot token is not configured for environment
    """
    return self._get_env_specific_value(env, "DEV_BOT_TOKEN", "PROD_BOT_TOKEN", "bot token")

@classmethod
def reset_for_testing(cls) -> None:
    """Reset the singleton instance for testing purposes."""
    cls._instance = None

def __new__(cls, *args: Any, **kwargs: Any) -> "EnvironmentManager":
    """Ensure singleton pattern."""
    if cls._instance is None:
        cls._instance = super().__new__(cls)
    return cls._instance

def configure(self, environment: Environment) -> None:
    """
    Configure the environment mode.

    Parameters
    ----------
    environment : Environment
        The environment mode to set (DEVELOPMENT or PRODUCTION)
    """
    self.environment = environment