Initialize docker stack repo

homeassistant/config/custom_components/yandex_pogoda/device_trigger.py

@@ -0,0 +1,74 @@
+"""Yandex.Pogoda triggers."""
+
+import logging
+
+import voluptuous as vol
+from homeassistant.components.device_automation import \
+    DEVICE_TRIGGER_BASE_SCHEMA as HA_TRIGGER_BASE_SCHEMA
+from homeassistant.components.homeassistant.triggers import \
+    event as event_trigger
+from homeassistant.const import (CONF_DEVICE_ID, CONF_DOMAIN, CONF_PLATFORM,
+                                 CONF_TYPE)
+from homeassistant.core import HomeAssistant
+
+from .const import DOMAIN, WEATHER_STATES_CONVERSION, map_state
+
+
+def generate_triggers() -> list:
+    """Generate triggers list."""
+    result = []
+    for s in WEATHER_STATES_CONVERSION:
+        for d in [True, False]:
+            result.append(map_state(src=s, is_day=d, mapping=WEATHER_STATES_CONVERSION))
+
+    return list(set(result))
+
+
+_LOGGER = logging.getLogger(__name__)
+
+TRIGGERS = generate_triggers()
+
+TRIGGER_SCHEMA = HA_TRIGGER_BASE_SCHEMA.extend(
+    {
+        vol.Required(CONF_TYPE): vol.In(TRIGGERS),
+    }
+)
+
+
+async def async_get_triggers(_hass: HomeAssistant, device_id):
+    """Return a list of triggers."""
+    triggers = []
+
+    for t in TRIGGERS:
+        triggers.append(
+            {
+                # Required fields of TRIGGER_BASE_SCHEMA
+                CONF_PLATFORM: "device",
+                CONF_DOMAIN: DOMAIN,
+                CONF_DEVICE_ID: device_id,
+                # Required fields of TRIGGER_SCHEMA
+                CONF_TYPE: t,
+            }
+        )
+
+    return triggers
+
+
+async def async_attach_trigger(hass: HomeAssistant, config, action, automation_info):
+    """Attach a trigger."""
+    config = TRIGGER_SCHEMA(config)
+    _LOGGER.debug("Got subscription to trigger: %s", config)
+    event_config = event_trigger.TRIGGER_SCHEMA(
+        {
+            event_trigger.CONF_PLATFORM: "event",
+            event_trigger.CONF_EVENT_TYPE: DOMAIN + "_event",
+            event_trigger.CONF_EVENT_DATA: {
+                CONF_DEVICE_ID: config[CONF_DEVICE_ID],
+                CONF_TYPE: config[CONF_TYPE],
+            },
+        }
+    )
+
+    return await event_trigger.async_attach_trigger(
+        hass, event_config, action, automation_info, platform_type="device"
+    )