Remove running binary_sensor for HAOS (#60597)

This commit is contained in:
Joakim Sørensen 2021-11-30 19:18:14 +01:00 committed by GitHub
parent 19b4cc7119
commit 153f15c93b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -31,13 +31,16 @@ class HassioBinarySensorEntityDescription(BinarySensorEntityDescription):
target: str | None = None target: str | None = None
ENTITY_DESCRIPTIONS = ( COMMON_ENTITY_DESCRIPTIONS = (
HassioBinarySensorEntityDescription( HassioBinarySensorEntityDescription(
device_class=DEVICE_CLASS_UPDATE, device_class=DEVICE_CLASS_UPDATE,
entity_registry_enabled_default=False, entity_registry_enabled_default=False,
key=ATTR_UPDATE_AVAILABLE, key=ATTR_UPDATE_AVAILABLE,
name="Update Available", name="Update Available",
), ),
)
ADDON_ENTITY_DESCRIPTIONS = COMMON_ENTITY_DESCRIPTIONS + (
HassioBinarySensorEntityDescription( HassioBinarySensorEntityDescription(
device_class=DEVICE_CLASS_RUNNING, device_class=DEVICE_CLASS_RUNNING,
entity_registry_enabled_default=False, entity_registry_enabled_default=False,
@ -58,7 +61,7 @@ async def async_setup_entry(
entities = [] entities = []
for entity_description in ENTITY_DESCRIPTIONS: for entity_description in ADDON_ENTITY_DESCRIPTIONS:
for addon in coordinator.data[DATA_KEY_ADDONS].values(): for addon in coordinator.data[DATA_KEY_ADDONS].values():
entities.append( entities.append(
HassioAddonBinarySensor( HassioAddonBinarySensor(
@ -68,7 +71,8 @@ async def async_setup_entry(
) )
) )
if coordinator.is_hass_os: if coordinator.is_hass_os:
for entity_description in COMMON_ENTITY_DESCRIPTIONS:
entities.append( entities.append(
HassioOSBinarySensor( HassioOSBinarySensor(
coordinator=coordinator, coordinator=coordinator,