extractors/collection/base: Reformat code

master
Felix Stupp 3 years ago
parent ed8fc4439e
commit a1b64d8644
Signed by: zocker
GPG Key ID: 93E1BD26F6B02FB7

@ -5,7 +5,11 @@ import logging
import math import math
from typing import Optional, TypeVar from typing import Optional, TypeVar
from ...models import CollectionUriMapping, MediaCollection, MediaElement from ...models import (
CollectionUriMapping,
MediaCollection,
MediaElement,
)
from ..generic import ExtractedData, ExtractionError, GeneralExtractor from ..generic import ExtractedData, ExtractionError, GeneralExtractor
from ..media import media_extract_uri from ..media import media_extract_uri
@ -14,7 +18,6 @@ T = TypeVar("T")
class CollectionExtractor(GeneralExtractor[MediaCollection, T]): class CollectionExtractor(GeneralExtractor[MediaCollection, T]):
@staticmethod @staticmethod
def check_uri(uri: str) -> Optional[MediaCollection]: def check_uri(uri: str) -> Optional[MediaCollection]:
mapping: CollectionUriMapping = CollectionUriMapping.get(uri=uri) mapping: CollectionUriMapping = CollectionUriMapping.get(uri=uri)
@ -23,8 +26,8 @@ class CollectionExtractor(GeneralExtractor[MediaCollection, T]):
elem: MediaCollection = MediaCollection.get(uri=uri) elem: MediaCollection = MediaCollection.get(uri=uri)
if elem: if elem:
logging.warning( logging.warning(
f"Add missing URI mapping entry for uri {uri!r}, " + f"Add missing URI mapping entry for uri {uri!r}, "
"this should not happen at this point and is considered a bug" + "this should not happen at this point and is considered a bug"
) )
elem.add_single_uri(uri) elem.add_single_uri(uri)
return elem return elem
@ -54,7 +57,13 @@ class CollectionExtractor(GeneralExtractor[MediaCollection, T]):
self.__configure_collection(collection) self.__configure_collection(collection)
return collection return collection
def _add_episode(self, collection: MediaCollection, uri: str, season: int = 0, episode: int = 0) -> Optional[MediaElement]: def _add_episode(
self,
collection: MediaCollection,
uri: str,
season: int = 0,
episode: int = 0,
) -> Optional[MediaElement]:
logging.debug(f"Add to collection {collection.title!r} media {uri!r} (Season {season}, Episode {episode})") logging.debug(f"Add to collection {collection.title!r} media {uri!r} (Season {season}, Episode {episode})")
try: try:
element = media_extract_uri(uri) element = media_extract_uri(uri)

Loading…
Cancel
Save