"""Support for sensors.""" from __future__ import annotations from fjaraskupan import Device from homeassistant.components.sensor import ( SensorDeviceClass, SensorEntity, SensorStateClass, ) from homeassistant.config_entries import ConfigEntry from homeassistant.const import SIGNAL_STRENGTH_DECIBELS_MILLIWATT from homeassistant.core import HomeAssistant from homeassistant.helpers.entity import DeviceInfo, Entity, EntityCategory from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.typing import StateType from homeassistant.helpers.update_coordinator import CoordinatorEntity from . import Coordinator, async_setup_entry_platform async def async_setup_entry( hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback, ) -> None: """Set up sensors dynamically through discovery.""" def _constructor(coordinator: Coordinator) -> list[Entity]: return [RssiSensor(coordinator, coordinator.device, coordinator.device_info)] async_setup_entry_platform(hass, config_entry, async_add_entities, _constructor) class RssiSensor(CoordinatorEntity[Coordinator], SensorEntity): """Sensor device.""" _attr_has_entity_name = True def __init__( self, coordinator: Coordinator, device: Device, device_info: DeviceInfo, ) -> None: """Init sensor.""" super().__init__(coordinator) self._attr_unique_id = f"{device.address}-signal-strength" self._attr_device_info = device_info self._attr_name = "Signal strength" self._attr_device_class = SensorDeviceClass.SIGNAL_STRENGTH self._attr_state_class = SensorStateClass.MEASUREMENT self._attr_native_unit_of_measurement = SIGNAL_STRENGTH_DECIBELS_MILLIWATT self._attr_entity_registry_enabled_default = False self._attr_entity_category = EntityCategory.DIAGNOSTIC @property def native_value(self) -> StateType: """Return the value reported by the sensor.""" if data := self.coordinator.data: return data.rssi return None