disabled many plugins

disabled_stuff/rdio.py

@@ -0,0 +1,131 @@
+import urllib
+import json
+import re
+
+import oauth2 as oauth
+
+from util import hook
+
+
+def getdata(inp, types, api_key, api_secret):
+    consumer = oauth.Consumer(api_key, api_secret)
+    client = oauth.Client(consumer)
+    response = client.request('http://api.rdio.com/1/', 'POST',
+                              urllib.urlencode({'method': 'search', 'query': inp, 'types': types, 'count': '1'}))
+    data = json.loads(response[1])
+    return data
+
+
+@hook.command
+def rdio(inp, bot=None):
+    """ rdio <search term> - alternatives: .rdiot (track), .rdioar (artist), .rdioal (album) """
+    api_key = bot.config.get("api_keys", {}).get("rdio_key")
+    api_secret = bot.config.get("api_keys", {}).get("rdio_secret")
+    if not api_key:
+        return "error: no api key set"
+    data = getdata(inp, "Track,Album,Artist", api_key, api_secret)
+    try:
+        info = data['result']['results'][0]
+    except IndexError:
+        return "No results."
+    if 'name' in info:
+        if 'artist' in info and 'album' in info:  # Track
+            name = info['name']
+            artist = info['artist']
+            album = info['album']
+            url = info['shortUrl']
+            return u"\x02{}\x02 by \x02{}\x02 - {} {}".format(name, artist, album, url)
+        elif 'artist' in info and not 'album' in info:  # Album
+            name = info['name']
+            artist = info['artist']
+            url = info['shortUrl']
+            return u"\x02{}\x02 by \x02{}\x02 - {}".format(name, artist, url)
+        else:  # Artist
+            name = info['name']
+            url = info['shortUrl']
+            return u"\x02{}\x02 - {}".format(name, url)
+
+
+@hook.command
+def rdiot(inp, bot=None):
+    """ rdiot <search term> - Search for tracks on rdio """
+    api_key = bot.config.get("api_keys", {}).get("rdio_key")
+    api_secret = bot.config.get("api_keys", {}).get("rdio_secret")
+    if not api_key:
+        return "error: no api key set"
+    data = getdata(inp, "Track", api_key, api_secret)
+    try:
+        info = data['result']['results'][0]
+    except IndexError:
+        return "No results."
+    name = info['name']
+    artist = info['artist']
+    album = info['album']
+    url = info['shortUrl']
+    return u"\x02{}\x02 by \x02{}\x02 - {} - {}".format(name, artist, album, url)
+
+
+@hook.command
+def rdioar(inp, bot=None):
+    """ rdioar <search term> - Search for artists on rdio """
+    api_key = bot.config.get("api_keys", {}).get("rdio_key")
+    api_secret = bot.config.get("api_keys", {}).get("rdio_secret")
+    if not api_key:
+        return "error: no api key set"
+    data = getdata(inp, "Artist", api_key, api_secret)
+    try:
+        info = data['result']['results'][0]
+    except IndexError:
+        return "No results."
+    name = info['name']
+    url = info['shortUrl']
+    return u"\x02{}\x02 - {}".format(name, url)
+
+
+@hook.command
+def rdioal(inp, bot=None):
+    """ rdioal <search term> - Search for albums on rdio """
+    api_key = bot.config.get("api_keys", {}).get("rdio_key")
+    api_secret = bot.config.get("api_keys", {}).get("rdio_secret")
+    if not api_key:
+        return "error: no api key set"
+    data = getdata(inp, "Album", api_key, api_secret)
+    try:
+        info = data['result']['results'][0]
+    except IndexError:
+        return "No results."
+    name = info['name']
+    artist = info['artist']
+    url = info['shortUrl']
+    return u"\x02{}\x02 by \x02{}\x02 - {}".format(name, artist, url)
+
+
+rdio_re = (r'(.*:)//(rd.io|www.rdio.com|rdio.com)(:[0-9]+)?(.*)', re.I)
+
+
+@hook.regex(*rdio_re)
+def rdio_url(match, bot=None):
+    api_key = bot.config.get("api_keys", {}).get("rdio_key")
+    api_secret = bot.config.get("api_keys", {}).get("rdio_secret")
+    if not api_key:
+        return None
+    url = match.group(1) + "//" + match.group(2) + match.group(4)
+    consumer = oauth.Consumer(api_key, api_secret)
+    client = oauth.Client(consumer)
+    response = client.request('http://api.rdio.com/1/', 'POST',
+                              urllib.urlencode({'method': 'getObjectFromUrl', 'url': url}))
+    data = json.loads(response[1])
+    info = data['result']
+    if 'name' in info:
+        if 'artist' in info and 'album' in info:  # Track
+            name = info['name']
+            artist = info['artist']
+            album = info['album']
+            return u"Rdio track: \x02{}\x02 by \x02{}\x02 - {}".format(name, artist, album)
+        elif 'artist' in info and not 'album' in info:  # Album
+            name = info['name']
+            artist = info['artist']
+            return u"Rdio album: \x02{}\x02 by \x02{}\x02".format(name, artist)
+        else:  # Artist
+            name = info['name']
+            return u"Rdio artist: \x02{}\x02".format(name)