mirrors/streamrip
1
0
Fork 0
mirror of https://github.com/nathom/streamrip.git synced 2024-09-19 19:28:46 -04:00
Code Issues Projects Releases Packages Wiki Activity

Fix searching related issues

Browse source
This commit is contained in:
nathom 2021-07-29 23:09:25 -07:00
parent f98ee5c8f7
commit 328879584d
1 changed files with 8 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
rip/core.py
View file

@ -334,13 +334,13 @@ class RipCore(list):
if arguments["conversion"]["enabled"]: if arguments["conversion"]["enabled"]:
item.convert(**arguments["conversion"]) item.convert(**arguments["conversion"])
def scrape(self, featured_list: str): def scrape(self, featured_list: str, max_items: int = 500):
"""Download all of the items in a Qobuz featured list. """Download all of the items in a Qobuz featured list.
:param featured_list: The name of the list. See `rip discover --help`. :param featured_list: The name of the list. See `rip discover --help`.
:type featured_list: str :type featured_list: str
""" """
self.extend(self.search("qobuz", featured_list, "featured", limit=500)) self.extend(self.search("qobuz", featured_list, "featured", limit=max_items))
def get_client(self, source: str) -> Client: def get_client(self, source: str) -> Client:
"""Get a client given the source and log in. """Get a client given the source and log in.
@ -384,7 +384,7 @@ class RipCore(list):
self.prompt_creds(client.source) self.prompt_creds(client.source)
creds = self.config.creds(client.source) creds = self.config.creds(client.source)
except MissingCredentials: except MissingCredentials:
logger.debug("Credentials are missing. Prompting..", fg="yellow") logger.debug("Credentials are missing. Prompting..")
self.prompt_creds(client.source) self.prompt_creds(client.source)
creds = self.config.creds(client.source) creds = self.config.creds(client.source)
@ -499,13 +499,15 @@ class RipCore(list):
""" """
def try_search(source) -> Optional[Track]: def try_search(source) -> Optional[Track]:
if source == lastfm_fallback_source:
click.secho("using fallback", fg="red")
try: try:
query = QUERY_FORMAT[lastfm_source].format( query = QUERY_FORMAT[lastfm_source].format(
title=title, artist=artist title=title, artist=artist
) )
return next(self.search(source, query, media_type="track")) track = next(self.search(source, query, media_type="track"))
# Because the track is searched as a single we need to set
# this manually
track.part_of_tracklist = True
return track
except (NoResultsFound, StopIteration): except (NoResultsFound, StopIteration):
return None return None