From 396bce25d3e3eeff8cdaaa13f5f68ce454829ae4 Mon Sep 17 00:00:00 2001
From: Vishnunarayan K I <appukuttancr@gmail.com>
Date: Mon, 23 Sep 2019 16:03:04 +0530
Subject: [PATCH] fix: add `__iter__` to anime class

This is done so that range in cli works.
How it was working before, I have no clue. We need proper tests to catch
stuff like these.
---
 anime_downloader/sites/anime.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/anime_downloader/sites/anime.py b/anime_downloader/sites/anime.py
index 65f8c51..a057553 100644
--- a/anime_downloader/sites/anime.py
+++ b/anime_downloader/sites/anime.py
@@ -192,6 +192,11 @@ class Anime:
             return anime
         return None
 
+    def __iter__(self):
+        episode_class = AnimeEpisode.subclasses[self.sitename]
+        for ep_id in self._episode_urls:
+            yield episode_class(ep_id[1], parent=self, ep_no=ep_id[0])
+
     def __repr__(self):
         return '''
 Site: {name}