Fix imports

Author: Bre77

custom_components/teslemetry/coordinator.py

@@ -5,7 +5,7 @@
 from typing import Any
 from random import randint
 
-from tesla_fleet_api.teslemetry.vehicles import TeslemetryVehicleFleet
+from tesla_fleet_api.teslemetry.vehicles import TeslemetryVehicle
 from tesla_fleet_api.tesla.energysite import EnergySite
 from tesla_fleet_api.const import VehicleDataEndpoint
 from tesla_fleet_api.exceptions import (
@@ -49,7 +49,7 @@ class TeslemetryVehicleDataCoordinator(DataUpdateCoordinator[dict[str, Any]]):
     """Class to manage fetching data from the Teslemetry API."""
 
     def __init__(
-        self, hass: HomeAssistant, api: TeslemetryVehicleFleet, product: dict
+        self, hass: HomeAssistant, api: TeslemetryVehicle, product: dict
     ) -> None:
         """Initialize Teslemetry Vehicle Update Coordinator."""
         super().__init__(

custom_components/teslemetry/entity.py

@@ -4,7 +4,7 @@
 from propcache import cached_property
 
 from homeassistant.helpers.entity import Entity
-from tesla_fleet_api.teslemetry.vehicles import TeslemetryVehicleFleet
+from tesla_fleet_api.teslemetry.vehicles import TeslemetryVehicle
 from tesla_fleet_api.tesla.energysite import EnergySite
 from tesla_fleet_api.const import Scope
 from teslemetry_stream import Signal
@@ -29,7 +29,7 @@ class TeslemetryRootEntity(Entity):
 
     _attr_has_entity_name = True
     scoped: bool
-    api: TeslemetryVehicleFleet | EnergySite
+    api: TeslemetryVehicle | EnergySite
 
     def raise_for_scope(self, scope: Scope):
         """Raise an error if a scope is not available."""
@@ -197,7 +197,7 @@ def __init__(
         | TeslemetryEnergySiteLiveCoordinator
         | TeslemetryEnergySiteInfoCoordinator
         | TeslemetryEnergyHistoryCoordinator,
-        api: TeslemetryVehicleFleet | EnergySite,
+        api: TeslemetryVehicle | EnergySite,
         key: str,
     ) -> None:
         """Initialize common aspects of a Teslemetry entity."""
@@ -254,7 +254,7 @@ def _async_update_attrs(self) -> None:
 class TeslemetryVehicleEntity(TeslemetryCoordinatorEntity):
     """Parent class for polled Teslemetry Vehicle entities."""
 
-    api: TeslemetryVehicleFleet
+    api: TeslemetryVehicle
 
     def __init__(
         self,

custom_components/teslemetry/media_player.py

@@ -12,7 +12,7 @@
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
-from tesla_fleet_api.teslemetry.vehicles import TeslemetryVehicleFleet
+from tesla_fleet_api.teslemetry.vehicles import TeslemetryVehicle
 
 from custom_components.teslemetry.helpers import handle_vehicle_command
 
@@ -59,7 +59,7 @@ async def async_setup_entry(
 class TeslemetryMediaEntity(TeslemetryRootEntity, MediaPlayerEntity):
     """Base vehicle media player class."""
 
-    api: TeslemetryVehicleFleet
+    api: TeslemetryVehicle
 
     _attr_device_class = MediaPlayerDeviceClass.SPEAKER
     _attr_volume_step = VOLUME_STEP

custom_components/teslemetry/models.py

@@ -8,7 +8,7 @@
 import asyncio
 from dataclasses import dataclass
 
-from tesla_fleet_api.teslemetry.vehicles import TeslemetryVehicleFleet
+from tesla_fleet_api.teslemetry.vehicles import TeslemetryVehicle
 from tesla_fleet_api.tesla.energysite import EnergySite
 from tesla_fleet_api.const import Scope
 
@@ -38,7 +38,7 @@ class TeslemetryData:
 class TeslemetryVehicleData:
     """Data for a vehicle in the Teslemetry integration."""
 
-    api: TeslemetryVehicleFleet
+    api: TeslemetryVehicle
     config_entry: ConfigEntry
     coordinator: TeslemetryVehicleDataCoordinator
     stream: TeslemetryStream

custom_components/teslemetry/select.py

@@ -9,7 +9,7 @@
     EnergyExportMode,
     EnergyOperationMode,
 )
-from tesla_fleet_api.teslemetry.vehicles import TeslemetryVehicleFleet
+from tesla_fleet_api.teslemetry.vehicles import TeslemetryVehicle
 from teslemetry_stream import Signal
 
 from dataclasses import dataclass
@@ -147,7 +147,7 @@ async def async_setup_entry(
 class TeslemetrySeatHeaterSelectEntity(TeslemetryRootEntity, SelectEntity):
     """Select entity for vehicle seat heater."""
 
-    api: TeslemetryVehicleFleet
+    api: TeslemetryVehicle
     entity_description: SeatHeaterDescription
     _attr_options = [
         TeslemetryHeaterOptions.OFF,