lastfm.py

import urllib2
import urllib
import urlparse
import xml.etree.ElementTree
import re
from htmlentitydefs import name2codepoint
import hashlib
import sys
import time
import os

key="5071d44086caaeb4dce9fba2053ac768"
secret="4c2b1b277e94f73d50429f891185db3f"

if os.path.exists(os.path.expanduser("~/.lastfmsess")):
	sessfp = open(os.path.expanduser("~/.lastfmsess"), "r")
	session = sessfp.read().strip()
	sessfp.close()

def htmlentitydecode(s):
    os= re.sub('&(%s);' % '|'.join(name2codepoint), 
            lambda m: unichr(name2codepoint[m.group(1)]), s)
    return os

def clean_trackid(trackid):
	m=re.match(".*([a-fA-Z0-9]{8}-[a-fA-Z0-9]{4}-[a-fA-Z0-9]{4}-[a-fA-Z0-9]{4}-[a-fA-Z0-9]{12}).*",trackid)
	assert m
	mbid=m.group(1)
	return mbid

def _cleanname(x):
	if x is None:
		return ''
	return htmlentitydecode(x)

def _etree_to_dict(etree):
	result={}
	for i in etree:
		if i.tag not in result:
			result[i.tag]=[]
		if len(i):
			result[i.tag].append(_etree_to_dict(i))
		else:
			result[i.tag].append(_cleanname(i.text))
	return result

def _do_raw_lastfm_query(url):
	f = urllib2.Request(url)
	f.add_header('User-Agent','Scrobbyl')
	try:
		f = urllib2.urlopen(f)
	except urllib2.URLError, e:
		raise

	tree = xml.etree.ElementTree.ElementTree(file=f)
	result=_etree_to_dict(tree.getroot())
	return result

def _do_lastfm_post(url, data):
	f = urllib2.Request(url)
	f.add_header('User-Agent','Scrobbyl')
	try:
		f = urllib2.urlopen(f, data)
	except urllib2.URLError, e:
		raise


def _do_lastfm_query(type, method,**kwargs):
	args = { 
		"method" : method,
		"api_key" : key,
	 	}
	for k,v in kwargs.items():
		args[k] = v.encode("utf8")
	s = ""
	for k in sorted(args.keys()):
		s+=k+args[k]
	s+=secret
	if "sk" in args.keys() or "token" in args.keys():
		args["api_sig"] = hashlib.md5(s).hexdigest()

	if type == "GET":
		url=urlparse.urlunparse(('http',
			'ws.audioscrobbler.com',
			'/2.0/',
			'',
			urllib.urlencode(args),
			''))
		return _do_raw_lastfm_query(url)
	elif type == "POST":
		url=urlparse.urlunparse(('http',
			'ws.audioscrobbler.com',
			'/2.0/', '', '', ''))
		_do_lastfm_post(url, urllib.urlencode(args))

def _get_auth_token():
	token = _do_lastfm_query("GET", "auth.getToken")
	return token["token"][0]

def _make_session(token):
	sess = _do_lastfm_query("GET", "auth.getSession", token=token)
	key = sess["session"][0]["key"][0]
	fp = open(os.path.expanduser("~/.lastfmsess"), "w")
	fp.write(key)
	fp.close()
	print "session successfully created. thank you."

def scrobble(artist, track):
	# OK, not the real start time. but that'll do
	ts = "%d" % (time.time() - 100)
	_do_lastfm_query("POST", "track.scrobble", timestamp=ts, artist=artist, track=track, sk=session)

if __name__=="__main__":
	if len(sys.argv) > 1:
		if sys.argv[1] == "auth":
			t = _get_auth_token()
			print "Please open this url then come back to this window: http://www.last.fm/api/auth/?api_key=%s&token=%s" % (key, t)
			i = 0
			while i < 5:
				try:
					_make_session(t)
					break
				except urllib2.HTTPError ,e:
					print e.code
				time.sleep(10)
	else:
		scrobble("The New Pornographers", "Failsafe")