bitbot-3.11-fork/modules/soundcloud.py

65 lines
2.1 KiB
Python

#--depends-on commands
#--require-config soundcloud-api-key
import json, re, time
from src import ModuleManager, utils
URL_SOUNDCLOUD_TRACK = "http://api.soundcloud.com/tracks"
URL_SOUNDCLOUD_RESOLVE = "http://api.soundcloud.com/resolve"
REGEX_SOUNDCLOUD = "https?://soundcloud.com/([^/]+)/([^/]+)"
class Module(ModuleManager.BaseModule):
_name = "SoundCloud"
@utils.hook("received.command.sc", alias_of="soundcloud")
@utils.hook("received.command.soundcloud")
def soundcloud(self, event):
"""
:help: Search SoundCloud
:usage: <term>
"""
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["target"].buffer.find(REGEX_SOUNDCLOUD)
if last_soundcloud:
url = last_soundcloud.match
if not query and not url:
raise utils.EventError("no search phrase provided")
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.http.request(
URL_SOUNDCLOUD_TRACK if has_query else URL_SOUNDCLOUD_RESOLVE,
get_params=get_params, json=True)
if page:
if len(page.data):
page = page.data[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("No results found")
else:
raise utils.EventResultsError()