bitbot-3.11-fork/modules/lastfm.py

66 lines
2.6 KiB
Python
Raw Normal View History

#--require-config lastfm-api-key
2016-03-29 11:56:58 +00:00
import Utils
URL_SCROBBLER = "http://ws.audioscrobbler.com/2.0/"
class Module(object):
def __init__(self, bot, events, exports):
2016-03-29 11:56:58 +00:00
self.bot = bot
exports.add("set", {"setting": "lastfm",
"help": "Set username on last.fm"})
events.on("received.command").on("np", "listening", "nowplaying"
).hook(self.np, help="Get the last listened to track from a user",
usage="[username]")
2016-03-29 11:56:58 +00:00
def np(self, event):
if event["args_split"]:
lastfm_username = event["args_split"][0]
shown_username = lastfm_username
2016-03-29 11:56:58 +00:00
else:
2018-08-15 09:34:58 +00:00
lastfm_username = event["user"].get_setting("lastfm",
2016-03-29 11:56:58 +00:00
event["user"].nickname)
shown_username = event["user"].nickname
2016-03-29 11:56:58 +00:00
page = Utils.get_url(URL_SCROBBLER, get_params={
"method": "user.getrecenttracks", "user": lastfm_username,
2016-03-29 11:56:58 +00:00
"api_key": self.bot.config["lastfm-api-key"],
"format": "json", "limit": "1"}, json=True)
if page:
if "recenttracks" in page and len(page["recenttracks"
]["track"]):
now_playing = page["recenttracks"]["track"][0]
track_name = now_playing["name"]
artist = now_playing["artist"]["#text"]
2016-04-04 11:37:17 +00:00
info_page = Utils.get_url(URL_SCROBBLER, get_params={
"method": "track.getInfo", "artist": artist,
2016-03-29 11:56:58 +00:00
"track": track_name, "autocorrect": "1",
"api_key": self.bot.config["lastfm-api-key"],
"user": lastfm_username, "format": "json"}, json=True)
2016-03-29 11:56:58 +00:00
tags = []
if "toptags" in info_page.get("track", []):
2016-04-04 11:37:17 +00:00
for tag in info_page["track"]["toptags"]["tag"]:
2016-03-29 11:56:58 +00:00
tags.append(tag["name"])
if tags:
tags = " (%s)" % ", ".join(tags)
else:
tags = ""
play_count = ""
if "userplaycount" in info_page.get("track", []):
play_count = int(info_page["track"]["userplaycount"])
play_count = " (%d play%s)" % (play_count,
"s" if play_count > 1 else "")
2016-03-29 11:56:58 +00:00
2016-04-04 11:37:17 +00:00
event["stdout"].write(
"%s is now playing: %s - %s%s%s" % (
shown_username, artist, track_name, play_count, tags))
2016-03-29 11:56:58 +00:00
else:
event["stderr"].write(
"The user '%s' has never scrobbled before" % (
shown_username))
2016-03-29 11:56:58 +00:00
else:
event["stderr"].write("Failed to load results")