yadunut/clan-core
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

API: init set_machine

Browse Source
This commit is contained in:
Johannes Kirschbauer
2024-08-21 15:22:56 +02:00
parent 3a254bee28
commit 9a222d62a7
1 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
pkgs/clan-cli/clan_cli/machines/list.py
View File

@@ -8,12 +8,21 @@ from typing import Literal
from clan_cli.api import API from clan_cli.api import API
from clan_cli.cmd import run_no_stdout from clan_cli.cmd import run_no_stdout
from clan_cli.errors import ClanCmdError, ClanError from clan_cli.errors import ClanCmdError, ClanError
from clan_cli.inventory import Machine, load_inventory_eval from clan_cli.inventory import Machine, load_inventory_eval, save_inventory
from clan_cli.nix import nix_eval, nix_shell from clan_cli.nix import nix_eval, nix_shell
log = logging.getLogger(__name__) log = logging.getLogger(__name__)
@API.register
def set_machine(flake_url: str | Path, machine_name: str, machine: Machine) -> None:
inventory = load_inventory_eval(flake_url)
inventory.machines[machine_name] = machine
save_inventory(inventory, flake_url, "machines: edit '{machine_name}'")
@API.register @API.register
def list_inventory_machines(flake_url: str | Path) -> dict[str, Machine]: def list_inventory_machines(flake_url: str | Path) -> dict[str, Machine]:
inventory = load_inventory_eval(flake_url) inventory = load_inventory_eval(flake_url)
@@ -86,7 +95,7 @@ def check_machine_online(
if not hostname: if not hostname:
raise ClanError(f"Machine {machine_name} does not specify a targetHost") raise ClanError(f"Machine {machine_name} does not specify a targetHost")
timeout = opts.timeout if opts and opts.timeout else 2 timeout = opts.timeout if opts and opts.timeout else 20
cmd = nix_shell( cmd = nix_shell(
["nixpkgs#util-linux", *(["nixpkgs#openssh"] if hostname else [])], ["nixpkgs#util-linux", *(["nixpkgs#openssh"] if hostname else [])],