[server]: Handle no trains

2025-03-25 00:27:39 +00:00
parent 4f829cc216
commit b0633797db
1 changed files with 2 additions and 2 deletions
--- a/server/src/functions/fetch_transient_data/lambda_function.py
+++ b/server/src/functions/fetch_transient_data/lambda_function.py
@ -55,8 +55,8 @@ def fetch_trains():

        punctuality_data = fetch_punctuality_data()

-        for train in trains_json["ArrayOfObjTrainPositions"]["objTrainPositions"]:
-            train_code = str(train["TrainCode"])
+        for train in trains_json["ArrayOfObjTrainPositions"].get("objTrainPositions", []):
+                train_code = str(train["TrainCode"])
            train_status = train["TrainStatus"]
            public_message = train["PublicMessage"]
            avg_punctuality = punctuality_data.get("IrishRailTrain-" + train_code, 0)  # Default to 0 if not found