mirror of
https://github.com/MuRuLOSE/limoka.git
synced 2026-06-16 14:34:17 +02:00
98 lines
3.1 KiB
Python
98 lines
3.1 KiB
Python
# -*- coding: utf-8 -*-
|
||
|
||
# Module author: Official Repo, @dekftgmodules
|
||
|
||
# requires: lyricsgenius ShazamAPI
|
||
|
||
import io
|
||
|
||
import lyricsgenius
|
||
from ShazamAPI import Shazam
|
||
|
||
from .. import loader, utils
|
||
|
||
|
||
@loader.tds
|
||
class LyricsMod(loader.Module):
|
||
"""Sings songs"""
|
||
|
||
strings = {
|
||
"name": "Lyrics",
|
||
"genius_api_token_doc": "The LyricsGenius API token from http://genius.com/api-clients",
|
||
"invalid_syntax": "<b>Please specify song and artist.</b>",
|
||
"song_not_found": "<b>Song not found</b>",
|
||
"missing_token": "<b>API Token missing</b>",
|
||
}
|
||
|
||
tag = "<b>[Shazam]</b> "
|
||
|
||
def __init__(self):
|
||
self.config = loader.ModuleConfig(
|
||
"GENIUS_API_TOKEN",
|
||
None,
|
||
lambda message: self.strings("genius_api_token_doc", message),
|
||
)
|
||
|
||
def config_complete(self):
|
||
if self.config["GENIUS_API_TOKEN"]:
|
||
self.genius = lyricsgenius.Genius(self.config["GENIUS_API_TOKEN"])
|
||
else:
|
||
self.genius = None
|
||
|
||
@loader.unrestricted
|
||
@loader.ratelimit
|
||
async def lyricscmd(self, message):
|
||
""".lyrics Song, Artist"""
|
||
if self.genius is None:
|
||
await utils.answer(message, self.strings("missing_token", message))
|
||
args = utils.get_args_split_by(message, ",")
|
||
if len(args) != 2:
|
||
await utils.answer(message, self.strings("invalid_syntax", message))
|
||
return
|
||
try:
|
||
song = await utils.run_sync(self.genius.search_song, args[0], args[1])
|
||
except TypeError:
|
||
# Song not found causes internal library error
|
||
song = None
|
||
if song is None:
|
||
await utils.answer(message, self.strings("song_not_found", message))
|
||
return
|
||
await utils.answer(message, utils.escape_html(song.lyrics))
|
||
|
||
async def shazamcmd(self, message):
|
||
""".shazam <reply to audio> - распознать трек"""
|
||
s = await get_audio_shazam(message)
|
||
if not s:
|
||
return
|
||
try:
|
||
shazam = Shazam(s.track.read())
|
||
recog = shazam.recognizeSong()
|
||
track = next(recog)[1]["track"]
|
||
await message.client.send_file(
|
||
message.to_id,
|
||
file=track["images"]["background"],
|
||
caption=self.tag + "Распознанный трек: " + track["share"]["subject"],
|
||
reply_to=s.reply.id,
|
||
)
|
||
await message.delete()
|
||
except:
|
||
await message.edit(self.tag + "Не удалось распознать...")
|
||
|
||
|
||
async def get_audio_shazam(message):
|
||
class rct:
|
||
track = io.BytesIO()
|
||
reply = None
|
||
|
||
reply = await message.get_reply_message()
|
||
if reply and reply.file and reply.file.mime_type.split("/")[0] == "audio":
|
||
ae = rct()
|
||
await utils.answer(message, "<b>Скачиваю...</b>")
|
||
ae.track = io.BytesIO(await reply.download_media(bytes))
|
||
ae.reply = reply
|
||
await message.edit("<b>Распознаю...</b>")
|
||
return ae
|
||
else:
|
||
await utils.answer(message, "<b>reply to audio...</b>")
|
||
return None
|