Add Sensors to Weatherflow Cloud

homeassistant/components/weatherflow_cloud/__init__.py

@@ -9,7 +9,7 @@
 from .const import DOMAIN
 from .coordinator import WeatherFlowCloudDataUpdateCoordinator
 
-PLATFORMS: list[Platform] = [Platform.WEATHER]
+PLATFORMS: list[Platform] = [Platform.SENSOR, Platform.WEATHER]
 
 
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:

homeassistant/components/weatherflow_cloud/coordinator.py

@@ -21,12 +21,11 @@ class WeatherFlowCloudDataUpdateCoordinator(
     def __init__(self, hass: HomeAssistant, api_token: str) -> None:
         """Initialize global WeatherFlow forecast data updater."""
         self.weather_api = WeatherFlowRestAPI(api_token=api_token)
-
         super().__init__(
             hass,
             LOGGER,
             name=DOMAIN,
-            update_interval=timedelta(minutes=15),
+            update_interval=timedelta(seconds=60),
         )
 
     async def _async_update_data(self) -> dict[int, WeatherFlowDataREST]:

homeassistant/components/weatherflow_cloud/entity.py

@@ -0,0 +1,38 @@
+"""Base entity class for WeatherFlow Cloud integration."""
+
+from weatherflow4py.models.rest.unified import WeatherFlowDataREST
+
+from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
+from homeassistant.helpers.update_coordinator import CoordinatorEntity
+
+from .const import ATTR_ATTRIBUTION, DOMAIN, MANUFACTURER
+from .coordinator import WeatherFlowCloudDataUpdateCoordinator
+
+
+class WeatherFlowCloudEntity(CoordinatorEntity[WeatherFlowCloudDataUpdateCoordinator]):
+    """Base entity class to use for everything."""
+
+    _attr_attribution = ATTR_ATTRIBUTION
+    _attr_has_entity_name = True
+
+    def __init__(
+        self,
+        coordinator: WeatherFlowCloudDataUpdateCoordinator,
+        station_id: int,
+    ) -> None:
+        """Class initializer."""
+        super().__init__(coordinator)
+        self.station_id = station_id
+
+        self._attr_device_info = DeviceInfo(
+            name=self.station.station.name,
+            entry_type=DeviceEntryType.SERVICE,
+            identifiers={(DOMAIN, str(station_id))},
+            manufacturer=MANUFACTURER,
+            configuration_url=f"https://tempestwx.com/station/{station_id}/grid",
+        )
+
+    @property
+    def station(self) -> WeatherFlowDataREST:
+        """Individual Station data."""
+        return self.coordinator.data[self.station_id]

homeassistant/components/weatherflow_cloud/icons.json

@@ -0,0 +1,42 @@
+{
+  "entity": {
+    "sensor": {
+      "air_temperature": {
+        "default": "mdi:thermometer"
+      },
+      "air_density": {
+        "default": "mdi:format-line-weight"
+      },
+      "feels_like": {
+        "default": "mdi:thermometer"
+      },
+      "heat_index": {
+        "default": "mdi:sun-thermometer"
+      },
+      "wet_bulb_temperature": {
+        "default": "mdi:thermometer-water"
+      },
+      "wet_bulb_globe_temperature": {
+        "default": "mdi:thermometer-water"
+      },
+      "lightning_strike_count": {
+        "default": "mdi:lightning-bolt"
+      },
+      "lightning_strike_count_last_1hr": {
+        "default": "mdi:lightning-bolt"
+      },
+      "lightning_strike_count_last_3hr": {
+        "default": "mdi:lightning-bolt"
+      },
+      "lightning_strike_last_distance": {
+        "default": "mdi:lightning-bolt"
+      },
+      "lightning_strike_last_epoch": {
+        "default": "mdi:lightning-bolt"
+      },
+      "wind_chill": {
+        "default": "mdi:snowflake-thermometer"
+      }
+    }
+  }
+}

homeassistant/components/weatherflow_cloud/manifest.json

@@ -5,5 +5,6 @@
   "config_flow": true,
   "documentation": "https://www.home-assistant.io/integrations/weatherflow_cloud",
   "iot_class": "cloud_polling",
+  "loggers": ["weatherflow4py"],
   "requirements": ["weatherflow4py==0.2.21"]
 }

homeassistant/components/weatherflow_cloud/sensor.py

@@ -0,0 +1,209 @@
+"""Sensors for cloud based weatherflow."""
+
+from __future__ import annotations
+
+from collections.abc import Callable
+from dataclasses import dataclass
+from datetime import UTC, date, datetime
+from decimal import Decimal
+
+from weatherflow4py.models.rest.observation import Observation
+
+from homeassistant.components.sensor import (
+    SensorDeviceClass,
+    SensorEntity,
+    SensorEntityDescription,
+    SensorStateClass,
+)
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import UnitOfLength, UnitOfPressure, UnitOfTemperature
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.typing import StateType
+
+from .const import DOMAIN
+from .coordinator import WeatherFlowCloudDataUpdateCoordinator
+from .entity import WeatherFlowCloudEntity
+
+
+@dataclass(frozen=True, kw_only=True)
+class WeatherFlowCloudSensorEntityDescription(
+    SensorEntityDescription,
+):
+    """Describes a weatherflow sensor."""
+
+    value_fn: Callable[[Observation], int | str | datetime | None]
+
+
+WF_SENSORS: tuple[WeatherFlowCloudSensorEntityDescription, ...] = (
+    # Air Sensors
+    WeatherFlowCloudSensorEntityDescription(
+        key="air_density",
+        translation_key="air_density",
+        state_class=SensorStateClass.MEASUREMENT,
+        suggested_display_precision=5,
+        value_fn=lambda data: data.air_density,
+        native_unit_of_measurement="kg/m³",
+    ),
+    # Temp Sensors
+    WeatherFlowCloudSensorEntityDescription(
+        key="air_temperature",
+        translation_key="air_temperature",
+        device_class=SensorDeviceClass.TEMPERATURE,
+        state_class=SensorStateClass.MEASUREMENT,
+        suggested_display_precision=1,
+        value_fn=lambda data: data.air_temperature,
+        native_unit_of_measurement=UnitOfTemperature.CELSIUS,
+    ),
+    WeatherFlowCloudSensorEntityDescription(
+        key="dew_point",
+        translation_key="dew_point",
+        value_fn=lambda data: data.dew_point,
+        native_unit_of_measurement=UnitOfTemperature.CELSIUS,
+        device_class=SensorDeviceClass.TEMPERATURE,
+        state_class=SensorStateClass.MEASUREMENT,
+        suggested_display_precision=1,
+    ),
+    WeatherFlowCloudSensorEntityDescription(
+        key="feels_like",
+        translation_key="feels_like",
+        device_class=SensorDeviceClass.TEMPERATURE,
+        state_class=SensorStateClass.MEASUREMENT,
+        suggested_display_precision=1,
+        value_fn=lambda data: data.feels_like,
+        native_unit_of_measurement=UnitOfTemperature.CELSIUS,
+    ),
+    WeatherFlowCloudSensorEntityDescription(
+        key="heat_index",
+        translation_key="heat_index",
+        device_class=SensorDeviceClass.TEMPERATURE,
+        state_class=SensorStateClass.MEASUREMENT,
+        suggested_display_precision=1,
+        value_fn=lambda data: data.heat_index,
+        native_unit_of_measurement=UnitOfTemperature.CELSIUS,
+    ),
+    WeatherFlowCloudSensorEntityDescription(
+        key="wind_chill",
+        translation_key="wind_chill",
+        device_class=SensorDeviceClass.TEMPERATURE,
+        state_class=SensorStateClass.MEASUREMENT,
+        suggested_display_precision=1,
+        value_fn=lambda data: data.wind_chill,
+        native_unit_of_measurement=UnitOfTemperature.CELSIUS,
+    ),
+    WeatherFlowCloudSensorEntityDescription(
+        key="wet_bulb_temperature",
+        translation_key="wet_bulb_temperature",
+        device_class=SensorDeviceClass.TEMPERATURE,
+        state_class=SensorStateClass.MEASUREMENT,
+        suggested_display_precision=1,
+        value_fn=lambda data: data.wet_bulb_temperature,
+        native_unit_of_measurement=UnitOfTemperature.CELSIUS,
+    ),
+    WeatherFlowCloudSensorEntityDescription(
+        key="wet_bulb_globe_temperature",
+        translation_key="wet_bulb_globe_temperature",
+        device_class=SensorDeviceClass.TEMPERATURE,
+        state_class=SensorStateClass.MEASUREMENT,
+        suggested_display_precision=1,
+        value_fn=lambda data: data.wet_bulb_globe_temperature,
+        native_unit_of_measurement=UnitOfTemperature.CELSIUS,
+    ),
+    # Pressure Sensors
+    WeatherFlowCloudSensorEntityDescription(
+        key="barometric_pressure",
+        translation_key="barometric_pressure",
+        value_fn=lambda data: data.barometric_pressure,
+        native_unit_of_measurement=UnitOfPressure.MBAR,
+        device_class=SensorDeviceClass.ATMOSPHERIC_PRESSURE,
+        state_class=SensorStateClass.MEASUREMENT,
+        suggested_display_precision=3,
+    ),
+    WeatherFlowCloudSensorEntityDescription(
+        key="sea_level_pressure",
+        translation_key="sea_level_pressure",
+        value_fn=lambda data: data.sea_level_pressure,
+        native_unit_of_measurement=UnitOfPressure.MBAR,
+        device_class=SensorDeviceClass.ATMOSPHERIC_PRESSURE,
+        state_class=SensorStateClass.MEASUREMENT,
+        suggested_display_precision=3,
+    ),
+    # Lightning Sensors
+    WeatherFlowCloudSensorEntityDescription(
+        key="lightning_strike_count",
+        translation_key="lightning_strike_count",
+        state_class=SensorStateClass.TOTAL,
+        value_fn=lambda data: data.lightning_strike_count,
+    ),
+    WeatherFlowCloudSensorEntityDescription(
+        key="lightning_strike_count_last_1hr",
+        translation_key="lightning_strike_count_last_1hr",
+        state_class=SensorStateClass.TOTAL,
+        value_fn=lambda data: data.lightning_strike_count_last_1hr,
+    ),
+    WeatherFlowCloudSensorEntityDescription(
+        key="lightning_strike_count_last_3hr",
+        translation_key="lightning_strike_count_last_3hr",
+        state_class=SensorStateClass.TOTAL,
+        value_fn=lambda data: data.lightning_strike_count_last_3hr,
+    ),
+    WeatherFlowCloudSensorEntityDescription(
+        key="lightning_strike_last_distance",
+        translation_key="lightning_strike_last_distance",
+        state_class=SensorStateClass.MEASUREMENT,
+        device_class=SensorDeviceClass.DISTANCE,
+        native_unit_of_measurement=UnitOfLength.KILOMETERS,
+        value_fn=lambda data: data.lightning_strike_last_distance,
+    ),
+    WeatherFlowCloudSensorEntityDescription(
+        key="lightning_strike_last_epoch",
+        translation_key="lightning_strike_last_epoch",
+        device_class=SensorDeviceClass.TIMESTAMP,
+        value_fn=lambda data: datetime.fromtimestamp(
+            data.lightning_strike_last_epoch, tz=UTC
+        ),
+    ),
+)
+
+
+async def async_setup_entry(
+    hass: HomeAssistant,
+    entry: ConfigEntry,
+    async_add_entities: AddEntitiesCallback,
+) -> None:
+    """Set up WeatherFlow sensors based on a config entry."""
+
+    coordinator: WeatherFlowCloudDataUpdateCoordinator = hass.data[DOMAIN][
+        entry.entry_id
+    ]
+
+    stations = coordinator.data.keys()
+
+    async_add_entities(
+        WeatherFlowCloudSensor(coordinator, sensor_description, station_id)
+        for station_id in stations
+        for sensor_description in WF_SENSORS
+    )
+
+
+class WeatherFlowCloudSensor(WeatherFlowCloudEntity, SensorEntity):
+    """Implementation of a WeatherFlow sensor."""
+
+    entity_description: WeatherFlowCloudSensorEntityDescription
+
+    def __init__(
+        self,
+        coordinator: WeatherFlowCloudDataUpdateCoordinator,
+        description: WeatherFlowCloudSensorEntityDescription,
+        station_id: int,
+    ) -> None:
+        """Initialize the sensor."""
+        # Initialize the Entity Class
+        super().__init__(coordinator, station_id)
+        self.entity_description = description
+        self._attr_unique_id = f"{station_id}_{description.key}"
+
+    @property
+    def native_value(self) -> StateType | date | datetime | Decimal | None:
+        """Return the state of the sensor."""
+        return self.entity_description.value_fn(self.station.observation.obs[0])

homeassistant/components/weatherflow_cloud/strings.json

@@ -23,5 +23,65 @@
       "already_configured": "[%key:common::config_flow::abort::already_configured_service%]",
       "reauth_successful": "[%key:common::config_flow::abort::reauth_successful%]"
     }
+  },
+  "entity": {
+    "sensor": {
+      "air_density": {
+        "name": "Air density"
+      },
+      "barometric_pressure": {
+        "name": "Pressure barometric"
+      },
+      "sea_level_pressure": {
+        "name": "Pressure sea level"
+      },
+
+      "dew_point": {
+        "name": "Dew point"
+      },
+      "lightning_strike_count": {
+        "name": "Lightning count"
+      },
+      "lightning_strike_count_last_1hr": {
+        "name": "Lightning count last 1 hr"
+      },
+      "lightning_strike_count_last_3hr": {
+        "name": "Lightning count last 3 hr"
+      },
+      "lightning_strike_last_distance": {
+        "name": "Lightning last distance"
+      },
+      "lightning_strike_last_epoch": {
+        "name": "Lightning last strike"
+      },
+
+      "wind_chill": {
+        "name": "Wind chill"
+      },
+      "wind_direction": {
+        "name": "Wind direction"
+      },
+      "wind_direction_cardinal": {
+        "name": "Wind direction (cardinal)"
+      },
+      "wind_gust": {
+        "name": "Wind gust"
+      },
+      "wind_lull": {
+        "name": "Wind lull"
+      },
+      "feels_like": {
+        "name": "Feels like"
+      },
+      "heat_index": {
+        "name": "Heat index"
+      },
+      "wet_bulb_temperature": {
+        "name": "Wet bulb temperature"
+      },
+      "wet_bulb_globe_temperature": {
+        "name": "Wet bulb globe temperature"
+      }
+    }
   }
 }