core/homeassistant/components/netgear_lte/entity.py

55 lines
1.9 KiB
Python

"""Entity representing a Netgear LTE entity."""
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_HOST
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.helpers.entity import Entity, EntityDescription
from . import ModemData
from .const import DISPATCHER_NETGEAR_LTE, DOMAIN, MANUFACTURER
class LTEEntity(Entity):
"""Base LTE entity."""
_attr_has_entity_name = True
_attr_should_poll = False
def __init__(
self,
config_entry: ConfigEntry,
modem_data: ModemData,
description: EntityDescription,
) -> None:
"""Initialize a Netgear LTE entity."""
self.entity_description = description
self.modem_data = modem_data
self._attr_unique_id = f"{description.key}_{modem_data.data.serial_number}"
self._attr_device_info = DeviceInfo(
configuration_url=f"http://{config_entry.data[CONF_HOST]}",
identifiers={(DOMAIN, modem_data.data.serial_number)},
manufacturer=MANUFACTURER,
model=modem_data.data.items["general.model"],
serial_number=modem_data.data.serial_number,
sw_version=modem_data.data.items["general.fwversion"],
hw_version=modem_data.data.items["general.hwversion"],
)
async def async_added_to_hass(self) -> None:
"""Register callback."""
self.async_on_remove(
async_dispatcher_connect(
self.hass, DISPATCHER_NETGEAR_LTE, self.async_write_ha_state
)
)
async def async_update(self) -> None:
"""Force update of state."""
await self.modem_data.async_update()
@property
def available(self) -> bool:
"""Return the availability of the sensor."""
return self.modem_data.data is not None