Start

sternwarte-homeassistant-integration/custom_components/sternwarte/datetime.py

@@ -0,0 +1,74 @@
+"""Datetime entities for Sternwarte schedules."""
+
+from __future__ import annotations
+
+from datetime import datetime
+
+from homeassistant.components.datetime import DateTimeEntity
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.util import dt as dt_util
+
+from .const import DOMAIN, ENTITY_SCHEDULE_CLOSE, ENTITY_SCHEDULE_OPEN
+from .entity import SternwarteEntity
+
+
+async def async_setup_entry(
+    hass: HomeAssistant,
+    entry: ConfigEntry,
+    async_add_entities: AddEntitiesCallback,
+) -> None:
+    """Set up Sternwarte datetime entities."""
+    coordinator = hass.data[DOMAIN][entry.entry_id]
+    async_add_entities(
+        [
+            SternwarteOpenScheduleEntity(coordinator),
+            SternwarteCloseScheduleEntity(coordinator),
+        ]
+    )
+
+
+class SternwarteScheduleEntity(SternwarteEntity, DateTimeEntity):
+    """Base datetime entity for schedules."""
+
+    _attr_icon = "mdi:calendar-clock"
+
+    async def _async_set_schedule(self, value: datetime | None) -> None:
+        raise NotImplementedError
+
+    async def async_set_value(self, value: datetime) -> None:
+        if value.tzinfo is None:
+            value = value.replace(tzinfo=dt_util.DEFAULT_TIME_ZONE)
+        await self._async_set_schedule(value)
+
+
+class SternwarteOpenScheduleEntity(SternwarteScheduleEntity):
+    _attr_name = "Öffnen geplant"
+
+    def __init__(self, coordinator):
+        super().__init__(coordinator, ENTITY_SCHEDULE_OPEN)
+
+    @property
+    def native_value(self) -> datetime | None:
+        value = self.coordinator.open_schedule_at
+        return dt_util.as_local(value) if value is not None else None
+
+    async def _async_set_schedule(self, value: datetime | None) -> None:
+        await self.coordinator.async_set_open_schedule(value)
+
+
+class SternwarteCloseScheduleEntity(SternwarteScheduleEntity):
+    _attr_name = "Schließen geplant"
+
+    def __init__(self, coordinator):
+        super().__init__(coordinator, ENTITY_SCHEDULE_CLOSE)
+
+    @property
+    def native_value(self) -> datetime | None:
+        value = self.coordinator.close_schedule_at
+        return dt_util.as_local(value) if value is not None else None
+
+    async def _async_set_schedule(self, value: datetime | None) -> None:
+        await self.coordinator.async_set_close_schedule(value)
+