PonyMirrors/fimfarchive
1
0
Fork 0
mirror of https://github.com/JockeTF/fimfarchive.git synced 2024-11-22 05:17:59 +01:00
Code Issues Projects Releases Packages Wiki Activity

Make FimfarchiveFetcher iterable

Browse source
This commit is contained in:
Joakim Soderlund 2018-02-14 09:47:53 +01:00
parent cb58cfaa53
commit b916699127
1 changed files with 14 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
fimfarchive/fetchers/fimfarchive.py
View file

@ -31,6 +31,7 @@ from jmespath import compile as jmes
from fimfarchive.exceptions import InvalidStoryError, StorySourceError from fimfarchive.exceptions import InvalidStoryError, StorySourceError
from fimfarchive.flavors import StorySource, DataFormat, MetaPurity from fimfarchive.flavors import StorySource, DataFormat, MetaPurity
from fimfarchive.stories import Story
from fimfarchive.utils import Empty from fimfarchive.utils import Empty
from .base import Fetcher from .base import Fetcher
@ -140,6 +141,19 @@ class FimfarchiveFetcher(Fetcher):
except ValueError as e: except ValueError as e:
raise StorySourceError(f"Malformed index key: {key}") from e raise StorySourceError(f"Malformed index key: {key}") from e
def __len__(self) -> int:
"""
Returns the total number of stories in the archive.
"""
return len(self.index)
def __iter__(self) -> Iterable[Story]:
"""
Yields all stories in the archive, ordered by ID.
"""
for key in sorted(self.index.keys()):
yield self.fetch(key)
def validate(self, key: int) -> int: def validate(self, key: int) -> int:
""" """
Ensures that the key matches a valid story Ensures that the key matches a valid story