From 4352283318c346e57cc466ec48a2187f8768a458 Mon Sep 17 00:00:00 2001
From: Casey Link <unnamedrambler@gmail.com>
Date: Sat, 5 Feb 2011 13:03:23 -0500
Subject: [PATCH] add tasteometer command to determine users' musical
 compatibility

---
 modules/lastfm.py | 53 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 53 insertions(+)

diff --git a/modules/lastfm.py b/modules/lastfm.py
index 916c75a..e33ca8f 100644
--- a/modules/lastfm.py
+++ b/modules/lastfm.py
@@ -121,6 +121,59 @@ def now_playing(phenny, input):
 
 now_playing.commands = ['np']
 
+def tasteometer(phenny, input):
+    input1 = input.group(2)
+    input2 = input.group(3)
+    user1 = resolve_username(input1)
+    if not user1:
+        user1 = input1
+    user2 = resolve_username(input2)
+    if not user2:
+        user2 = input2
+    try:
+        req = urlopen("%smethod=tasteometer.compare&type1=user&type2=user&value1=%s&value2=%s" % (APIURL, urlquote(user1), urlquote(user2)))
+    except HTTPError, e:
+        if e.code == 400:
+            phenny.say("uhoh, someone doesn't exist on last.fm, perhaps they need to set user")
+            return
+        else:
+            phenny.say("uhoh. try again later, mmkay?")
+            return
+    doc = etree.parse(req)
+    root = doc.getroot()
+    score = root.xpath('comparison/result/score')
+    if len(score) == 0:
+        phenny.say("something isn't right. have those users scrobbled?")
+
+    score = float(score[0].text)
+    rating = ""
+    if score >= 0.9:
+        rating = "Super"
+    elif score >= 0.7:
+        rating = "Very High"
+    elif score >= 0.5:
+        rating = "High"
+    elif score >= 0.3:
+        rating = "Medium"
+    elif score >= 0.1:
+        rating = "Low"
+    else:
+        rating = "Very Low"
+
+    artists = root.xpath("comparison/result/artists/artist/name")
+    common_artists = ""
+    if len(artists) == 0:
+        common_artists = ". they don't have any artists in common."
+
+    names = []
+    map(lambda a: names.append(a.text) ,artists)
+    common_artists = "and music they have in common includes: %s" % ", ".join(names)
+
+
+    phenny.say("%s's and %s's musical compatibility rating is %s %s" % (user1, user2, rating, common_artists))
+
+tasteometer.rule = (['taste'], r'(\S+)\s+(\S+)')
+
 def save_config():
     configfile  = open(config_filename, 'wb')
     config.write(configfile)