From fec88b6ca89ab94d3286e43142c9547f05d21519 Mon Sep 17 00:00:00 2001
From: Wu Tingfeng <wutingfeng@outlook.com>
Date: Tue, 16 Jan 2024 21:07:05 +0800
Subject: [PATCH] Handle API output edge case

---
 train_arrival.py | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/train_arrival.py b/train_arrival.py
index 24a41d5..3c628dd 100644
--- a/train_arrival.py
+++ b/train_arrival.py
@@ -91,9 +91,13 @@ def get_all_station_names():  # type: () -> list[str]
     all_stations_info = json.loads(get_all_station_info())  # type: dict
     if not isinstance(all_stations_info, dict):
         all_stations_info = {}
+
+    results = all_stations_info.get("results", [])
+    if not isinstance(results, list):
+        return []
     station_names = set()
 
-    for station_info in all_stations_info.get("results", []):
+    for station_info in results:
         if isinstance(station_info, dict) and "name" in station_info:
             station_name = station_info[
                 "name"
@@ -125,10 +129,14 @@ def get_train_arrival_time_by_id(station_name):  # type: (str) -> str
             "https://connectv3.smrt.wwprojects.com/smrt/api/train_arrival_time_by_id",
             params,
         )
+
         d = json.loads(data)
-        if not d.get("results", []):
+        results = d.get("results", [])
+        if not isinstance(results, list):
             continue
-        mrt_names = set(result.get("mrt", "") for result in d["results"]) - set([""])
+        mrt_names = set(
+            result.get("mrt", "") for result in results if isinstance(result, dict)
+        ) - set([""])
         if (
             len(mrt_names) != 1 or station_name not in mrt_names
         ):  # Ensure that the 'mrt' field matches station name.