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.
master
Vishnunarayan K I 2019-09-23 16:03:04 +05:30
parent d7756a8fcc
commit 396bce25d3
1 changed files with 5 additions and 0 deletions

View File

@ -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}