Simone Chemelli 2750535928
Use runtime_data in UptimeRobot (#142848)
* Use runtime_data in UptimeRobot

* fix unload
2025-04-14 09:51:06 +02:00

59 lines
1.6 KiB
Python

"""UptimeRobot sensor platform."""
from __future__ import annotations
from homeassistant.components.sensor import (
SensorDeviceClass,
SensorEntity,
SensorEntityDescription,
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import UptimeRobotConfigEntry
from .entity import UptimeRobotEntity
SENSORS_INFO = {
0: "pause",
1: "not_checked_yet",
2: "up",
8: "seems_down",
9: "down",
}
# Coordinator is used to centralize the data updates
PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: UptimeRobotConfigEntry,
async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the UptimeRobot sensors."""
coordinator = entry.runtime_data
async_add_entities(
UptimeRobotSensor(
coordinator,
SensorEntityDescription(
key=str(monitor.id),
entity_category=EntityCategory.DIAGNOSTIC,
device_class=SensorDeviceClass.ENUM,
options=["down", "not_checked_yet", "pause", "seems_down", "up"],
translation_key="monitor_status",
),
monitor=monitor,
)
for monitor in coordinator.data
)
class UptimeRobotSensor(UptimeRobotEntity, SensorEntity):
"""Representation of a UptimeRobot sensor."""
@property
def native_value(self) -> str:
"""Return the status of the monitor."""
return SENSORS_INFO[self.monitor.status]