From cc7eb41ac7964ce62ef73fedc134538d63ad321d Mon Sep 17 00:00:00 2001 From: Hellow2 Date: Thu, 30 Mar 2023 12:31:37 +0200 Subject: [PATCH] made the direct downloads work --- src/actual_donwload.py | 11 ++++++++++- src/music_kraken/objects/source.py | 3 +++ .../pages/download_center/page_attributes.py | 2 +- 3 files changed, 14 insertions(+), 2 deletions(-) diff --git a/src/actual_donwload.py b/src/actual_donwload.py index 2aa5103..882b73d 100644 --- a/src/actual_donwload.py +++ b/src/actual_donwload.py @@ -17,7 +17,16 @@ def search_pages(): search.choose(0) print(search) + +def direct_download(): + search = pages.Search() + + search.search_url("https://www.metal-archives.com/bands/Ghost_Bath/3540372489") + print(search) + + search.search_url("https://musify.club/artist/ghost-bath-280348") + print(search) if __name__ == "__main__": - music_kraken.cli() + direct_download() diff --git a/src/music_kraken/objects/source.py b/src/music_kraken/objects/source.py index b48cb64..901a623 100644 --- a/src/music_kraken/objects/source.py +++ b/src/music_kraken/objects/source.py @@ -83,6 +83,9 @@ class Source(DatabaseObject): """ parsed = urlparse(url) url = parsed.geturl() + + if "musify" in parsed.netloc: + return cls(SourcePages.MUSIFY, url) if url.startswith("https://www.youtube"): return cls(SourcePages.YOUTUBE, url) diff --git a/src/music_kraken/pages/download_center/page_attributes.py b/src/music_kraken/pages/download_center/page_attributes.py index b5fd4f0..9989d01 100644 --- a/src/music_kraken/pages/download_center/page_attributes.py +++ b/src/music_kraken/pages/download_center/page_attributes.py @@ -7,7 +7,7 @@ from ..musify import Musify NAME_PAGE_MAP: Dict[str, Type[Page]] = dict() PAGE_NAME_MAP: Dict[Type[Page], str] = dict() -SOURCE_PAGE_MAP: Dict[SourcePages, Type[Page]] +SOURCE_PAGE_MAP: Dict[SourcePages, Type[Page]] = dict() ALL_PAGES: Tuple[Type[Page]] = ( EncyclopaediaMetallum,