Skip to content

check

zeus.util.check

Tools for checking stuff.

get_env

get_env(name, valtype, default=None)

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

Source code in zeus/util/check.py
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
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 == bool:
            val = os.environ[name].lower()
            if val not in ["true", "false"]:
                raise RuntimeError(
                    f"Strange boolean environment variable value '{val}'"
                )
            return cast(T, val == "true")
        return valtype(os.environ[name])
    except KeyError:
        if default is not None:
            return default
        raise RuntimeError(f"Zeus requires the environment variable '{name}'") from None