Use JUJU_ENV (or jujus.get_env_name()) seems to be a better match as the same source tree may be used against different clouds (caveat: unit_config.yaml needs to be generated again when switching envs).