chore: better error message if index out of range

anime-dl · Aug 7, 2019 · 16ea97d · 16ea97d
1 parent d5eb40f
commit 16ea97d
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/anime_downloader/sites/anime.py b/anime_downloader/sites/anime.py
@@ -177,12 +177,18 @@ def get_data(self):
     def __getitem__(self, index):
         episode_class = AnimeEpisode.subclasses[self.sitename]
         if isinstance(index, int):
-            ep_id = self._episode_urls[index]
+            try:
+                ep_id = self._episode_urls[index]
+            except IndexError as e:
+                raise RuntimeError("No episode found with index") from e
             return episode_class(ep_id[1], parent=self,
                                  ep_no=ep_id[0])
         elif isinstance(index, slice):
             anime = copy.deepcopy(self)
-            anime._episode_urls = anime._episode_urls[index]
+            try:
+                anime._episode_urls = anime._episode_urls[index]
+            except IndexError as e:
+                raise RuntimeError("No episode found with index") from e
             return anime
         return None