Skip to content

env

zeus.utils.env

Tools related to environment variables.

get_env

get_env(name, valtype, default=None)

Fetch an environment variable and cast it to the given type.

Source code in zeus/utils/env.py
25
26
27
28
29
30
31
32
33
34
35
36
37
def get_env(name: str, valtype: Type[T], default: T | None = None) -> T:
    """Fetch an environment variable and cast it to the given type."""
    try:
        if valtype is bool:
            val = os.environ[name].lower()
            if val not in ["true", "false"]:
                raise ValueError(f"Strange boolean environment variable value '{val}'")
            return cast(T, val == "true")
        return valtype(os.environ[name])  # type: ignore
    except KeyError:
        if default is not None:
            return default
        raise ValueError(f"Missing environment variable '{name}'") from None