maloja/external.py

import urllib.parse, urllib.request
import json
import base64
from doreah.settings import get_settings
from doreah.logging import log


apis_artists = [
	{
		"name":"LastFM + Fanart.tv",
		"check":get_settings("LASTFM_API_KEY") not in [None,"ASK"] and get_settings("FANARTTV_API_KEY") not in [None,"ASK"],
		"steps":[
			("get","http://ws.audioscrobbler.com/2.0/?method=artist.getinfo&artist={artiststring}&api_key=" + get_settings("LASTFM_API_KEY") + "&format=json"),
			("parse",["artist","mbid"]),
			("get","http://webservice.fanart.tv/v3/music/{var}?api_key=" + get_settings("FANARTTV_API_KEY")),
			("parse",["artistthumb",0,"url"])
		]
	},
	{
		"name":"Spotify",
		"check":get_settings("SPOTIFY_API_ID") not in [None,"ASK"] and get_settings("SPOTIFY_API_SECRET") not in [None,"ASK"],
		"steps":[
			("post","https://accounts.spotify.com/api/token",{"Authorization":"Basic " + base64.b64encode(bytes(get_settings("SPOTIFY_API_ID") + ":" + get_settings("SPOTIFY_API_SECRET"),encoding="utf-8")).decode("utf-8")},{"grant_type":"client_credentials"}),
			("parse",["access_token"]),
			("get","https://api.spotify.com/v1/search?q={artiststring}&type=artist&access_token={var}"),
			("parse",["artists","items",0,"images",0,"url"])
		]
	}
]

apis_tracks = [
	{
		"name":"LastFM",
		"check":get_settings("LASTFM_API_KEY") not in [None,"ASK"],
		"steps":[
			("get","https://ws.audioscrobbler.com/2.0/?method=track.getinfo&track={titlestring}&artist={artiststring}&api_key=" + get_settings("LASTFM_API_KEY") + "&format=json"),
			("parse",["track","album","image",3,"#text"])
		]
	},
	{
		"name":"Spotify",
		"check":get_settings("SPOTIFY_API_ID") not in [None,"ASK"] and get_settings("SPOTIFY_API_SECRET") not in [None,"ASK"],
		"steps":[
			("post","https://accounts.spotify.com/api/token",{"Authorization":"Basic " + base64.b64encode(bytes(get_settings("SPOTIFY_API_ID") + ":" + get_settings("SPOTIFY_API_SECRET"),encoding="utf-8")).decode("utf-8")},{"grant_type":"client_credentials"}),
			("parse",["access_token"]),
			("get","https://api.spotify.com/v1/search?q={artiststring}%20{titlestring}&type=track&access_token={var}"),
			("parse",["tracks","items",0,"album","images",0,"url"])
		]
	}
]

def api_request_artist(artist):
	for api in apis_artists:
		if api["check"]:
			log("API: " + api["name"] + "; Image request: " + artist,module="external")
			try:
				artiststring = urllib.parse.quote(artist)
				var = artiststring
				for step in api["steps"]:
					if step[0] == "get":
						response = urllib.request.urlopen(step[1].format(artiststring=artiststring,var=var))
						var = json.loads(response.read())
					elif step[0] == "post":
						keys = {
							"url":step[1].format(artiststring=artiststring,var=var),
							"method":"POST",
							"headers":step[2],
							"data":bytes(urllib.parse.urlencode(step[3]),encoding="utf-8")
						}
						req = urllib.request.Request(**keys)
						response = urllib.request.urlopen(req)
						var = json.loads(response.read())
					elif step[0] == "parse":
						for node in step[1]:
							var = var[node]
				assert isinstance(var,str) and var != ""
			except:
				continue

			return var
		else:
			pass
	return None

def api_request_track(track):
	artists, title = track
	for api in apis_tracks:
		if api["check"]:
			log("API: " + api["name"] + "; Image request: " + "/".join(artists) + " - " + title,module="external")
			try:
				artiststring = urllib.parse.quote(", ".join(artists))
				titlestring = urllib.parse.quote(title)
				var = artiststring + titlestring
				for step in api["steps"]:
					if step[0] == "get":
						response = urllib.request.urlopen(step[1].format(artiststring=artiststring,titlestring=titlestring,var=var))
						var = json.loads(response.read())
					elif step[0] == "post":
						keys = {
							"url":step[1].format(artiststring=artiststring,titlestring=titlestring,var=var),
							"method":"POST",
							"headers":step[2],
							"data":bytes(urllib.parse.urlencode(step[3]),encoding="utf-8")
						}
						req = urllib.request.Request(**keys)
						response = urllib.request.urlopen(req)
						var = json.loads(response.read())
					elif step[0] == "parse":
						for node in step[1]:
							var = var[node]
				assert isinstance(var,str) and var != ""
			except:
				if len(artists) != 1:
					# try the same track with every single artist
					for a in artists:
						result = api_request_track(([a],title))
						if result is not None:
							return result
				continue

			return var
		else:
			pass

	return None