2011-04-02 15:47:51 +02:00
|
|
|
import clementine
|
|
|
|
|
|
|
|
from PyQt4.QtCore import QString, QUrl
|
|
|
|
from PyQt4.QtNetwork import QNetworkRequest
|
|
|
|
|
|
|
|
from simplejson import loads
|
|
|
|
import urllib
|
|
|
|
|
|
|
|
class GoogleImagesCoversScript(clementine.CoverProvider):
|
|
|
|
|
|
|
|
def __init__(self):
|
|
|
|
clementine.CoverProvider.__init__(self, "Google Images")
|
|
|
|
|
|
|
|
self.api_url = 'https://ajax.googleapis.com/ajax/services/search/images?{0}'
|
|
|
|
self.api_args = {
|
2011-04-06 17:24:56 +02:00
|
|
|
'v' : '1.0',
|
2011-04-02 15:47:51 +02:00
|
|
|
# at most five results
|
2011-04-06 17:24:56 +02:00
|
|
|
'rsz' : '5',
|
|
|
|
# only larger sizes
|
|
|
|
'imgsz' : 'large|xlarge'
|
2011-04-02 15:47:51 +02:00
|
|
|
}
|
|
|
|
self.network = clementine.NetworkAccessManager(self)
|
|
|
|
self.queries = {}
|
|
|
|
|
|
|
|
# register in the repository of cover providers
|
|
|
|
clementine.cover_providers.AddCoverProvider(self)
|
|
|
|
|
|
|
|
def SendRequest(self, query):
|
|
|
|
url = self.GetQueryURL(query)
|
|
|
|
|
|
|
|
reply = self.network.get(QNetworkRequest(url))
|
|
|
|
self.queries[reply] = query
|
|
|
|
|
|
|
|
return reply
|
|
|
|
|
|
|
|
def ParseReply(self, reply):
|
|
|
|
results = loads(str(reply.readAll()))
|
|
|
|
|
|
|
|
parsed = []
|
|
|
|
|
|
|
|
if 'Error' in results:
|
|
|
|
return parsed
|
|
|
|
|
|
|
|
query = self.queries.pop(reply)
|
|
|
|
for result in results['responseData']['results']:
|
|
|
|
current = clementine.CoverSearchResult()
|
|
|
|
current.description = query
|
|
|
|
current.image_url = QString(result['url'])
|
|
|
|
|
|
|
|
parsed.append(current)
|
|
|
|
|
|
|
|
return parsed
|
|
|
|
|
|
|
|
def GetQueryURL(self, query):
|
|
|
|
current_args = self.api_args.copy()
|
|
|
|
current_args['q'] = query
|
|
|
|
|
|
|
|
return QUrl(self.api_url.format(urllib.urlencode(current_args)))
|
|
|
|
|
|
|
|
|
|
|
|
script = GoogleImagesCoversScript()
|