Improve history coordinator in Teslemetry (#128235)

homeassistant/components/teslemetry/__init__.py

@@ -135,11 +135,11 @@ async def async_setup_entry(hass: HomeAssistant, entry: TeslemetryConfigEntry) -
 
         elif "energy_site_id" in product and Scope.ENERGY_DEVICE_DATA in scopes:
             site_id = product["energy_site_id"]
-            if not (
+            powerwall = (
-                product["components"]["battery"]
+                product["components"]["battery"] or product["components"]["solar"]
-                or product["components"]["solar"]
+            )
-                or "wall_connectors" in product["components"]
+            wall_connector = "wall_connectors" in product["components"]
-            ):
+            if not powerwall and not wall_connector:
                 LOGGER.debug(
                     "Skipping Energy Site %s as it has no components",
                     site_id,
@@ -162,7 +162,11 @@ async def async_setup_entry(hass: HomeAssistant, entry: TeslemetryConfigEntry) -
                     info_coordinator=TeslemetryEnergySiteInfoCoordinator(
                         hass, api, product
                     ),
-                    history_coordinator=TeslemetryEnergyHistoryCoordinator(hass, api),
+                    history_coordinator=(
+                        TeslemetryEnergyHistoryCoordinator(hass, api)
+                        if powerwall
+                        else None
+                    ),
                     id=site_id,
                     device=device,
                 )
@@ -185,6 +189,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: TeslemetryConfigEntry) -
         *(
             energysite.history_coordinator.async_config_entry_first_refresh()
             for energysite in energysites
+            if energysite.history_coordinator
         ),
     )
 

homeassistant/components/teslemetry/entity.py

@@ -175,6 +175,8 @@ class TeslemetryEnergyHistoryEntity(TeslemetryEntity):
     ) -> None:
         """Initialize common aspects of a Teslemetry Energy Site Info entity."""
 
+        assert data.history_coordinator
+
         self.api = data.api
         self._attr_unique_id = f"{data.id}-{key}"
         self._attr_device_info = data.device

homeassistant/components/teslemetry/models.py

@@ -49,6 +49,6 @@ class TeslemetryEnergyData:
     api: EnergySpecific
     live_coordinator: TeslemetryEnergySiteLiveCoordinator
     info_coordinator: TeslemetryEnergySiteInfoCoordinator
-    history_coordinator: TeslemetryEnergyHistoryCoordinator
+    history_coordinator: TeslemetryEnergyHistoryCoordinator | None
     id: int
     device: DeviceInfo

homeassistant/components/teslemetry/sensor.py

@@ -482,8 +482,7 @@ async def async_setup_entry(
                 TeslemetryEnergyHistorySensorEntity(energysite, description)
                 for energysite in entry.runtime_data.energysites
                 for description in ENERGY_HISTORY_DESCRIPTIONS
-                if energysite.info_coordinator.data.get("components_battery")
+                if energysite.history_coordinator
-                or energysite.info_coordinator.data.get("components_solar")
             ),
         )
     )