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

Add iterable and sized protocols to fetchers

Browse source
This commit is contained in:
Joakim Soderlund 2019-05-25 22:15:00 +02:00
parent 7a9191cf36
commit cbee5ead0d
2 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
fimfarchive/fetchers/directory.py
View file

@ -25,7 +25,7 @@ Directory fetcher.
import json
from itertools import chain
from pathlib import Path
from typing import Any, Dict, Iterable, Iterator, Optional, Set, Union
from typing import Any, Dict, Iterable, Iterator, Optional, Set, Sized, Union
from fimfarchive.exceptions import InvalidStoryError, StorySourceError
from fimfarchive.flavors import Flavor
@ -40,7 +40,7 @@ __all__ = (
)
class DirectoryFetcher(Fetcher):
class DirectoryFetcher(Iterable[Story], Sized, Fetcher):
"""
Fetches stories from file system.
"""

6
fimfarchive/fetchers/fimfarchive.py
View file

@ -27,8 +27,8 @@ import marshal
from io import BufferedReader
from multiprocessing import Pool
from typing import (
cast, Any, Callable, Dict, IO, Iterator,
Mapping, Optional, Tuple, Union,
cast, Any, Callable, Dict, IO, Iterable, Iterator,
Mapping, Optional, Sized, Tuple, Union,
)
from zipfile import ZipFile, BadZipFile
@ -144,7 +144,7 @@ class MemoryIndex(Index):
self.data.clear()
class FimfarchiveFetcher(Fetcher):
class FimfarchiveFetcher(Iterable[Story], Sized, Fetcher):
"""
Fetcher for Fimfarchive.
"""