device_tracker.py

Go to the documentation of this file.

"""Device Tracker platform for Tessie integration."""
 
from __future__ import annotations
 
from homeassistant.components.device_tracker.config_entry import TrackerEntity
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import StateType
 
from . import TessieConfigEntry
from .entity import TessieEntity
from .models import TessieVehicleData
 
PARALLEL_UPDATES = 0
 
 
async def async_setup_entry(
    hass: HomeAssistant,
    entry: TessieConfigEntry,
    async_add_entities: AddEntitiesCallback,
) -> None:
    """Set up the Tessie device tracker platform from a config entry."""
    data = entry.runtime_data
 
    async_add_entities(
        klass(vehicle)
        for klass in (
            TessieDeviceTrackerLocationEntity,
            TessieDeviceTrackerRouteEntity,
        )
        for vehicle in data.vehicles
    )
 
 
class TessieDeviceTrackerEntity(TessieEntity, TrackerEntity):
    """Base class for Tessie Tracker Entities."""
 
    def __init__(
        self,
        vehicle: TessieVehicleData,
    ) -> None:
        """Initialize the device tracker."""
        super().__init__(vehicle, self.keykey)
 
 
class TessieDeviceTrackerLocationEntity(TessieDeviceTrackerEntity):
    """Vehicle Location Device Tracker Class."""
 
    key = "location"
 
    @property
    def longitude(self) -> float | None:
        """Return the longitude of the device tracker."""
        return self.getget("drive_state_longitude")
 
    @property
    def latitude(self) -> float | None:
        """Return the latitude of the device tracker."""
        return self.getget("drive_state_latitude")
 
    @property
    def extra_state_attributes(self) -> dict[str, StateType] | None:
        """Return device state attributes."""
        return {
            "heading": self.getget("drive_state_heading"),
            "speed": self.getget("drive_state_speed"),
        }
 
 
class TessieDeviceTrackerRouteEntity(TessieDeviceTrackerEntity):
    """Vehicle Navigation Device Tracker Class."""
 
    key = "route"
 
    @property
    def longitude(self) -> float | None:
        """Return the longitude of the device tracker."""
        return self.getget("drive_state_active_route_longitude")
 
    @property
    def latitude(self) -> float | None:
        """Return the latitude of the device tracker."""
        return self.getget("drive_state_active_route_latitude")