68 lines
1.6 KiB
Python
68 lines
1.6 KiB
Python
from clan_cli.inventory import (
|
|
ServiceMeta,
|
|
ServiceSingleDisk,
|
|
ServiceSingleDiskRole,
|
|
ServiceSingleDiskRoleDefault,
|
|
SingleDiskConfig,
|
|
load_inventory_eval,
|
|
load_inventory_json,
|
|
save_inventory,
|
|
)
|
|
|
|
from . import API
|
|
|
|
|
|
def get_instance_name(machine_name: str) -> str:
|
|
return f"{machine_name}-single-disk"
|
|
|
|
|
|
@API.register
|
|
def set_single_disk_uuid(
|
|
base_path: str,
|
|
machine_name: str,
|
|
disk_uuid: str,
|
|
) -> None:
|
|
"""
|
|
Set the disk UUID of single disk machine
|
|
"""
|
|
inventory = load_inventory_json(base_path)
|
|
|
|
instance_name = get_instance_name(machine_name)
|
|
|
|
single_disk_config: ServiceSingleDisk = ServiceSingleDisk(
|
|
meta=ServiceMeta(name=instance_name),
|
|
roles=ServiceSingleDiskRole(
|
|
default=ServiceSingleDiskRoleDefault(
|
|
config=SingleDiskConfig(device=f"/dev/disk/by-id/{disk_uuid}"),
|
|
machines=[machine_name],
|
|
)
|
|
),
|
|
)
|
|
|
|
inventory.services.single_disk[instance_name] = single_disk_config
|
|
|
|
save_inventory(
|
|
inventory,
|
|
base_path,
|
|
f"Set disk UUID: '{disk_uuid}' on machine: '{machine_name}'",
|
|
)
|
|
|
|
|
|
@API.register
|
|
def get_single_disk_uuid(
|
|
base_path: str,
|
|
machine_name: str,
|
|
) -> str | None:
|
|
"""
|
|
Get the disk UUID of single disk machine
|
|
"""
|
|
inventory = load_inventory_eval(base_path)
|
|
|
|
instance_name = get_instance_name(machine_name)
|
|
|
|
single_disk_config: ServiceSingleDisk = inventory.services.single_disk[
|
|
instance_name
|
|
]
|
|
|
|
return single_disk_config.roles.default.config.device
|