Merge pull request #30 from mohakkataria/patch-1

Allow platform filter on buses
morosanmihail · Jul 28, 2024 · 7d52d1b · 7d52d1b
2 parents bc9018d + 76698e0
commit 7d52d1b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/custom_components/london_tfl/tfl_data.py b/custom_components/london_tfl/tfl_data.py
@@ -58,7 +58,7 @@ def filter_by_platform(self, filter_platform):
         if filter_platform != '':
             self._raw_result = [
                 item for item in self._raw_result
-                if filter_platform in item['platformName']
+                if ('platformName' in item and filter_platform in item['platformName']) or ('lineName' in item and filter_platform in item['lineName'])
             ]
 
     def sort_data(self, max_items):