Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2026-02-15 07:36:16 +00:00
Code Activity

Portainer refactor availability (#161589)

Browse Source
This commit is contained in:
Erwin Douna
2026-01-26 08:59:25 +01:00
committed by GitHub
parent b18a8791ef
commit 98b7ffcbdd
3 changed files with 14 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
homeassistant/components/portainer/binary_sensor.py
View File

@@ -132,11 +132,6 @@ class PortainerEndpointSensor(PortainerEndpointEntity, BinarySensorEntity):
self._attr_unique_id = f"{coordinator.config_entry.entry_id}_{device_info.id}_{entity_description.key}"
@property
def available(self) -> bool:
"""Return if the device is available."""
return super().available and self.device_id in self.coordinator.data
@property
def is_on(self) -> bool | None:
"""Return true if the binary sensor is on."""
@@ -161,15 +156,6 @@ class PortainerContainerSensor(PortainerContainerEntity, BinarySensorEntity):
self._attr_unique_id = f"{coordinator.config_entry.entry_id}_{self.device_name}_{entity_description.key}"
@property
def available(self) -> bool:
"""Return if the device is available."""
return (
super().available
and self.endpoint_id in self.coordinator.data
and self.device_name in self.coordinator.data[self.endpoint_id].containers
)
@property
def is_on(self) -> bool | None:
"""Return true if the binary sensor is on."""

14
homeassistant/components/portainer/entity.py
View File

@@ -44,6 +44,11 @@ class PortainerEndpointEntity(PortainerCoordinatorEntity):
name=device_info.endpoint.name,
)
@property
def available(self) -> bool:
"""Return if the device is available."""
return super().available and self.device_id in self.coordinator.data
class PortainerContainerEntity(PortainerCoordinatorEntity):
"""Base implementation for Portainer container."""
@@ -84,6 +89,15 @@ class PortainerContainerEntity(PortainerCoordinatorEntity):
translation_key=None if self.device_name else "unknown_container",
)
@property
def available(self) -> bool:
"""Return if the device is available."""
return (
super().available
and self.endpoint_id in self.coordinator.data
and self.device_name in self.coordinator.data[self.endpoint_id].containers
)
@property
def container_data(self) -> PortainerContainerData:
"""Return the coordinator data for this container."""

14
homeassistant/components/portainer/sensor.py
View File

@@ -300,15 +300,6 @@ class PortainerContainerSensor(PortainerContainerEntity, SensorEntity):
self._attr_unique_id = f"{coordinator.config_entry.entry_id}_{self.device_name}_{entity_description.key}"
@property
def available(self) -> bool:
"""Return if the device is available."""
return (
super().available
and self.endpoint_id in self.coordinator.data
and self.device_name in self.coordinator.data[self.endpoint_id].containers
)
@property
def native_value(self) -> StateType:
"""Return the state of the sensor."""
@@ -332,11 +323,6 @@ class PortainerEndpointSensor(PortainerEndpointEntity, SensorEntity):
self._attr_unique_id = f"{coordinator.config_entry.entry_id}_{device_info.id}_{entity_description.key}"
@property
def available(self) -> bool:
"""Return if the device is available."""
return super().available and self.device_id in self.coordinator.data
@property
def native_value(self) -> StateType:
"""Return the state of the sensor."""