Backup_Repos/beets
1
0
Fork 0
mirror of https://github.com/beetbox/beets.git synced 2025-12-15 21:14:19 +01:00
Code Issues Projects Releases Wiki Activity

fetchart: add available() method to sources

Browse source 
in order to move (some of) the source-specific code out of plugin
initialization and into the source classes
This commit is contained in:
wisp3rwind 2023-04-30 10:04:54 +02:00
parent 843fbcbaa0
commit 254bb297c8
1 changed files with 33 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

52
beetsplug/fetchart.py
View file

@ -254,6 +254,13 @@ class ArtSource(RequestMixin):
self._config = config
self.match_by = match_by or self.VALID_MATCHING_CRITERIA
@classmethod
def available(cls, log, config):
"""Return whether or not all dependencies are met and the art source is
in fact usable.
"""
return True
def get(self, album, plugin, paths):
raise NotImplementedError()
@ -466,6 +473,13 @@ class GoogleImages(RemoteArtSource):
self.key = self._config['google_key'].get(),
self.cx = self._config['google_engine'].get(),
@classmethod
def available(cls, log, config):
has_key = bool(config['google_key'].get())
if not has_key:
log.debug("google: Disabling art source due to missing key")
return has_key
def get(self, album, plugin, paths):
"""Return art URL from google custom search engine
given an album title and interpreter.
@ -855,6 +869,13 @@ class LastFM(RemoteArtSource):
super().__init__(*args, **kwargs)
self.key = self._config['lastfm_key'].get(),
@classmethod
def available(cls, log, config):
has_key = bool(config['lastfm_key'].get())
if not has_key:
log.debug("lastfm: Disabling art source due to missing key")
return has_key
def get(self, album, plugin, paths):
if not album.mb_albumid:
return
@ -901,6 +922,14 @@ class Spotify(RemoteArtSource):
SPOTIFY_ALBUM_URL = 'https://open.spotify.com/album/'
@classmethod
def available(cls, log, config):
if not HAS_BEAUTIFUL_SOUP:
log.debug('To use Spotify as an album art source, '
'you must install the beautifulsoup4 module. See '
'the documentation for further details.')
return HAS_BEAUTIFUL_SOUP
def get(self, album, plugin, paths):
url = self.SPOTIFY_ALBUM_URL + album.mb_albumid
try:
@ -922,9 +951,6 @@ class Spotify(RemoteArtSource):
return
# Try each source in turn.
SOURCES_ALL = ['filesystem', 'coverart', 'itunes', 'amazon', 'albumart',
'wikipedia', 'google', 'fanarttv', 'lastfm', 'spotify']
ART_SOURCES = {
'filesystem': FileSystem,
'coverart': CoverArtArchive,
@ -1017,22 +1043,10 @@ class FetchArtPlugin(plugins.BeetsPlugin, RequestMixin):
self.import_stages = [self.fetch_art]
self.register_listener('import_task_files', self.assign_art)
available_sources = list(SOURCES_ALL)
if not self.config['google_key'].get() and \
'google' in available_sources:
available_sources.remove('google')
if not self.config['lastfm_key'].get() and \
'lastfm' in available_sources:
available_sources.remove('lastfm')
if not HAS_BEAUTIFUL_SOUP and \
'spotify' in available_sources:
self._log.debug('To use Spotify as an album art source, '
'you must install the beautifulsoup4 module. See '
'the documentation for further details.')
available_sources.remove('spotify')
available_sources = [(s, c)
for s in available_sources
for c in ART_SOURCES[s].VALID_MATCHING_CRITERIA]
available_sources = [(s_name, c)
for (s_name, s_cls) in ART_SOURCES.items()
if s_cls.available(self._log, self.config)
for c in s_cls.VALID_MATCHING_CRITERIA]
sources = plugins.sanitize_pairs(
self.config['sources'].as_pairs(default_value='*'),
available_sources)