mirror of
https://github.com/home-assistant/core.git
synced 2026-07-02 04:06:41 +01:00
Joost Lekkerkerker
88 lines
2.7 KiB
Python
88 lines
2.7 KiB
Python
"""Support for Ridwell sensors."""
|
|
|
|
from collections.abc import Mapping
|
|
from datetime import date
|
|
from typing import Any
|
|
|
|
from aioridwell.model import RidwellAccount
|
|
|
|
from homeassistant.components.sensor import (
|
|
SensorDeviceClass,
|
|
SensorEntity,
|
|
SensorEntityDescription,
|
|
)
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
|
|
|
|
from .const import SENSOR_TYPE_NEXT_PICKUP
|
|
from .coordinator import RidwellConfigEntry, RidwellDataUpdateCoordinator
|
|
from .entity import RidwellEntity
|
|
|
|
ATTR_CATEGORY = "category"
|
|
ATTR_PICKUP_STATE = "pickup_state"
|
|
ATTR_PICKUP_TYPES = "pickup_types"
|
|
ATTR_QUANTITY = "quantity"
|
|
|
|
SENSOR_DESCRIPTION = SensorEntityDescription(
|
|
key=SENSOR_TYPE_NEXT_PICKUP,
|
|
translation_key="next_pickup",
|
|
device_class=SensorDeviceClass.DATE,
|
|
)
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant,
|
|
entry: RidwellConfigEntry,
|
|
async_add_entities: AddConfigEntryEntitiesCallback,
|
|
) -> None:
|
|
"""Set up Ridwell sensors based on a config entry."""
|
|
coordinator = entry.runtime_data
|
|
|
|
async_add_entities(
|
|
RidwellSensor(coordinator, account, SENSOR_DESCRIPTION)
|
|
for account in coordinator.accounts.values()
|
|
)
|
|
|
|
|
|
class RidwellSensor(RidwellEntity, SensorEntity):
|
|
"""Define a Ridwell pickup sensor."""
|
|
|
|
def __init__(
|
|
self,
|
|
coordinator: RidwellDataUpdateCoordinator,
|
|
account: RidwellAccount,
|
|
description: SensorEntityDescription,
|
|
) -> None:
|
|
"""Initialize."""
|
|
super().__init__(coordinator, account)
|
|
|
|
self._attr_unique_id = f"{account.account_id}_{description.key}"
|
|
self.entity_description = description
|
|
|
|
@property
|
|
def extra_state_attributes(self) -> Mapping[str, Any]:
|
|
"""Return entity specific state attributes."""
|
|
attrs: dict[str, Any] = {
|
|
ATTR_PICKUP_TYPES: {},
|
|
ATTR_PICKUP_STATE: self.next_pickup_event.state.value,
|
|
}
|
|
|
|
for pickup in self.next_pickup_event.pickups:
|
|
if pickup.name not in attrs[ATTR_PICKUP_TYPES]:
|
|
attrs[ATTR_PICKUP_TYPES][pickup.name] = {
|
|
ATTR_CATEGORY: pickup.category.value,
|
|
ATTR_QUANTITY: pickup.quantity,
|
|
}
|
|
else:
|
|
# Ridwell's API will return distinct objects, even if they have the
|
|
# same name (e.g. two pickups of Latex Paint will show up as two
|
|
# objects) - so, we sum the quantities:
|
|
attrs[ATTR_PICKUP_TYPES][pickup.name][ATTR_QUANTITY] += pickup.quantity
|
|
|
|
return attrs
|
|
|
|
@property
|
|
def native_value(self) -> date:
|
|
"""Return the value reported by the sensor."""
|
|
return self.next_pickup_event.pickup_date
|