Remove trains from the api response

Remove trains from the API response when the destination isn't found in the calling points. Is this a temporary fix? Why is the API returning wrong info?
plutomedia987 · Dec 19, 2024 · 6d70905 · 6d70905
1 parent 27f3d67
commit 6d70905
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/custom_components/nationalrailuk/client.py b/custom_components/nationalrailuk/client.py
@@ -328,9 +328,6 @@ def process_data(self, json_message_in):
                     ############################################################
                     # Assign outputs
                     ############################################################
-                    if otherEnd:
-                        train["otherEnd"] = otherEnd
-
                     train["scheduled"] = times["sheduled"]
                     train["expected"] = times["estimated"]
                     train["origin"] = service["origin"]["location"][0]["locationName"]
@@ -343,7 +340,9 @@ def process_data(self, json_message_in):
                     train["length"] = service["length"]
                     train["callingPoints"] = callingPoints
 
-                    status["trains"].append(train)
+                    if otherEnd:
+                        train["otherEnd"] = otherEnd
+                        status["trains"].append(train)
 
                 # with open("output_test.txt", "w") as convert_file:
                 #     convert_file.write(str(status["trains"]))