110 lines
3.3 KiB
Python
110 lines
3.3 KiB
Python
"""Sensor for PG LAB Electronics."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from pypglab.const import SENSOR_REBOOT_TIME, SENSOR_TEMPERATURE, SENSOR_VOLTAGE
|
|
from pypglab.device import Device as PyPGLabDevice
|
|
|
|
from homeassistant.components.sensor import (
|
|
SensorDeviceClass,
|
|
SensorEntity,
|
|
SensorEntityDescription,
|
|
SensorStateClass,
|
|
)
|
|
from homeassistant.const import Platform, UnitOfElectricPotential, UnitOfTemperature
|
|
from homeassistant.core import HomeAssistant, callback
|
|
from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
|
|
|
|
from . import PGLabConfigEntry
|
|
from .coordinator import PGLabSensorsCoordinator
|
|
from .discovery import PGLabDiscovery
|
|
from .entity import PGLabSensorEntity
|
|
|
|
PARALLEL_UPDATES = 0
|
|
|
|
SENSOR_INFO: list[SensorEntityDescription] = [
|
|
SensorEntityDescription(
|
|
key=SENSOR_TEMPERATURE,
|
|
native_unit_of_measurement=UnitOfTemperature.CELSIUS,
|
|
device_class=SensorDeviceClass.TEMPERATURE,
|
|
state_class=SensorStateClass.MEASUREMENT,
|
|
),
|
|
SensorEntityDescription(
|
|
key=SENSOR_VOLTAGE,
|
|
translation_key="mpu_voltage",
|
|
native_unit_of_measurement=UnitOfElectricPotential.VOLT,
|
|
device_class=SensorDeviceClass.VOLTAGE,
|
|
state_class=SensorStateClass.MEASUREMENT,
|
|
),
|
|
SensorEntityDescription(
|
|
key=SENSOR_REBOOT_TIME,
|
|
translation_key="runtime",
|
|
device_class=SensorDeviceClass.TIMESTAMP,
|
|
icon="mdi:progress-clock",
|
|
),
|
|
]
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant,
|
|
config_entry: PGLabConfigEntry,
|
|
async_add_entities: AddConfigEntryEntitiesCallback,
|
|
) -> None:
|
|
"""Set up sensor for device."""
|
|
|
|
@callback
|
|
def async_discover(
|
|
pglab_device: PyPGLabDevice,
|
|
pglab_coordinator: PGLabSensorsCoordinator,
|
|
) -> None:
|
|
"""Discover and add a PG LAB Sensor."""
|
|
pglab_discovery = config_entry.runtime_data
|
|
|
|
sensors: list[PGLabSensor] = [
|
|
PGLabSensor(
|
|
description,
|
|
pglab_discovery,
|
|
pglab_device,
|
|
pglab_coordinator,
|
|
)
|
|
for description in SENSOR_INFO
|
|
]
|
|
|
|
async_add_entities(sensors)
|
|
|
|
# Register the callback to create the sensor entity when discovered.
|
|
pglab_discovery = config_entry.runtime_data
|
|
await pglab_discovery.register_platform(hass, Platform.SENSOR, async_discover)
|
|
|
|
|
|
class PGLabSensor(PGLabSensorEntity, SensorEntity):
|
|
"""A PGLab sensor."""
|
|
|
|
def __init__(
|
|
self,
|
|
description: SensorEntityDescription,
|
|
pglab_discovery: PGLabDiscovery,
|
|
pglab_device: PyPGLabDevice,
|
|
pglab_coordinator: PGLabSensorsCoordinator,
|
|
) -> None:
|
|
"""Initialize the Sensor class."""
|
|
|
|
super().__init__(pglab_discovery, pglab_device, pglab_coordinator)
|
|
|
|
self._attr_unique_id = f"{pglab_device.id}_{description.key}"
|
|
self.entity_description = description
|
|
|
|
@callback
|
|
def _handle_coordinator_update(self) -> None:
|
|
"""Update attributes when the coordinator updates."""
|
|
|
|
self._attr_native_value = self.coordinator.get_sensor_value(
|
|
self.entity_description.key
|
|
)
|
|
super()._handle_coordinator_update()
|
|
|
|
@property
|
|
def available(self) -> bool:
|
|
"""Return PG LAB sensor availability."""
|
|
return super().available and self.native_value is not None
|