bitbot-3.11-fork/modules/soundcloud.py

63 lines
2 KiB
Python

#--require-config soundcloud-api-key
import json, re, time
from src import Utils
URL_SOUNDCLOUD_TRACK = "http://api.soundcloud.com/tracks"
URL_SOUNDCLOUD_RESOLVE = "http://api.soundcloud.com/resolve"
REGEX_SOUNDCLOUD = "https?://soundcloud.com/([^/]+)/([^/]+)"
class Module(object):
_name = "SoundCloud"
def __init__(self, bot, events, exports):
self.bot = bot
@Utils.hook("received.command.soundcloud|sc")
def soundcloud(self, event):
"""
Search SoundCloud
"""
query = None
url = None
if event["args"]:
match = re.search(REGEX_SOUNDCLOUD, event["args"])
if match:
url = match.string
else:
query = event["args"]
else:
last_soundcloud = event["buffer"].find(REGEX_SOUNDCLOUD)
if last_soundcloud:
url = re.match(REGEX_SOUNDCLOUD,
last_soundcloud.message).string
if not query and not url:
event["stderr"].write("no search phrase provided")
return
has_query = not query == None
get_params = {"limit": 1,
"client_id": self.bot.config["soundcloud-api-key"]}
if query:
get_params["q"] = query
else:
get_params["url"] = url
page = Utils.get_url(
URL_SOUNDCLOUD_TRACK if has_query else URL_SOUNDCLOUD_RESOLVE,
get_params=get_params, json=True)
if page:
page = page[0] if has_query else page
title = page["title"]
user = page["user"]["username"]
duration = time.strftime("%H:%M:%S", time.gmtime(page[
"duration"]/1000))
if duration.startswith("00:"):
duration = duration[3:]
link = page["permalink_url"]
event["stdout"].write("%s [%s] (posted by %s) %s" % (title,
duration, user, link))
else:
event["stderr"].write("Failed to load results")