Home Assistant Git Exporter

config/custom_components/vigieau/sensor.py

@@ -0,0 +1,29 @@
+import logging
+
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.config_entries import ConfigEntry
+
+from . import (
+    UsageRestrictionEntity,
+    AlertLevelEntity,
+)
+from .const import DOMAIN, SENSOR_DEFINITIONS, SENSOR_DEFINITIONS
+
+_LOGGER = logging.getLogger(__name__)
+
+
+async def async_setup_entry(
+    hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+) -> None:
+    vigieau_coordinator = hass.data[DOMAIN][entry.entry_id]["vigieau_coordinator"]
+    sensors = [
+        UsageRestrictionEntity(
+            vigieau_coordinator, hass, "mydi", entry, sensor_description
+        )
+        for sensor_description in SENSOR_DEFINITIONS
+    ]
+    sensors.append(AlertLevelEntity(vigieau_coordinator, hass, entry))
+
+    async_add_entities(sensors)
+    await vigieau_coordinator.async_config_entry_first_refresh()