home-assistant/core
1
0
Fork 1
mirror of https://github.com/home-assistant/core.git synced 2025-06-25 01:21:51 +02:00
Code Issues Packages Projects Releases Wiki Activity

Small nmbs sensor attributes refactoring (#145956)

Browse Source 
Attributes refactoring
This commit is contained in:
Simon Lamon
2025-06-02 15:13:23 +02:00
committed by GitHub
parent e5f95b3aff
commit 93b8cc38d8
1 changed files with 7 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
homeassistant/components/nmbs/sensor.py
View File

@ -247,7 +247,6 @@ class NMBSSensor(SensorEntity):
delay = get_delay_in_minutes(self._attrs.departure.delay) delay = get_delay_in_minutes(self._attrs.departure.delay)
departure = get_time_until(self._attrs.departure.time) departure = get_time_until(self._attrs.departure.time)
canceled = self._attrs.departure.canceled
attrs = { attrs = {
"destination": self._attrs.departure.station, "destination": self._attrs.departure.station,
@ -257,14 +256,13 @@ class NMBSSensor(SensorEntity):
"vehicle_id": self._attrs.departure.vehicle, "vehicle_id": self._attrs.departure.vehicle,
} }
if not canceled: attrs["canceled"] = self._attrs.departure.canceled
attrs["departure"] = f"In {departure} minutes" if attrs["canceled"]:
attrs["departure_minutes"] = departure
attrs["canceled"] = False
else:
attrs["departure"] = None attrs["departure"] = None
attrs["departure_minutes"] = None attrs["departure_minutes"] = None
attrs["canceled"] = True else:
attrs["departure"] = f"In {departure} minutes"
attrs["departure_minutes"] = departure
if self._show_on_map and self.station_coordinates: if self._show_on_map and self.station_coordinates:
attrs[ATTR_LATITUDE] = self.station_coordinates[0] attrs[ATTR_LATITUDE] = self.station_coordinates[0]
@ -280,9 +278,8 @@ class NMBSSensor(SensorEntity):
via.timebetween via.timebetween
) + get_delay_in_minutes(via.departure.delay) ) + get_delay_in_minutes(via.departure.delay)
if delay > 0: attrs["delay"] = f"{delay} minutes"
attrs["delay"] = f"{delay} minutes" attrs["delay_minutes"] = delay
attrs["delay_minutes"] = delay
return attrs return attrs