Add check for segment structure

bertybuttface · Jan 28, 2024 · 13ba850 · 13ba850
1 parent c71660b
commit 13ba850
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/iSponsorBlockTV/api_helpers.py b/src/iSponsorBlockTV/api_helpers.py
@@ -153,6 +153,10 @@ def process_segments(response):
         if "segments" not in response:
             return [], True
 
+        for seg in response["segments"]:
+            if not isinstance(seg, dict) or "segment" not in seg or not isinstance(seg["segment"], list):
+                raise TypeError("Invalid segment structure")
+
         # Sort segments by their start time
         sorted_segments = sorted(response["segments"], key=lambda x: x["segment"][0])