From 90dba36f80a5f724c611ecdb1c394917383f8a08 Mon Sep 17 00:00:00 2001 From: Corbeno Date: Thu, 16 Jun 2022 00:35:58 -0500 Subject: [PATCH] Proxmoxve code cleanup (#73571) code cleanup --- .../components/proxmoxve/binary_sensor.py | 27 +++++-------------- 1 file changed, 7 insertions(+), 20 deletions(-) diff --git a/homeassistant/components/proxmoxve/binary_sensor.py b/homeassistant/components/proxmoxve/binary_sensor.py index 780e7240267..97995431778 100644 --- a/homeassistant/components/proxmoxve/binary_sensor.py +++ b/homeassistant/components/proxmoxve/binary_sensor.py @@ -35,31 +35,18 @@ async def async_setup_platform( for node_config in host_config["nodes"]: node_name = node_config["node"] - for vm_id in node_config["vms"]: - coordinator = host_name_coordinators[node_name][vm_id] + for dev_id in node_config["vms"] + node_config["containers"]: + coordinator = host_name_coordinators[node_name][dev_id] - # unfound vm case + # unfound case if (coordinator_data := coordinator.data) is None: continue - vm_name = coordinator_data["name"] - vm_sensor = create_binary_sensor( - coordinator, host_name, node_name, vm_id, vm_name + name = coordinator_data["name"] + sensor = create_binary_sensor( + coordinator, host_name, node_name, dev_id, name ) - sensors.append(vm_sensor) - - for container_id in node_config["containers"]: - coordinator = host_name_coordinators[node_name][container_id] - - # unfound container case - if (coordinator_data := coordinator.data) is None: - continue - - container_name = coordinator_data["name"] - container_sensor = create_binary_sensor( - coordinator, host_name, node_name, container_id, container_name - ) - sensors.append(container_sensor) + sensors.append(sensor) add_entities(sensors)