completed new cli
This commit is contained in:
parent
308e34a91c
commit
4b4e05239f
@ -1,14 +1,15 @@
|
||||
import music_kraken
|
||||
from music_kraken import pages
|
||||
from music_kraken.download import Search
|
||||
from music_kraken.utils.enums.source import SourcePages
|
||||
from music_kraken.objects import Song, Target, Source, Album
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
music_kraken.cli(genre="test", command_list=[
|
||||
# "https://musify.club/release/molchat-doma-etazhi-2018-1092949",
|
||||
# "https://musify.club/release/ghost-bath-self-loather-2021-1554266",
|
||||
normally_download = [
|
||||
"s: #a Ghost Bath",
|
||||
"1",
|
||||
"d: 1, 5"
|
||||
]
|
||||
|
||||
])
|
||||
direct_download = [
|
||||
"d: https://musify.club/release/crystal-f-x-2012-795181"
|
||||
]
|
||||
|
||||
music_kraken.cli(genre="test", command_list=direct_download)
|
||||
|
@ -178,7 +178,9 @@ class Shell:
|
||||
|
||||
def print_current_options(self):
|
||||
self.page_dict = dict()
|
||||
|
||||
|
||||
print()
|
||||
|
||||
page_count = 0
|
||||
for option in self.current_results.formated_generator(max_items_per_page=self.max_displayed_options):
|
||||
if isinstance(option, Option):
|
||||
@ -191,7 +193,9 @@ class Shell:
|
||||
self.page_dict[option.__name__] = option
|
||||
|
||||
page_count += 1
|
||||
|
||||
|
||||
print()
|
||||
|
||||
def set_current_options(self, current_options: Results):
|
||||
self.current_results = current_options
|
||||
|
||||
@ -216,7 +220,8 @@ class Shell:
|
||||
|
||||
def search(self, query: str):
|
||||
if re.match(URL_PATTERN, query) is not None:
|
||||
self.set_current_options(*self.pages.fetch_url(re.match(URL_PATTERN, query)))
|
||||
page, data_object = self.pages.fetch_url(query)
|
||||
self.set_current_options(PageResults(page, data_object.options))
|
||||
self.print_current_options()
|
||||
return
|
||||
|
||||
@ -290,7 +295,7 @@ class Shell:
|
||||
to_download: List[DatabaseObject] = []
|
||||
|
||||
if re.match(URL_PATTERN, download_str) is not None:
|
||||
_, music_objects = self.pages.fetch_url(re.match(URL_PATTERN, download_str))
|
||||
_, music_objects = self.pages.fetch_url(download_str)
|
||||
to_download.append(music_objects)
|
||||
|
||||
else:
|
||||
@ -322,7 +327,7 @@ class Shell:
|
||||
print(music_object.option_string)
|
||||
print(result)
|
||||
|
||||
return False
|
||||
return True
|
||||
|
||||
def process_input(self, input_str: str) -> bool:
|
||||
input_str = input_str.strip()
|
||||
|
@ -36,6 +36,11 @@ class DatabaseObject:
|
||||
|
||||
self.build_version = -1
|
||||
|
||||
def __hash__(self):
|
||||
if self.dynamic:
|
||||
raise TypeError("Dynamic DatabaseObjects are unhashable.")
|
||||
return self.id
|
||||
|
||||
def __eq__(self, other) -> bool:
|
||||
if not isinstance(other, type(self)):
|
||||
return False
|
||||
|
@ -104,7 +104,7 @@ def merge_together(old_object: DatabaseObject, new_object: DatabaseObject) -> Da
|
||||
return old_object
|
||||
|
||||
|
||||
class Page():
|
||||
class Page:
|
||||
"""
|
||||
This is an abstract class, laying out the
|
||||
functionality for every other class fetching something
|
||||
@ -270,7 +270,7 @@ class Page():
|
||||
for collection_name in naming_music_object.UPWARDS_COLLECTION_ATTRIBUTES:
|
||||
collection: Collection = getattr(naming_music_object, collection_name)
|
||||
|
||||
if collection.empty():
|
||||
if collection.empty:
|
||||
continue
|
||||
if collection.element_type in naming_objects:
|
||||
continue
|
||||
@ -280,10 +280,10 @@ class Page():
|
||||
|
||||
fill_naming_objects(music_object)
|
||||
|
||||
return self._download(music_object, {}, genre, download_all)
|
||||
return self._download(music_object, naming_objects, download_all)
|
||||
|
||||
|
||||
def _download(self, music_object: DatabaseObject, naming_objects: Dict[Type[DatabaseObject], DatabaseObject], download_all: bool = False) -> list:
|
||||
def _download(self, music_object: DatabaseObject, naming_objects: Dict[Union[Type[DatabaseObject], str], DatabaseObject], download_all: bool = False) -> DownloadResult:
|
||||
# Skips all releases, that are defined in shared.ALBUM_TYPE_BLACKLIST, if download_all is False
|
||||
if isinstance(music_object, Album):
|
||||
if not download_all and music_object.album_type in shared.ALBUM_TYPE_BLACKLIST:
|
||||
@ -298,7 +298,7 @@ class Page():
|
||||
download_result: DownloadResult = DownloadResult()
|
||||
|
||||
for collection_name in music_object.DOWNWARDS_COLLECTION_ATTRIBUTES:
|
||||
collection: Collection = getattr(self, collection_name)
|
||||
collection: Collection = getattr(music_object, collection_name)
|
||||
|
||||
sub_ordered_music_object: DatabaseObject
|
||||
for sub_ordered_music_object in collection:
|
||||
@ -306,11 +306,12 @@ class Page():
|
||||
|
||||
return download_result
|
||||
|
||||
def _download_song(self, song: Song, naming_objects: Dict[Type[DatabaseObject], DatabaseObject]):
|
||||
def _download_song(self, song: Song, naming_objects: Dict[Type[DatabaseObject], Union[DatabaseObject, str]]):
|
||||
name_attribute = DEFAULT_VALUES.copy()
|
||||
|
||||
# song
|
||||
name_attribute["genre"] = naming_objects["genre"]
|
||||
if "genre" in naming_objects:
|
||||
name_attribute["genre"] = naming_objects["genre"]
|
||||
name_attribute["song"] = song.title
|
||||
|
||||
if Album in naming_objects:
|
||||
@ -344,7 +345,7 @@ class Page():
|
||||
file=str(random.randint(0, 999999))
|
||||
)
|
||||
|
||||
r = self._download_song_to_targets(source=sources[0], target=temp_target, desc=song.title)
|
||||
r = self.download_song_to_target(source=sources[0], target=temp_target, desc=song.title)
|
||||
|
||||
if not r.is_fatal_error:
|
||||
r.merge(self._post_process_targets(song, temp_target))
|
||||
|
@ -164,7 +164,6 @@ class Musify(Page):
|
||||
artist_thumbnail = image_soup.get("src")
|
||||
|
||||
return Artist(
|
||||
_id=_id,
|
||||
name=name,
|
||||
source_list=source_list
|
||||
)
|
||||
@ -269,7 +268,6 @@ class Musify(Page):
|
||||
self.LOGGER.warning("got an unequal ammount than 3 small elements")
|
||||
|
||||
return Album(
|
||||
_id=_id,
|
||||
title=title,
|
||||
source_list=source_list,
|
||||
date=ID3Timestamp(year=year),
|
||||
@ -316,7 +314,6 @@ class Musify(Page):
|
||||
_id = href.split("-")[-1]
|
||||
|
||||
artist_list.append(Artist(
|
||||
_id=_id,
|
||||
name=artist_anchor.get_text(strip=True),
|
||||
source_list=[artist_source]
|
||||
))
|
||||
@ -352,7 +349,6 @@ class Musify(Page):
|
||||
_id = raw_id
|
||||
|
||||
return Song(
|
||||
_id=_id,
|
||||
title=song_title,
|
||||
main_artist_list=artist_list,
|
||||
source_list=source_list
|
||||
@ -838,7 +834,6 @@ class Musify(Page):
|
||||
self.LOGGER.debug("there is not even 1 footer in the album card")
|
||||
|
||||
return Album(
|
||||
_id=_id,
|
||||
title=name,
|
||||
source_list=source_list,
|
||||
date=timestamp,
|
||||
@ -998,7 +993,7 @@ class Musify(Page):
|
||||
def fetch_label(self, source: Source, stop_at_level: int = 1) -> Label:
|
||||
return Label()
|
||||
|
||||
def download_song_to_targets(self, source: Source, target: Target, desc: str = None) -> DownloadResult:
|
||||
def download_song_to_target(self, source: Source, target: Target, desc: str = None) -> DownloadResult:
|
||||
"""
|
||||
https://musify.club/track/im-in-a-coffin-life-never-was-waste-of-skin-16360302
|
||||
https://musify.club/track/dl/16360302/im-in-a-coffin-life-never-was-waste-of-skin.mp3
|
||||
|
Loading…
Reference in New Issue
Block a user