Backup_Repos/beets
1
0
Fork 0
mirror of https://github.com/beetbox/beets.git synced 2026-01-05 23:43:31 +01:00
Code Issues Projects Releases Wiki Activity

fetchart: Add new google backend using google custom search

Browse source 
Requires an API key from google similar to lyrics plugin.
Limited to 100 queries/day.
This commit is contained in:
Lachlan Charlick 2015-12-28 23:16:12 +10:30
parent b25b4db9e1
commit c14abe56a5
1 changed files with 46 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

50
beetsplug/fetchart.py
View file

@ -96,8 +96,9 @@ class RequestMixin(object):
# ART SOURCES ################################################################
class ArtSource(RequestMixin):
def __init__(self, log):
def __init__(self, log, config):
self._log = log
self._config = config
def get(self, album):
raise NotImplementedError()
@ -157,6 +158,43 @@ class AlbumArtOrg(ArtSource):
self._log.debug(u'no image found on page')
class GoogleImages(ArtSource):
URL = u'https://www.googleapis.com/customsearch/v1'
def get(self, album):
"""Return art URL from google custom search engine
given an album title and interpreter.
"""
self._log.debug('fetching art from google...')
api_key = self._config['google_api_key'].get()
engine_id = self._config['google_engine_ID'].get()
self._log.debug('API Key: ' + api_key)
self._log.debug('Engine ID: ' + engine_id)
if not api_key:
self._log.debug(u'google api key not set')
return
if not (album.albumartist and album.album):
return
search_string = (album.albumartist + ',' + album.album).encode('utf-8')
response = self.request(self.URL, params={
'key': api_key,
'cx': engine_id,
'q': search_string,
'searchType': 'image'
})
# Get results using JSON.
data = response.json()
if 'error' in data:
reason = data['error']['errors'][0]['reason']
self._log.debug(u'google fetchart error: {0}', reason)
return
if 'items' in data.keys():
for item in data['items']:
yield item['link']
class ITunesStore(ArtSource):
# Art from the iTunes Store.
def get(self, album):
@ -361,7 +399,7 @@ class FileSystem(ArtSource):
# Try each source in turn.
SOURCES_ALL = [u'coverart', u'itunes', u'amazon', u'albumart',
u'wikipedia']
u'wikipedia', u'google']
ART_SOURCES = {
u'coverart': CoverArtArchive,
@ -369,6 +407,7 @@ ART_SOURCES = {
u'albumart': AlbumArtOrg,
u'amazon': Amazon,
u'wikipedia': Wikipedia,
u'google': GoogleImages,
}
# PLUGIN LOGIC ###############################################################
@ -387,7 +426,10 @@ class FetchArtPlugin(plugins.BeetsPlugin, RequestMixin):
'cautious': False,
'cover_names': ['cover', 'front', 'art', 'album', 'folder'],
'sources': ['coverart', 'itunes', 'amazon', 'albumart'],
'google_API_key': None,
'google_engine_ID': u'001442825323518660753:hrh5ch1gjzm',
})
self.config['google_API_key'].redact = True
# Holds paths to downloaded images between fetching them and
# placing them in the filesystem.
@ -407,8 +449,8 @@ class FetchArtPlugin(plugins.BeetsPlugin, RequestMixin):
available_sources.remove(u'itunes')
sources_name = plugins.sanitize_choices(
self.config['sources'].as_str_seq(), available_sources)
self.sources = [ART_SOURCES[s](self._log) for s in sources_name]
self.fs_source = FileSystem(self._log)
self.sources = [ART_SOURCES[s](self._log, self.config) for s in sources_name]
self.fs_source = FileSystem(self._log, self.config)
# Asynchronous; after music is added to the library.
def fetch_art(self, session, task):