Cleanup

Teslemetry · Jun 7, 2024 · 244eb81 · 244eb81
1 parent f442326
commit 244eb81
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 8 deletions.
diff --git a/custom_components/teslemetry/__init__.py b/custom_components/teslemetry/__init__.py
@@ -120,8 +120,8 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
         elif "energy_site_id" in product and Scope.ENERGY_DEVICE_DATA in scopes:
             site_id = product["energy_site_id"]
             api = EnergySpecific(teslemetry.energy, site_id)
-            live_coordinator = TeslemetryEnergySiteLiveCoordinator(hass, api, entry.entry_id)
-            info_coordinator = TeslemetryEnergySiteInfoCoordinator(hass, api, entry.entry_id, product)
+            live_coordinator = TeslemetryEnergySiteLiveCoordinator(hass, api)
+            info_coordinator = TeslemetryEnergySiteInfoCoordinator(hass, api, product)
 
             device = DeviceInfo(
                 identifiers={(DOMAIN, str(site_id))},

diff --git a/custom_components/teslemetry/coordinator.py b/custom_components/teslemetry/coordinator.py
@@ -129,7 +129,7 @@ async def _async_update_data(self) -> dict[str, Any]:
 class TeslemetryEnergySiteLiveCoordinator(DataUpdateCoordinator[dict[str, Any]]):
     """Class to manage fetching energy site live status from the Teslemetry API."""
 
-    def __init__(self, hass: HomeAssistant, api: EnergySpecific, uid: str, ) -> None:
+    def __init__(self, hass: HomeAssistant, api: EnergySpecific) -> None:
         """Initialize Teslemetry Energy Site Live coordinator."""
         super().__init__(
             hass,
@@ -147,8 +147,6 @@ async def _async_update_data(self) -> dict[str, Any]:
             data = (await self.api.live_status())["response"]
         except InvalidToken as e:
             raise ConfigEntryAuthFailed from e
-        except SubscriptionRequired as e:
-            raise ConfigEntryAuthFailed from e
         except TeslaFleetError as e:
             raise UpdateFailed(e.message) from e
         except TypeError as e:
@@ -167,7 +165,7 @@ async def _async_update_data(self) -> dict[str, Any]:
 class TeslemetryEnergySiteInfoCoordinator(DataUpdateCoordinator[dict[str, Any]]):
     """Class to manage fetching energy site info from the Teslemetry API."""
 
-    def __init__(self, hass: HomeAssistant, api: EnergySpecific, uid: str, product: dict) -> None:
+    def __init__(self, hass: HomeAssistant, api: EnergySpecific, product: dict) -> None:
         """Initialize Teslemetry Energy Info coordinator."""
         super().__init__(
             hass,
@@ -186,8 +184,6 @@ async def _async_update_data(self) -> dict[str, Any]:
             data = (await self.api.site_info())["response"]
         except InvalidToken as e:
             raise ConfigEntryAuthFailed from e
-        except SubscriptionRequired as e:
-            raise ConfigEntryAuthFailed from e
         except TeslaFleetError as e:
             raise UpdateFailed(e.message) from e
         except TypeError as e: