bitbot-3.11-fork/modules/spotify.py

25 lines
930 B
Python

import json
from src import ModuleManager, Utils
URL_SPOTIFY = "https://api.spotify.com/v1/search"
class Module(ModuleManager.BaseModule):
@Utils.hook("received.command.spotify", min_args=1)
def spotify(self, event):
"""
Search for a track on spotify
"""
page = Utils.get_url(URL_SPOTIFY, get_params={"type": "track",
"limit": 1, "q": event["args"]}, json=True)
if page:
if len(page["tracks"]["items"]):
item = page["tracks"]["items"][0]
title = item["name"]
artist_name = item["artists"][0]["name"]
url = item["external_urls"]["spotify"]
event["stdout"].write("%s (by %s) %s" % (title, artist_name,
url))
else:
event["stderr"].write("No results found")
else:
event["stderr"].write("Failed to load results")