73 lines
1.9 KiB
Python
73 lines
1.9 KiB
Python
import clementine
|
|
|
|
from PythonQt.QtCore import QUrl
|
|
from PythonQt.QtNetwork import QNetworkRequest
|
|
|
|
import json
|
|
import logging
|
|
import urllib
|
|
|
|
LOGGER = logging.getLogger("google_images")
|
|
|
|
|
|
class GoogleImagesCoverProvider(clementine.CoverProvider):
|
|
API_URL = 'https://ajax.googleapis.com/ajax/services/search/images?{0}'
|
|
|
|
def __init__(self, parent=None):
|
|
clementine.CoverProvider.__init__(self, "Google Images", parent)
|
|
|
|
self.api_args = {
|
|
'v' : '1.0',
|
|
# at most five results
|
|
'rsz' : '5',
|
|
# only larger sizes
|
|
'imgsz' : 'large|xlarge'
|
|
}
|
|
self.network = clementine.NetworkAccessManager()
|
|
|
|
def StartSearch(self, artist, album, id):
|
|
url = self.GetQueryURL(artist + " " + album)
|
|
LOGGER.info("Id %d - sending request to '%s'" % (id, url))
|
|
|
|
reply = self.network.get(QNetworkRequest(url))
|
|
|
|
def QueryFinished():
|
|
LOGGER.debug("Id %d - finished" % id)
|
|
|
|
self.SearchFinished(id, self.ParseReply(artist, album, reply))
|
|
|
|
reply.connect("finished()", QueryFinished)
|
|
return True
|
|
|
|
def ParseReply(self, artist, album, reply):
|
|
results = json.loads(str(reply.readAll()))
|
|
|
|
parsed = []
|
|
|
|
if "responseStatus" not in results or results["responseStatus"] != 200:
|
|
LOGGER.warning("Error parsing reply: %s", results["responseDetails"])
|
|
return parsed
|
|
|
|
query = "%s - %s" % (artist, album)
|
|
|
|
LOGGER.info("Parsing reply for query '%s'" % query)
|
|
for result in results['responseData']['results']:
|
|
current = clementine.CoverSearchResult()
|
|
|
|
current.description = query
|
|
current.image_url = result['url']
|
|
|
|
parsed.append(current)
|
|
|
|
return parsed
|
|
|
|
def GetQueryURL(self, query):
|
|
current_args = self.api_args.copy()
|
|
current_args['q'] = query
|
|
|
|
return QUrl.fromEncoded(self.API_URL.format(urllib.urlencode(current_args)))
|
|
|
|
|
|
provider = GoogleImagesCoverProvider()
|
|
clementine.cover_providers.AddProvider(provider)
|