fix: handle missing highlights

toddrob99 · Mar 6, 2023 · 88c64be · 88c64be
1 parent de0ea50
commit 88c64be
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/statsapi/__init__.py b/statsapi/__init__.py
@@ -1039,17 +1039,19 @@ def game_highlight_data(gamePk):
         },
     )
     gameHighlights = (
-        r["dates"][0]["games"][0]["content"]
+        r["dates"][0]["games"][0]
+        .get("content", {})
         .get("highlights", {})
         .get("highlights", {})
-        .get("items", [])
     )
-    if not len(gameHighlights):
+    if not gameHighlights or not len(gameHighlights.get("items", [])):
         return []
 
     unorderedHighlights = {}
     for v in (
-        x for x in gameHighlights if isinstance(x, dict) and x["type"] == "video"
+        x
+        for x in gameHighlights["items"]
+        if isinstance(x, dict) and x["type"] == "video"
     ):
         unorderedHighlights.update({v["date"]: v})