diff --git a/src/music_kraken/pages/download_center/download.py b/src/music_kraken/pages/download_center/download.py index 7cba7d6..d5e3902 100644 --- a/src/music_kraken/pages/download_center/download.py +++ b/src/music_kraken/pages/download_center/download.py @@ -6,12 +6,13 @@ from ...objects import Song, Album, Artist, Label, Source MusicObject = Union[Song, Album, Artist, Label] + class Download: def __init__( - self, - pages: Tuple[Type[Page]] = page_attributes.ALL_PAGES, - exclude_pages: Set[Type[Page]] = set(), - exclude_shady: bool = False, + self, + pages: Tuple[Type[Page]] = page_attributes.ALL_PAGES, + exclude_pages: Set[Type[Page]] = set(), + exclude_shady: bool = False, ) -> None: _page_list: List[Type[Page]] = [] _audio_page_list: List[Type[Page]] = [] @@ -29,7 +30,7 @@ class Download: self.pages: Tuple[Type[Page]] = tuple(_page_list) self.audio_pages: Tuple[Type[Page]] = tuple(_audio_page_list) - + def fetch_details(self, music_object: MusicObject) -> MusicObject: for page in self.pages: page.fetch_details(music_object=music_object) @@ -37,10 +38,8 @@ class Download: def fetch_source(self, source: Source) -> Optional[MusicObject]: source_page = page_attributes.SOURCE_PAGE_MAP[source.page_enum] - + if source_page not in self.pages: return - + return source_page.fetch_object_from_source(source) - - \ No newline at end of file