Add preview to Threshold config & option flow

.coveragerc

@@ -1203,7 +1203,6 @@ omit =
     homeassistant/components/screenlogic/light.py
     homeassistant/components/screenlogic/number.py
     homeassistant/components/screenlogic/sensor.py
-    homeassistant/components/screenlogic/services.py
     homeassistant/components/screenlogic/switch.py
     homeassistant/components/scsgate/*
     homeassistant/components/sendgrid/notify.py

.github/workflows/codeql.yml

@@ -24,11 +24,11 @@ jobs:
         uses: actions/[email protected]
 
       - name: Initialize CodeQL
-        uses: github/codeql-action/[email protected].3
+        uses: github/codeql-action/[email protected].4
         with:
           languages: python
 
       - name: Perform CodeQL Analysis
-        uses: github/codeql-action/[email protected].3
+        uses: github/codeql-action/[email protected].4
         with:
           category: "/language:python"

.pre-commit-config.yaml

@@ -1,6 +1,6 @@
 repos:
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.4.3
+    rev: v0.4.4
     hooks:
       - id: ruff
         args:

.strict-typing

@@ -48,6 +48,7 @@ homeassistant.components.adax.*
 homeassistant.components.adguard.*
 homeassistant.components.aftership.*
 homeassistant.components.air_quality.*
+homeassistant.components.airgradient.*
 homeassistant.components.airly.*
 homeassistant.components.airnow.*
 homeassistant.components.airq.*

CODEOWNERS

@@ -56,6 +56,8 @@ build.json @home-assistant/supervisor
 /tests/components/agent_dvr/ @ispysoftware
 /homeassistant/components/air_quality/ @home-assistant/core
 /tests/components/air_quality/ @home-assistant/core
+/homeassistant/components/airgradient/ @airgradienthq @joostlek
+/tests/components/airgradient/ @airgradienthq @joostlek
 /homeassistant/components/airly/ @bieniu
 /tests/components/airly/ @bieniu
 /homeassistant/components/airnow/ @asymworks

homeassistant/components/airgradient/__init__.py

@@ -0,0 +1,34 @@
+"""The Airgradient integration."""
+
+from __future__ import annotations
+
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_HOST, Platform
+from homeassistant.core import HomeAssistant
+
+from .const import DOMAIN
+from .coordinator import AirGradientDataUpdateCoordinator
+
+PLATFORMS: list[Platform] = [Platform.SENSOR]
+
+
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+    """Set up Airgradient from a config entry."""
+
+    coordinator = AirGradientDataUpdateCoordinator(hass, entry.data[CONF_HOST])
+
+    await coordinator.async_config_entry_first_refresh()
+
+    hass.data.setdefault(DOMAIN, {})[entry.entry_id] = coordinator
+
+    await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
+
+    return True
+
+
+async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+    """Unload a config entry."""
+    if unload_ok := await hass.config_entries.async_unload_platforms(entry, PLATFORMS):
+        hass.data[DOMAIN].pop(entry.entry_id)
+
+    return unload_ok

homeassistant/components/airgradient/config_flow.py

@@ -0,0 +1,83 @@
+"""Config flow for Airgradient."""
+
+from typing import Any
+
+from airgradient import AirGradientClient, AirGradientError
+import voluptuous as vol
+
+from homeassistant.components import zeroconf
+from homeassistant.config_entries import ConfigFlow, ConfigFlowResult
+from homeassistant.const import CONF_HOST, CONF_MODEL
+from homeassistant.helpers.aiohttp_client import async_get_clientsession
+
+from .const import DOMAIN
+
+
+class AirGradientConfigFlow(ConfigFlow, domain=DOMAIN):
+    """AirGradient config flow."""
+
+    def __init__(self) -> None:
+        """Initialize the config flow."""
+        self.data: dict[str, Any] = {}
+
+    async def async_step_zeroconf(
+        self, discovery_info: zeroconf.ZeroconfServiceInfo
+    ) -> ConfigFlowResult:
+        """Handle zeroconf discovery."""
+        self.data[CONF_HOST] = host = discovery_info.host
+        self.data[CONF_MODEL] = discovery_info.properties["model"]
+
+        await self.async_set_unique_id(discovery_info.properties["serialno"])
+        self._abort_if_unique_id_configured(updates={CONF_HOST: host})
+
+        session = async_get_clientsession(self.hass)
+        air_gradient = AirGradientClient(host, session=session)
+        await air_gradient.get_current_measures()
+
+        self.context["title_placeholders"] = {
+            "model": self.data[CONF_MODEL],
+        }
+        return await self.async_step_discovery_confirm()
+
+    async def async_step_discovery_confirm(
+        self, user_input: dict[str, Any] | None = None
+    ) -> ConfigFlowResult:
+        """Confirm discovery."""
+        if user_input is not None:
+            return self.async_create_entry(
+                title=self.data[CONF_MODEL],
+                data={CONF_HOST: self.data[CONF_HOST]},
+            )
+
+        self._set_confirm_only()
+        return self.async_show_form(
+            step_id="discovery_confirm",
+            description_placeholders={
+                "model": self.data[CONF_MODEL],
+            },
+        )
+
+    async def async_step_user(
+        self, user_input: dict[str, Any] | None = None
+    ) -> ConfigFlowResult:
+        """Handle a flow initialized by the user."""
+        errors: dict[str, str] = {}
+        if user_input:
+            session = async_get_clientsession(self.hass)
+            air_gradient = AirGradientClient(user_input[CONF_HOST], session=session)
+            try:
+                current_measures = await air_gradient.get_current_measures()
+            except AirGradientError:
+                errors["base"] = "cannot_connect"
+            else:
+                await self.async_set_unique_id(current_measures.serial_number)
+                self._abort_if_unique_id_configured()
+                return self.async_create_entry(
+                    title=current_measures.model,
+                    data={CONF_HOST: user_input[CONF_HOST]},
+                )
+        return self.async_show_form(
+            step_id="user",
+            data_schema=vol.Schema({vol.Required(CONF_HOST): str}),
+            errors=errors,
+        )

homeassistant/components/airgradient/const.py

@@ -0,0 +1,7 @@
+"""Constants for the Airgradient integration."""
+
+import logging
+
+DOMAIN = "airgradient"
+
+LOGGER = logging.getLogger(__package__)

homeassistant/components/airgradient/coordinator.py

@@ -0,0 +1,32 @@
+"""Define an object to manage fetching AirGradient data."""
+
+from datetime import timedelta
+
+from airgradient import AirGradientClient, AirGradientError, Measures
+
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.aiohttp_client import async_get_clientsession
+from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
+
+from .const import LOGGER
+
+
+class AirGradientDataUpdateCoordinator(DataUpdateCoordinator[Measures]):
+    """Class to manage fetching AirGradient data."""
+
+    def __init__(self, hass: HomeAssistant, host: str) -> None:
+        """Initialize coordinator."""
+        super().__init__(
+            hass,
+            logger=LOGGER,
+            name=f"AirGradient {host}",
+            update_interval=timedelta(minutes=1),
+        )
+        session = async_get_clientsession(hass)
+        self.client = AirGradientClient(host, session=session)
+
+    async def _async_update_data(self) -> Measures:
+        try:
+            return await self.client.get_current_measures()
+        except AirGradientError as error:
+            raise UpdateFailed(error) from error

homeassistant/components/airgradient/entity.py

@@ -0,0 +1,24 @@
+"""Base class for AirGradient entities."""
+
+from homeassistant.helpers.device_registry import DeviceInfo
+from homeassistant.helpers.update_coordinator import CoordinatorEntity
+
+from .const import DOMAIN
+from .coordinator import AirGradientDataUpdateCoordinator
+
+
+class AirGradientEntity(CoordinatorEntity[AirGradientDataUpdateCoordinator]):
+    """Defines a base AirGradient entity."""
+
+    _attr_has_entity_name = True
+
+    def __init__(self, coordinator: AirGradientDataUpdateCoordinator) -> None:
+        """Initialize airgradient entity."""
+        super().__init__(coordinator)
+        self._attr_device_info = DeviceInfo(
+            identifiers={(DOMAIN, coordinator.data.serial_number)},
+            model=coordinator.data.model,
+            manufacturer="AirGradient",
+            serial_number=coordinator.data.serial_number,
+            sw_version=coordinator.data.firmware_version,
+        )

homeassistant/components/airgradient/icons.json

@@ -0,0 +1,15 @@
+{
+  "entity": {
+    "sensor": {
+      "total_volatile_organic_component_index": {
+        "default": "mdi:molecule"
+      },
+      "nitrogen_index": {
+        "default": "mdi:molecule"
+      },
+      "pm003_count": {
+        "default": "mdi:blur"
+      }
+    }
+  }
+}

homeassistant/components/airgradient/manifest.json

@@ -0,0 +1,11 @@
+{
+  "domain": "airgradient",
+  "name": "Airgradient",
+  "codeowners": ["@airgradienthq", "@joostlek"],
+  "config_flow": true,
+  "documentation": "https://www.home-assistant.io/integrations/airgradient",
+  "integration_type": "device",
+  "iot_class": "local_polling",
+  "requirements": ["airgradient==0.4.0"],
+  "zeroconf": ["_airgradient._tcp.local."]
+}