diff --git a/tasks.py b/tasks.py index 3d13233..3b677ba 100644 --- a/tasks.py +++ b/tasks.py @@ -111,17 +111,14 @@ def get_hosts(hosts: str) -> List[DeployHost]: systems = data["nixosConfigurations"] return [DeployHost(f"{n}.nix-community.org") for n in systems] - if "darwin01" in hosts: - return [ - DeployHost(f"{h}.nix-community.org", user="customer") - for h in hosts.split(",") - ] - - if "darwin02" or "darwin03" in hosts: - return [ - DeployHost(f"{h}.nix-community.org", user="hetzner") - for h in hosts.split(",") - ] + if "darwin" in hosts: + if hosts == "darwin01": + return [DeployHost("darwin01.nix-community.org", user="customer")] + else: + return [ + DeployHost(f"{h}.nix-community.org", user="hetzner") + for h in hosts.split(",") + ] return [DeployHost(f"{h}.nix-community.org") for h in hosts.split(",")]