Address late review for Ping (#112123)

* Address late review for Ping

* Fix import
This commit is contained in:
Jan-Philipp Benecke 2024-03-03 18:10:22 +01:00 committed by GitHub
parent a37c82e34c
commit 911cf60678
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 7 additions and 3 deletions

View File

@ -95,7 +95,7 @@ class PingBinarySensor(PingEntity, BinarySensorEntity):
self, config_entry: ConfigEntry, coordinator: PingUpdateCoordinator
) -> None:
"""Initialize the Ping Binary sensor."""
super().__init__(coordinator, config_entry.entry_id)
super().__init__(config_entry, coordinator, config_entry.entry_id)
# if this was imported just enable it when it was enabled before
if CONF_IMPORTED_BY in config_entry.data:

View File

@ -1,4 +1,5 @@
"""Base entity for the Ping component."""
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import DOMAIN
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@ -13,6 +14,7 @@ class PingEntity(CoordinatorEntity[PingUpdateCoordinator]):
def __init__(
self,
config_entry: ConfigEntry,
coordinator: PingUpdateCoordinator,
unique_id: str,
) -> None:
@ -21,6 +23,6 @@ class PingEntity(CoordinatorEntity[PingUpdateCoordinator]):
self._attr_unique_id = unique_id
self._attr_device_info = DeviceInfo(
identifiers={(DOMAIN, self.coordinator.data.ip_address)},
identifiers={(DOMAIN, config_entry.entry_id)},
manufacturer="Ping",
)

View File

@ -101,7 +101,9 @@ class PingSensor(PingEntity, SensorEntity):
coordinator: PingUpdateCoordinator,
) -> None:
"""Initialize the sensor."""
super().__init__(coordinator, f"{config_entry.entry_id}-{description.key}")
super().__init__(
config_entry, coordinator, f"{config_entry.entry_id}-{description.key}"
)
self.entity_description = description