From b5761a83c018ba77fcb7f1c50af962138b190a3e Mon Sep 17 00:00:00 2001 From: Brett Adams Date: Tue, 12 Mar 2024 07:17:42 +1000 Subject: [PATCH] Fix location data in Teslemetry (#112866) Fix location data in coordinator --- homeassistant/components/teslemetry/coordinator.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/homeassistant/components/teslemetry/coordinator.py b/homeassistant/components/teslemetry/coordinator.py index a14ed347193..d8ded1016fa 100644 --- a/homeassistant/components/teslemetry/coordinator.py +++ b/homeassistant/components/teslemetry/coordinator.py @@ -4,6 +4,7 @@ from datetime import timedelta from typing import Any from tesla_fleet_api import EnergySpecific, VehicleSpecific +from tesla_fleet_api.const import VehicleDataEndpoint from tesla_fleet_api.exceptions import TeslaFleetError, VehicleOffline from homeassistant.core import HomeAssistant @@ -13,6 +14,13 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda from .const import LOGGER, TeslemetryState SYNC_INTERVAL = 60 +ENDPOINTS = [ + VehicleDataEndpoint.CHARGE_STATE, + VehicleDataEndpoint.CLIMATE_STATE, + VehicleDataEndpoint.DRIVE_STATE, + VehicleDataEndpoint.LOCATION_DATA, + VehicleDataEndpoint.VEHICLE_STATE, +] class TeslemetryDataCoordinator(DataUpdateCoordinator[dict[str, Any]]): @@ -54,7 +62,7 @@ class TeslemetryVehicleDataCoordinator(TeslemetryDataCoordinator): """Update vehicle data using Teslemetry API.""" try: - data = await self.api.vehicle_data() + data = await self.api.vehicle_data(endpoints=ENDPOINTS) except VehicleOffline: self.data["state"] = TeslemetryState.OFFLINE return self.data