Add self.api to energy entities

Teslemetry · Feb 4, 2024 · 708b8c5 · 708b8c5
1 parent 77e868c
commit 708b8c5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/custom_components/teslemetry/entity.py b/custom_components/teslemetry/entity.py
@@ -119,6 +119,7 @@ def __init__(
         """Initialize common aspects of a Teslemetry entity."""
         super().__init__(energysite.live_coordinator, key)
         self._attr_unique_id = f"{energysite.id}-{key}"
+        self.api = energysite.api
 
         self._attr_device_info = DeviceInfo(
             identifiers={(DOMAIN, str(energysite.id))},
@@ -141,6 +142,7 @@ def __init__(
         """Initialize common aspects of a Teslemetry entity."""
         super().__init__(energysite.info_coordinator, key)
         self._attr_unique_id = f"{energysite.id}-{key}"
+        self.api = energysite.api
 
         self._attr_device_info = DeviceInfo(
             identifiers={(DOMAIN, str(energysite.id))},