From b0633797dbf4509ddd88b0cf8b1c69fba9c07aa9 Mon Sep 17 00:00:00 2001 From: Andrew Date: Tue, 25 Mar 2025 00:27:39 +0000 Subject: [PATCH] [server]: Handle no trains --- server/src/functions/fetch_transient_data/lambda_function.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/server/src/functions/fetch_transient_data/lambda_function.py b/server/src/functions/fetch_transient_data/lambda_function.py index 2a2b443..39ddefa 100644 --- 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