Fix line filter

warrior25 · Oct 9, 2023 · a1bae5d · a1bae5d
1 parent c573e00
commit a1bae5d
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/custom_components/Nysse/sensor.py b/custom_components/Nysse/sensor.py
@@ -88,11 +88,14 @@ def remove_unwanted_data(self, departures, journeys):
  removed_journey_count = 0
  removed_departures_count = 0
 
- # Remove unwanted journeys based on departure time and stop code
+ # Remove unwanted journeys based on departure time, stop code and line number
  for journey in journeys[:]:
- if journey["departureTime"] < self._last_update_time + timedelta(
- minutes=self.timelimit
- ) or (journey["stopCode"] != self.stop_code):
+ if (
+ journey["departureTime"]
+ < self._last_update_time + timedelta(minutes=self.timelimit)
+ or (journey["stopCode"] != self.stop_code)
+ or (journey["line"] not in self.lines)
+ ):
  journeys.remove(journey)
  removed_journey_count += 1