music-kraken/music-kraken-core
3
2
Fork 0
Code Issues 10 Pull Requests 1 Packages Projects 2 Releases Wiki Activity

finished most search functionality from the metadata downloader

Browse Source
This commit is contained in:
Lars Noack 2022-10-14 16:29:13 +02:00
parent 9d9c7960c9
commit 2ecd18eec9
1 changed files with 60 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

88
src/metadata.py
View File

@ -4,6 +4,7 @@ musicbrainzngs.set_useragent("metadata receiver", "0.1", "https://github.com/HeI
KNOWN_KIND_OF_OPTIONS = ["artist", "release", "track"] KNOWN_KIND_OF_OPTIONS = ["artist", "release", "track"]
class Search: class Search:
def __init__(self, query: str = None, artist: str = None): def __init__(self, query: str = None, artist: str = None):
if query is None and artist is None: if query is None and artist is None:
@ -15,47 +16,72 @@ class Search:
# initial search # initial search
if query is not None: if query is not None:
self.set_options([musicbrainzngs.search_artists(query), musicbrainzngs.search_release_groups(query), musicbrainzngs.search_releases(query)]) self.set_options(self.Options([musicbrainzngs.search_artists(query), musicbrainzngs.search_releases(query),
musicbrainzngs.search_recordings(query)]))
elif artist is not None: elif artist is not None:
self.set_options([musicbrainzngs.search_artists(artist=artist)]) self.set_options(self.Options([musicbrainzngs.search_artists(artist=artist)]))
def browse_artist(self, artist): def browse_artist(self, artist: dict, limit: int = 25):
return options_sets = [
{"artist-list": [artist, ], "artist-count": 1},
musicbrainzngs.browse_releases(artist=artist["id"], limit=limit),
musicbrainzngs.browse_recordings(artist=artist["id"], limit=limit)
]
return self.set_options(self.Options(options_sets))
def browse_option(self, option, index: int): def browse_release(self, release: dict, limit: int = 25):
if not option.choose(index): options_sets = [
return False musicbrainzngs.browse_artists(release=release["id"], limit=limit),
{"release-list": [release, ], "release-count": 1},
musicbrainzngs.browse_recordings(release=release["id"], limit=limit)
]
return self.set_options(self.Options(options_sets))
self.current_chosen_option = option.get_current_option() def browse_track(self, track: dict, limit: int = 25):
option_kind = self.current_chosen_option['kind'] options_sets = [
if option_kind == "artist": musicbrainzngs.browse_artists(recording=track["id"], limit=limit),
return self.browse_artist(self.current_chosen_option) musicbrainzngs.browse_releases(recording=track["id"], limit=limit),
{"recording-list": [track, ], "recording-count": 1}
]
return self.set_options(self.Options(options_sets))
def choose(self, index): def choose(self, index, limit: int = 25, ignore_limit_for_tracklist: bool = True):
if not self.current_options.choose(index): if not self.current_options.choose(index):
return self.current_options return self.current_options
self.current_chosen_option = self.current_options.get_current_option()
kind = self.current_chosen_option['kind']
if kind == 'artist':
return self.browse_artist(self.current_chosen_option, limit=limit)
if kind == 'release':
release_limit = limit if not ignore_limit_for_tracklist else 100
return self.browse_release(self.current_chosen_option, limit=release_limit)
if kind == 'track':
track_limit = limit if not ignore_limit_for_tracklist else 100
return self.browse_track(self.current_chosen_option, limit=track_limit)
return self.current_options
def get_options(self): def get_options(self):
return self.current_options return self.current_options
def set_options(self, results): def set_options(self, option_instance):
option_instance = self.Options(results=results)
self.options_history.append(option_instance) self.options_history.append(option_instance)
self.current_options = option_instance self.current_options = option_instance
return option_instance return option_instance
def get_previous_options(self):
self.options_history.pop(-1)
self.current_options = self.options_history[-1]
return self.current_options
options = property(fget=get_options) options = property(fget=get_options)
class Options: class Options:
def __init__(self, results: list): def __init__(self, results: list):
self.results = results self.results = results
self.headers = {
"artist": "found {count} artists:\n"
}
self.artist_count = 0 self.artist_count = 0
self.release_count = 0 self.release_count = 0
self.track_count = 0 self.track_count = 0
@ -86,7 +112,6 @@ class Search:
def get_string_for_release(self, release: dict) -> str: def get_string_for_release(self, release: dict) -> str:
string = "" string = ""
print(release)
if "type" in release: if "type" in release:
string += f"the {release['type']} titled " string += f"the {release['type']} titled "
string += f"'{release['title']}'" string += f"'{release['title']}'"
@ -120,13 +145,12 @@ class Search:
if "artist-list" in option_set: if "artist-list" in option_set:
self.set_artist_values(option_set) self.set_artist_values(option_set)
continue continue
if "release-group-list" in option_set: if "release-list" in option_set:
self.set_release_values(option_set) self.set_release_values(option_set)
continue continue
if "release-list" in option_set: if "recording-list" in option_set:
self.set_track_values(option_set) self.set_track_values(option_set)
continue continue
print(option_set)
def set_artist_values(self, option_set: dict): def set_artist_values(self, option_set: dict):
self.artist_count += option_set['artist-count'] self.artist_count += option_set['artist-count']
@ -135,21 +159,27 @@ class Search:
self.result_list.append(artist) self.result_list.append(artist)
def set_release_values(self, option_set: dict): def set_release_values(self, option_set: dict):
self.release_count += option_set['release-group-count'] self.release_count += option_set['release-count']
for release in option_set['release-group-list']: for release in option_set['release-list']:
release['kind'] = "release" release['kind'] = "release"
self.result_list.append(release) self.result_list.append(release)
def set_track_values(self, option_set: dict): def set_track_values(self, option_set: dict):
self.artist_count += option_set['release-count'] self.track_count += option_set['recording-count']
for track in option_set['release-list']: for track in option_set['recording-list']:
track['kind'] = "track" track['kind'] = "track"
self.result_list.append(track) self.result_list.append(track)
def automated_demo():
search = Search(query="psychonaut 4")
print(search.options)
print(search.choose(0))
print(search.choose(2))
print(search.choose(4))
print(search.get_previous_options())
print(search.choose(6))
if __name__ == "__main__": if __name__ == "__main__":
search = Search(query="psychonaut 4") automated_demo()
print(search.options)
search.choose(0)