jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-17 10:17:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove deprecated attributes from Swiss public transport integration (#120256)

Browse Source
This commit is contained in:
Cyrill Raccaud 2024-06-24 11:18:10 +02:00 committed by GitHub
parent e32a27a8ff
commit 59dd63ea86
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 0 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
homeassistant/components/swiss_public_transport/coordinator.py
View File

@ -23,8 +23,6 @@ class DataConnection(TypedDict):
"""A connection data class.""" """A connection data class."""
departure: datetime | None departure: datetime | None
next_departure: datetime | None
next_on_departure: datetime | None
duration: int | None duration: int | None
platform: str platform: str
remaining_time: str remaining_time: str
@ -88,8 +86,6 @@ class SwissPublicTransportDataUpdateCoordinator(
return [ return [
DataConnection( DataConnection(
departure=self.nth_departure_time(i), departure=self.nth_departure_time(i),
next_departure=self.nth_departure_time(i + 1),
next_on_departure=self.nth_departure_time(i + 2),
train_number=connections[i]["number"], train_number=connections[i]["number"],
platform=connections[i]["platform"], platform=connections[i]["platform"],
transfers=connections[i]["transfers"], transfers=connections[i]["transfers"],

28
homeassistant/components/swiss_public_transport/sensor.py
View File

@ -15,7 +15,6 @@ from homeassistant.components.sensor import (
SensorEntityDescription, SensorEntityDescription,
) )
from homeassistant.const import UnitOfTime from homeassistant.const import UnitOfTime
from homeassistant.core import callback
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import StateType from homeassistant.helpers.typing import StateType
@ -36,7 +35,6 @@ class SwissPublicTransportSensorEntityDescription(SensorEntityDescription):
value_fn: Callable[[DataConnection], StateType | datetime] value_fn: Callable[[DataConnection], StateType | datetime]
index: int = 0 index: int = 0
has_legacy_attributes: bool = False
SENSORS: tuple[SwissPublicTransportSensorEntityDescription, ...] = ( SENSORS: tuple[SwissPublicTransportSensorEntityDescription, ...] = (
@ -45,7 +43,6 @@ SENSORS: tuple[SwissPublicTransportSensorEntityDescription, ...] = (
key=f"departure{i or ''}", key=f"departure{i or ''}",
translation_key=f"departure{i}", translation_key=f"departure{i}",
device_class=SensorDeviceClass.TIMESTAMP, device_class=SensorDeviceClass.TIMESTAMP,
has_legacy_attributes=i == 0,
value_fn=lambda data_connection: data_connection["departure"], value_fn=lambda data_connection: data_connection["departure"],
index=i, index=i,
) )
@ -127,28 +124,3 @@ class SwissPublicTransportSensor(
return self.entity_description.value_fn( return self.entity_description.value_fn(
self.coordinator.data[self.entity_description.index] self.coordinator.data[self.entity_description.index]
) )
async def async_added_to_hass(self) -> None:
"""Prepare the extra attributes at start."""
if self.entity_description.has_legacy_attributes:
self._async_update_attrs()
await super().async_added_to_hass()
@callback
def _handle_coordinator_update(self) -> None:
"""Handle the state update and prepare the extra state attributes."""
if self.entity_description.has_legacy_attributes:
self._async_update_attrs()
return super()._handle_coordinator_update()
@callback
def _async_update_attrs(self) -> None:
"""Update the extra state attributes based on the coordinator data."""
if self.entity_description.has_legacy_attributes:
self._attr_extra_state_attributes = {
key: value
for key, value in self.coordinator.data[
self.entity_description.index
].items()
if key not in {"departure"}
}