iView handle missing episodes in series

[webdl] / iview.py

diff --git a/iview.py b/iview.py
index 55101d1f4950b5607aed928b74fd8435094a5eb8..e9c5904b1c51a5aadd3fc95f3f6630263d0d0e2a 100644 (file)
--- a/iview.py
+++ b/iview.py
@@ -68,7 +68,7 @@ class IviewIndexNode(Node):
         info = grab_json(self.url)
         for key in ["carousels", "collections", "index"]:
             for collection_list in info[key]:
-                for ep_info in collection_list["episodes"]:
+                for ep_info in collection_list.get("episodes", []):
                     self.add_series(ep_info)
 
     def add_series(self, ep_info):
@@ -89,7 +89,7 @@ class IviewSeriesNode(Node):
         series_slug = ep_info["href"].split("/")[1]
         series_url = API_URL + "/series/" + series_slug + "/" + ep_info["seriesHouseNumber"]
         info = grab_json(series_url)
-        for ep_info in info["episodes"]:
+        for ep_info in info.get("episodes", []):
             add_episode(self, ep_info)
 
 class IviewFlatNode(Node):