jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-04-23 16:57:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Simplify Swiss public transport coordinator (#128891)

Browse Source
This commit is contained in:
Cyrill Raccaud 2024-10-21 14:15:18 +02:00 committed by GitHub
parent 28a8ed62f3
commit 62773fa88a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -75,13 +75,6 @@ class SwissPublicTransportDataUpdateCoordinator(
return departure_datetime - dt_util.as_local(dt_util.utcnow())
return None
def nth_departure_time(self, i: int) -> datetime | None:
"""Get nth departure time."""
connections = self._opendata.connections
if len(connections) > i and connections[i] is not None:
return dt_util.parse_datetime(connections[i]["departure"])
return None
async def _async_update_data(self) -> list[DataConnection]:
return await self.fetch_connections(limit=CONNECTIONS_COUNT)
@ -101,7 +94,7 @@ class SwissPublicTransportDataUpdateCoordinator(
connections = self._opendata.connections
return [
DataConnection(
departure=self.nth_departure_time(i),
departure=dt_util.parse_datetime(connections[i]["departure"]),
train_number=connections[i]["number"],
platform=connections[i]["platform"],
transfers=connections[i]["transfers"],

2
tests/components/swiss_public_transport/fixtures/connections.json
View File

@ -99,7 +99,7 @@
"line": "T10"
},
{
"departure": "2024-01-06T18:14:00+0100",
"departure": "invalid",
"number": 11,
"platform": 11,
"transfers": 0,