Add integration platform helper (#29914)

homeassistant/helpers/integration_platform.py

@@ -0,0 +1,46 @@
+"""Helpers to help with integration platforms."""
+import asyncio
+import logging
+from typing import Any, Awaitable, Callable
+
+from homeassistant.core import Event, HomeAssistant
+from homeassistant.loader import IntegrationNotFound, async_get_integration, bind_hass
+from homeassistant.setup import ATTR_COMPONENT, EVENT_COMPONENT_LOADED
+
+_LOGGER = logging.getLogger(__name__)
+
+
+@bind_hass
+async def async_process_integration_platforms(
+    hass: HomeAssistant,
+    platform_name: str,
+    # Any = platform.
+    process_platform: Callable[[HomeAssistant, str, Any], Awaitable[None]],
+) -> None:
+    """Process a specific platform for all current and future loaded integrations."""
+
+    async def _process(component_name: str) -> None:
+        """Process the intents of a component."""
+        try:
+            integration = await async_get_integration(hass, component_name)
+            platform = integration.get_platform(platform_name)
+        except (IntegrationNotFound, ImportError):
+            return
+
+        try:
+            await process_platform(hass, component_name, platform)
+        except Exception:  # pylint: disable=broad-except
+            _LOGGER.exception(
+                "Error processing platform %s.%s", component_name, platform_name
+            )
+
+    async def async_component_loaded(event: Event) -> None:
+        """Handle a new component loaded."""
+        await _process(event.data[ATTR_COMPONENT])
+
+    hass.bus.async_listen(EVENT_COMPONENT_LOADED, async_component_loaded)
+
+    tasks = [_process(comp) for comp in hass.config.components]
+
+    if tasks:
+        await asyncio.gather(*tasks)