jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-13 00:07:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add base entity for Airgradient (#117135)

Browse Source
This commit is contained in:
Joost Lekkerkerker 2024-05-09 11:12:47 +02:00 committed by GitHub
parent 6485973d9b
commit b30a02dee6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 26 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
homeassistant/components/airgradient/entity.py Normal file
View File

@ -0,0 +1,24 @@
"""Base class for AirGradient entities."""
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
from .coordinator import AirGradientDataUpdateCoordinator
class AirGradientEntity(CoordinatorEntity[AirGradientDataUpdateCoordinator]):
"""Defines a base AirGradient entity."""
_attr_has_entity_name = True
def __init__(self, coordinator: AirGradientDataUpdateCoordinator) -> None:
"""Initialize airgradient entity."""
super().__init__(coordinator)
self._attr_device_info = DeviceInfo(
identifiers={(DOMAIN, coordinator.data.serial_number)},
model=coordinator.data.model,
manufacturer="AirGradient",
serial_number=coordinator.data.serial_number,
sw_version=coordinator.data.firmware_version,
)

17
homeassistant/components/airgradient/sensor.py
View File

@ -21,13 +21,12 @@ from homeassistant.const import (
UnitOfTemperature, UnitOfTemperature,
) )
from homeassistant.core import HomeAssistant, callback from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import StateType from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import AirGradientDataUpdateCoordinator from . import AirGradientDataUpdateCoordinator
from .const import DOMAIN from .const import DOMAIN
from .entity import AirGradientEntity
@dataclass(frozen=True, kw_only=True) @dataclass(frozen=True, kw_only=True)
@ -159,13 +158,9 @@ async def async_setup_entry(
add_entities() add_entities()
class AirGradientSensor( class AirGradientSensor(AirGradientEntity, SensorEntity):
CoordinatorEntity[AirGradientDataUpdateCoordinator], SensorEntity
):
"""Defines an AirGradient sensor.""" """Defines an AirGradient sensor."""
_attr_has_entity_name = True
entity_description: AirGradientSensorEntityDescription entity_description: AirGradientSensorEntityDescription
def __init__( def __init__(
@ -175,16 +170,8 @@ class AirGradientSensor(
) -> None: ) -> None:
"""Initialize airgradient sensor.""" """Initialize airgradient sensor."""
super().__init__(coordinator) super().__init__(coordinator)
self.entity_description = description self.entity_description = description
self._attr_unique_id = f"{coordinator.data.serial_number}-{description.key}" self._attr_unique_id = f"{coordinator.data.serial_number}-{description.key}"
self._attr_device_info = DeviceInfo(
identifiers={(DOMAIN, coordinator.data.serial_number)},
model=coordinator.data.model,
manufacturer="AirGradient",
serial_number=coordinator.data.serial_number,
sw_version=coordinator.data.firmware_version,
)
@property @property
def native_value(self) -> StateType: def native_value(self) -> StateType: