1
Fork 0
mirror of https://github.com/RYGhub/royalnet.git synced 2024-11-23 19:44:20 +00:00
royalnet/discordbot.py

737 lines
30 KiB
Python
Raw Normal View History

2017-11-11 17:55:13 +00:00
import random
2018-02-26 15:38:47 +00:00
import re
2017-10-27 11:38:32 +00:00
import discord
2017-10-31 22:50:05 +00:00
import discord.opus
2018-02-25 18:50:57 +00:00
import discord.voice_client
2017-11-01 18:23:11 +00:00
import functools
2017-11-09 19:34:18 +00:00
import sys
2017-10-27 11:38:32 +00:00
import db
2017-11-05 21:50:37 +00:00
import youtube_dl
2018-01-15 12:42:40 +00:00
import concurrent.futures
2018-02-25 18:50:57 +00:00
import platform
import typing
2018-02-25 19:47:12 +00:00
import os
2018-02-26 09:48:17 +00:00
import asyncio
import configparser
2018-04-09 17:51:05 +00:00
import subprocess
2018-04-09 20:55:48 +00:00
import async_timeout
import raven
2018-05-31 19:43:43 +00:00
import logging
import errors
2018-07-26 17:26:03 +00:00
import datetime
2018-05-31 19:43:43 +00:00
logging.basicConfig()
2017-10-27 11:38:32 +00:00
2018-02-26 15:38:47 +00:00
# Queue emojis
2018-05-25 17:58:30 +00:00
queue_emojis = [":one:",
":two:",
":three:",
":four:",
":five:",
":six:",
":seven:",
":eight:",
":nine:",
":keycap_ten:"]
2018-02-26 15:38:47 +00:00
2017-10-27 11:38:32 +00:00
# Init the event loop
loop = asyncio.get_event_loop()
# Init the config reader
config = configparser.ConfigParser()
config.read("config.ini")
2018-08-05 21:24:41 +00:00
# Radio messages
radio_messages = ["https://www.youtube.com/watch?v=3-yeK1Ck4yk&feature=youtu.be"]
radio_messages_enabled = False
2018-08-05 21:36:48 +00:00
radio_message_in = int(config["Discord"]["radio_messages_every"])
2018-08-05 21:24:41 +00:00
2018-02-25 19:47:12 +00:00
class DurationError(Exception):
pass
class InfoNotRetrievedError(Exception):
pass
class FileNotDownloadedError(Exception):
pass
class AlreadyDownloadedError(Exception):
2018-05-26 08:50:54 +00:00
pass
2018-02-26 09:48:17 +00:00
2018-05-26 08:50:54 +00:00
class Video:
2018-07-25 22:08:53 +00:00
def __init__(self, url: str=None, file: str=None, info: dict=None, enqueuer: discord.Member=None):
2018-05-26 08:50:54 +00:00
self.url = url
if file is None and info is None:
self.file = str(hash(url)) + ".opus"
elif info is not None:
2018-05-31 19:43:43 +00:00
self.file = re.sub(r'[/\\?*"<>|!:]', "_", info["title"]) + ".opus"
else:
self.file = file
self.downloaded = False if file is None else True
self.info = info
2018-07-25 22:08:53 +00:00
self.enqueuer = enqueuer
2018-05-26 08:50:54 +00:00
def __str__(self):
if self.info is None or "title" not in self.info:
return f"`{self.file}`"
return f"_{self.info['title']}_"
2018-05-29 21:50:47 +00:00
def plain_text(self):
if self.info is None or "title" not in self.info:
return self.file
return self.info['title']
async def download(self, progress_hooks: typing.List["function"]=None):
# File already downloaded
if self.downloaded:
raise AlreadyDownloadedError()
# No progress hooks
if progress_hooks is None:
progress_hooks = []
# Check if under max duration
if self.info is not None and self.info.get("duration", 0) > int(config["YouTube"]["max_duration"]):
raise DurationError()
# Download the file
with youtube_dl.YoutubeDL({"noplaylist": True,
"format": "best",
"postprocessors": [{
"key": 'FFmpegExtractAudio',
"preferredcodec": 'opus'
}],
"outtmpl": f"./opusfiles/{self.file}",
"progress_hooks": progress_hooks,
"quiet": True}) as ytdl:
await loop.run_in_executor(executor, functools.partial(ytdl.download, [self.url]))
self.downloaded = True
async def create_player(self) -> discord.voice_client.ProcessPlayer:
# Check if the file has been downloaded
if not self.downloaded:
raise FileNotDownloadedError()
global voice_client
return voice_client.create_ffmpeg_player(f"./opusfiles/{self.file}")
2018-05-26 08:50:54 +00:00
# noinspection PyUnreachableCode
2018-02-25 18:50:57 +00:00
if __debug__:
version = "Dev"
commit_msg = "_in sviluppo_"
2018-02-25 18:50:57 +00:00
else:
# Find the latest git tag
old_wd = os.getcwd()
try:
os.chdir(os.path.dirname(__file__))
version = str(subprocess.check_output(["git", "describe", "--tags"]), encoding="utf8").strip()
commit_msg = str(subprocess.check_output(["git", "log", "-1", "--pretty=%B"]), encoding="utf8").strip()
2018-02-26 09:48:17 +00:00
except Exception:
2018-02-26 15:38:47 +00:00
version = ""
2018-02-25 18:50:57 +00:00
finally:
os.chdir(old_wd)
2017-11-06 15:54:36 +00:00
2018-02-25 18:50:57 +00:00
# Init the discord bot
client = discord.Client()
if platform.system() == "Linux":
discord.opus.load_opus("/usr/lib/x86_64-linux-gnu/libopus.so")
elif platform.system() == "Windows":
2018-05-25 17:58:30 +00:00
discord.opus.load_opus("libopus-0.dll")
2018-02-25 18:50:57 +00:00
voice_client = None
voice_player = None
2018-05-29 21:50:47 +00:00
now_playing = None
voice_queue = []
2018-02-25 18:50:57 +00:00
# Init the executor
executor = concurrent.futures.ThreadPoolExecutor(max_workers=3)
2017-11-06 15:54:36 +00:00
# Init the Sentry client
sentry = raven.Client(config["Sentry"]["token"],
2018-05-25 17:58:30 +00:00
release=version,
install_logging_hook=False,
hook_libraries=[])
2018-03-12 12:29:12 +00:00
2017-11-10 07:53:48 +00:00
async def on_error(event, *args, **kwargs):
ei = sys.exc_info()
print("ERRORE CRITICO:\n" + repr(ei[1]) + "\n\n" + repr(ei))
2017-11-10 07:53:48 +00:00
try:
2018-05-31 19:43:43 +00:00
await client.send_message(client.get_channel(config["Discord"]["main_channel"]),
f"☢️ **ERRORE CRITICO NELL'EVENTO** `{event}`\n"
f"Il bot si è chiuso e si dovrebbe riavviare entro qualche minuto.\n"
f"Una segnalazione di errore è stata automaticamente mandata a Steffo.\n\n"
2018-02-26 09:48:17 +00:00
f"Dettagli dell'errore:\n"
f"```python\n"
f"{repr(ei[1])}\n"
2018-02-26 09:48:17 +00:00
f"```")
if voice_client is not None:
await voice_client.disconnect()
2017-11-10 07:53:48 +00:00
await client.change_presence(status=discord.Status.invisible)
await client.close()
except Exception as e:
print("ERRORE CRITICO PIU' CRITICO:\n" + repr(e) + "\n\n" + repr(sys.exc_info()))
2017-11-10 07:53:48 +00:00
loop.stop()
sentry.captureException(exc_info=ei)
2017-11-15 09:48:58 +00:00
os._exit(1)
pass
2017-11-09 19:34:18 +00:00
2017-11-10 07:53:48 +00:00
@client.event
async def on_ready():
2018-05-31 19:43:43 +00:00
await client.send_message(client.get_channel(config["Discord"]["main_channel"]),
f" Royal Bot avviato e pronto a ricevere comandi!\n"
f"Ultimo aggiornamento: `{version}: {commit_msg}`")
2017-11-10 07:53:48 +00:00
await client.change_presence(game=None, status=discord.Status.online)
2017-10-27 11:38:32 +00:00
@client.event
async def on_message(message: discord.Message):
global voice_client
2017-11-07 17:44:00 +00:00
global voice_player
2018-06-27 20:21:17 +00:00
if message.channel != client.get_channel(config["Discord"]["main_channel"]) or message.author.bot:
return
sentry.user_context({
"discord": {
"discord_id": message.author.id,
"name": message.author.name,
"discriminator": message.author.discriminator
}
})
2018-06-27 20:21:17 +00:00
if not message.content.startswith("!"):
2018-06-27 20:15:53 +00:00
client.send_message(message.channel,
":warning: In questa chat sono consentiti solo comandi per il bot.\n"
"Riinvia il tuo messaggio in <#425780562805129226>!")
client.delete_message(message)
return
data = message.content.split(" ")
if data[0] not in commands:
await client.send_message(message.channel, ":warning: Comando non riconosciuto.")
return
await commands[data[0]](channel=client.get_channel(config["Discord"]["main_channel"]),
author=message.author,
params=data)
2017-11-05 21:50:37 +00:00
2017-10-30 12:45:38 +00:00
async def update_users_pipe(users_connection):
await client.wait_until_ready()
while True:
2018-01-15 12:42:40 +00:00
msg = await loop.run_in_executor(executor, users_connection.recv)
2018-06-13 22:10:57 +00:00
if msg == "get cv":
2017-10-30 12:45:38 +00:00
discord_members = list(client.get_server(config["Discord"]["server_id"]).members)
users_connection.send(discord_members)
2018-07-31 16:31:06 +00:00
elif msg == "stop":
await client.logout()
exit(0)
2018-06-13 22:10:57 +00:00
elif msg.startswith("!"):
data = msg.split(" ")
if data[0] not in commands:
users_connection.send("error")
continue
await commands[data[0]](channel=client.get_channel(config["Discord"]["main_channel"]),
author=None,
2018-06-20 17:46:35 +00:00
params=data)
2018-06-13 22:10:57 +00:00
users_connection.send("success")
2017-10-30 12:45:38 +00:00
2018-05-25 17:58:30 +00:00
def command(func):
"""Decorator. Runs the function as a Discord command."""
async def new_func(channel: discord.Channel, author: discord.Member, params: typing.List[str], *args, **kwargs):
2018-06-13 22:10:57 +00:00
if author is not None:
sentry.user_context({
"discord_id": author.id,
"username": f"{author.name}#{author.discriminator}"
})
else:
sentry.user_context({
"source": "Telegram"
})
2018-05-25 17:58:30 +00:00
try:
result = await func(channel=channel, author=author, params=params, *args, **kwargs)
except Exception:
ei = sys.exc_info()
2018-05-25 17:58:30 +00:00
try:
await client.send_message(channel,
f"☢ **ERRORE DURANTE L'ESECUZIONE DEL COMANDO {params[0]}**\n"
f"Il comando è stato ignorato.\n"
f"Una segnalazione di errore è stata automaticamente mandata a Steffo.\n\n"
f"Dettagli dell'errore:\n"
f"```python\n"
f"{repr(ei[1])}\n"
f"```")
2018-05-31 19:43:43 +00:00
except Exception:
2018-05-25 17:58:30 +00:00
pass
sentry.captureException(exc_info=ei)
else:
return result
return new_func
2018-07-15 12:41:42 +00:00
def requires_voice_client(func):
2018-05-25 17:58:30 +00:00
"Decorator. Ensures the voice client is connected before running the command."
async def new_func(channel: discord.Channel, author: discord.Member, params: typing.List[str], *args, **kwargs):
2018-05-25 17:58:30 +00:00
global voice_client
if voice_client is None or not voice_client.is_connected():
await client.send_message(channel,
"⚠️ Non sono connesso alla cv!\n"
"Fammi entrare scrivendo `!cv` mentre sei in chat vocale.")
return
return await func(channel=channel, author=author, params=params, *args, **kwargs)
return new_func
2018-05-26 08:50:54 +00:00
def requires_rygdb(func, optional=False):
2018-05-25 17:58:30 +00:00
async def new_func(channel: discord.Channel, author: discord.Member, params: typing.List[str], *args, **kwargs):
2018-07-31 21:41:05 +00:00
session = db.Session()
2018-05-25 17:58:30 +00:00
dbuser = await loop.run_in_executor(executor,
session.query(db.Discord)
.filter_by(discord_id=author.id)
.join(db.Royal)
.first)
2018-07-25 22:08:53 +00:00
await loop.run_in_executor(executor, session.close)
2018-05-26 08:50:54 +00:00
if not optional and dbuser is None:
await client.send_message(channel,
"⚠️ Devi essere registrato su Royalnet per poter utilizzare questo comando.")
return
2018-05-25 17:58:30 +00:00
return await func(channel=channel, author=author, params=params, dbuser=dbuser, *args, **kwargs)
return new_func
@command
2018-05-26 08:50:54 +00:00
async def cmd_ping(channel: discord.Channel, author: discord.Member, params: typing.List[str]):
2018-05-25 17:58:30 +00:00
await client.send_message(channel, f"Pong!")
@command
async def cmd_cv(channel: discord.Channel, author: discord.Member, params: typing.List[str]):
2018-06-13 22:10:57 +00:00
if author is None:
await client.send_message(channel, "⚠ Questo comando richiede un autore.")
if author.voice is None or author.voice.voice_channel is None:
2018-05-25 17:58:30 +00:00
await client.send_message(channel, "⚠ Non sei in nessun canale!")
return
global voice_client
if voice_client is not None and voice_client.is_connected():
await voice_client.move_to(author.voice.voice_channel)
else:
voice_client = await client.join_voice_channel(author.voice.voice_channel)
await client.send_message(channel, f"✅ Mi sono connesso in <#{author.voice.voice_channel.id}>.")
2018-07-26 17:40:21 +00:00
async def add_video_from_url(url, index: typing.Optional[int]=None, enqueuer: discord.Member=None):
# Retrieve info
with youtube_dl.YoutubeDL({"quiet": True,
"ignoreerrors": True,
"simulate": True}) as ytdl:
info = await loop.run_in_executor(executor,
functools.partial(ytdl.extract_info, url=url, download=False))
if info is None:
await client.send_message(client.get_channel(config["Discord"]["main_channel"]),
f"⚠ Non è stato trovato nessun video all'URL `{url}`,"
f" pertanto non è stato aggiunto alla coda.")
return
if "entries" in info:
# This is a playlist
for entry in info["entries"]:
2018-07-26 17:40:21 +00:00
if index is not None:
voice_queue.insert(index, Video(url=entry["webpage_url"], info=entry, enqueuer=enqueuer))
else:
voice_queue.append(Video(url=entry["webpage_url"], info=entry, enqueuer=enqueuer))
return
# This is a single video
2018-07-26 17:40:21 +00:00
if index is not None:
voice_queue.insert(index, Video(url=url, info=info, enqueuer=enqueuer))
else:
voice_queue.append(Video(url=url, info=info, enqueuer=enqueuer))
2018-07-26 17:40:21 +00:00
async def add_video_from_file(file, index: typing.Optional[int]=None, enqueuer: discord.Member=None):
if index is not None:
voice_queue.insert(index, Video(file=file, enqueuer=enqueuer))
else:
voice_queue.append(Video(file=file, enqueuer=enqueuer))
2018-05-26 08:50:54 +00:00
@command
2018-07-15 12:41:42 +00:00
@requires_voice_client
2018-05-26 08:50:54 +00:00
async def cmd_play(channel: discord.Channel, author: discord.Member, params: typing.List[str]):
if len(params) < 2:
await client.send_message(channel, "⚠ Non hai specificato una canzone da riprodurre!\n"
"Sintassi: `!play <url|ricercayoutube|nomefile>`")
return
2018-08-05 21:24:41 +00:00
# If the radio messages are enabled...
global radio_messages_enabled
if radio_messages_enabled:
global radio_message_in
radio_message_in -= 1
if radio_message_in <= 0:
radio_message = random.sample(radio_messages, 1)[0]
2018-08-05 21:36:48 +00:00
radio_message_in = int(config["Discord"]["radio_messages_every"])
2018-08-05 21:24:41 +00:00
await add_video_from_url(radio_message)
await client.send_message(channel, f"✅ Aggiunto un messaggio radio, disattiva con `!radiomessages off`.")
2018-05-26 08:50:54 +00:00
# Parse the parameter as URL
url = re.match(r"(?:https?://|ytsearch[0-9]*:).*", " ".join(params[1:]).strip("<>"))
if url is not None:
2018-05-26 08:50:54 +00:00
# This is a url
2018-07-25 22:08:53 +00:00
await add_video_from_url(url.group(0), enqueuer=author)
await client.send_message(channel, f"✅ Video aggiunto alla coda.")
2018-05-26 08:50:54 +00:00
return
# Parse the parameter as file
file_path = os.path.join(os.path.join(os.path.curdir, "opusfiles"), " ".join(params[1:]))
if os.path.exists(file_path):
# This is a file
2018-07-25 22:08:53 +00:00
await add_video_from_file(file=file_path, enqueuer=author)
await client.send_message(channel, f"✅ Video aggiunto alla coda.")
return
file_path += ".opus"
2018-05-26 08:50:54 +00:00
if os.path.exists(file_path):
# This is a file
2018-07-25 22:08:53 +00:00
await add_video_from_file(file=file_path, enqueuer=author)
await client.send_message(channel, f"✅ Video aggiunto alla coda.")
2018-05-26 08:50:54 +00:00
return
# Search the parameter on youtube
search = " ".join(params[1:])
2018-05-26 08:50:54 +00:00
# This is a search
2018-07-25 22:08:53 +00:00
await add_video_from_url(url=f"ytsearch:{search}", enqueuer=author)
await client.send_message(channel, f"✅ Video aggiunto alla coda.")
@command
2018-07-15 12:41:42 +00:00
@requires_voice_client
async def cmd_skip(channel: discord.Channel, author: discord.Member, params: typing.List[str]):
global voice_player
2018-05-29 19:55:05 +00:00
if voice_player is None:
await client.send_message(channel, "⚠ Non c'è nessun video in riproduzione.")
return
voice_player.stop()
await client.send_message(channel, f"⏩ Video saltato.")
@command
2018-07-15 12:41:42 +00:00
@requires_voice_client
async def cmd_remove(channel: discord.Channel, author: discord.Member, params: typing.List[str]):
if len(voice_queue) == 0:
await client.send_message(channel, "⚠ Non c'è nessun video in coda.")
return
2018-06-20 17:46:35 +00:00
if len(params) == 1:
index = len(voice_queue) - 1
2018-07-26 17:40:21 +00:00
else:
try:
index = int(params[1]) - 1
except ValueError:
await client.send_message(channel, "⚠ Il numero inserito non è valido.\n"
2018-06-20 17:46:35 +00:00
"Sintassi: `!remove [numerovideoiniziale] [numerovideofinale]`")
return
2018-06-20 17:46:35 +00:00
if len(params) < 3:
if abs(index) >= len(voice_queue):
await client.send_message(channel, "⚠ Il numero inserito non corrisponde a nessun video nella playlist.\n"
"Sintassi: `!remove [numerovideoiniziale] [numerovideofinale]`")
return
2018-07-26 17:40:21 +00:00
video = voice_queue.pop(index)
2018-06-20 17:46:35 +00:00
await client.send_message(channel, f":regional_indicator_x: {str(video)} è stato rimosso dalla coda.")
return
try:
start = int(params[1]) - 1
except ValueError:
await client.send_message(channel, "⚠ Il numero iniziale inserito non è valido.\n"
"Sintassi: `!remove [numerovideoiniziale] [numerovideofinale]`")
return
if start >= len(voice_queue):
await client.send_message(channel, "⚠ Il numero iniziale inserito non corrisponde a nessun video nella"
" playlist.\n"
"Sintassi: `!remove [numerovideoiniziale] [numerovideofinale]`")
return
try:
end = int(params[2]) - 2
except ValueError:
await client.send_message(channel, "⚠ Il numero finale inserito non è valido.\n"
"Sintassi: `!remove [numerovideoiniziale] [numerovideofinale]`")
return
if end >= len(voice_queue):
await client.send_message(channel, "⚠ Il numero finale inserito non corrisponde a nessun video nella"
" playlist.\n"
"Sintassi: `!remove [numerovideoiniziale] [numerovideofinale]`")
return
if start > end:
await client.send_message(channel, "⚠ Il numero iniziale è maggiore del numero finale.\n"
"Sintassi: `!remove [numerovideoiniziale] [numerovideofinale]`")
return
2018-06-20 17:46:35 +00:00
del voice_queue[start:end]
await client.send_message(channel, f":regional_indicator_x: {end - start} video rimossi dalla coda.")
@command
async def cmd_queue(channel: discord.Channel, author: discord.Member, params: typing.List[str]):
if len(voice_queue) == 0:
await client.send_message(channel, "**Video in coda:**\n"
"nessuno")
return
msg = "**Video in coda:**\n"
for index, video in enumerate(voice_queue[:10]):
msg += f"{queue_emojis[index]} {str(video)}\n"
if len(voice_queue) > 10:
msg += f"più altri {len(voice_queue) - 10} video!"
await client.send_message(channel, msg)
2018-05-31 19:43:43 +00:00
@command
2018-07-15 12:41:42 +00:00
@requires_voice_client
2018-05-31 19:43:43 +00:00
async def cmd_shuffle(channel: discord.Channel, author: discord.Member, params: typing.List[str]):
if len(voice_queue) == 0:
await client.send_message(channel, "⚠ Non ci sono video in coda!")
return
random.shuffle(voice_queue)
2018-06-04 16:53:25 +00:00
await client.send_message(channel, "♠️ ♦️ ♣️ ♥️ Shuffle completo!")
2018-05-31 19:43:43 +00:00
2018-06-08 22:41:28 +00:00
@command
2018-07-15 12:41:42 +00:00
@requires_voice_client
2018-06-08 22:41:28 +00:00
async def cmd_clear(channel: discord.Channel, author: discord.Member, params: typing.List[str]):
global voice_queue
if len(voice_queue) == 0:
await client.send_message(channel, "⚠ Non ci sono video in coda!")
return
voice_queue = []
2018-06-20 17:46:35 +00:00
await client.send_message(channel, ":regional_indicator_x: Tutti i video in coda rimossi.")
2018-06-08 22:41:28 +00:00
2018-07-15 12:41:42 +00:00
@command
@requires_voice_client
async def cmd_dump_voice_player_error(channel: discord.Channel, author: discord.Member, params: typing.List[str]):
global voice_player
if voice_player is None:
return
await client.send_message(channel, f"```\n{str(voice_player.error)}\n```")
@command
async def cmd_register(channel: discord.Channel, author: discord.Member, params: typing.List[str]):
2018-07-31 21:41:05 +00:00
session = db.Session()
if len(params) < 1:
await client.send_message(channel, "⚠️ Non hai specificato un username!\n"
2018-07-25 22:08:53 +00:00
"Sintassi corretta: `!register <username_ryg>`")
return
try:
d = db.Discord.create(session,
royal_username=params[0],
discord_user=author)
except errors.AlreadyExistingError:
await client.send_message(channel,
"⚠ Il tuo account Discord è già collegato a un account RYG "
"o l'account RYG che hai specificato è già collegato a un account Discord.")
return
session.add(d)
session.commit()
session.close()
await client.send_message(channel, "✅ Sincronizzazione completata!")
2018-07-25 22:08:53 +00:00
2018-07-26 17:40:21 +00:00
@command
2018-07-26 17:46:35 +00:00
@requires_voice_client
2018-07-26 17:40:21 +00:00
async def cmd_forceplay(channel: discord.Channel, author: discord.Member, params: typing.List[str]):
if voice_player is not None:
voice_player.stop()
if len(params) < 2:
await client.send_message(channel, "⚠ Non hai specificato una canzone da riprodurre!\n"
2018-08-05 21:24:41 +00:00
"Sintassi: `!forceplay <url|ricercayoutube|nomefile>`")
2018-07-26 17:40:21 +00:00
return
# Parse the parameter as URL
url = re.match(r"(?:https?://|ytsearch[0-9]*:).*", " ".join(params[1:]).strip("<>"))
if url is not None:
# This is a url
await add_video_from_url(url.group(0), enqueuer=author, index=0)
await client.send_message(channel, f"✅ Riproduzione del video forzata.")
return
# Parse the parameter as file
file_path = os.path.join(os.path.join(os.path.curdir, "opusfiles"), " ".join(params[1:]))
if os.path.exists(file_path):
# This is a file
await add_video_from_file(file=file_path, enqueuer=author, index=0)
await client.send_message(channel, f"✅ Riproduzione del video forzata.")
return
file_path += ".opus"
if os.path.exists(file_path):
# This is a file
await add_video_from_file(file=file_path, enqueuer=author, index=0)
await client.send_message(channel, f"✅ Riproduzione del video forzata.")
return
# Search the parameter on youtube
search = " ".join(params[1:])
# This is a search
await add_video_from_url(url=f"ytsearch:{search}", enqueuer=author, index=0)
await client.send_message(channel, f"✅ Riproduzione del video forzata.")
2018-08-05 21:24:41 +00:00
@command
async def cmd_radiomessages(channel: discord.Channel, author: discord.Member, params: typing.List[str]):
global radio_messages_enabled
if len(params) < 2:
radio_messages_enabled = not radio_messages_enabled
else:
if params[1].lower() == "on":
radio_messages_enabled = True
elif params[1].lower() == "off":
radio_messages_enabled = False
else:
await client.send_message(channel, "⚠ Sintassi del comando non valida.\n"
"Sintassi: `!radiomessages [on|off]`")
return
await client.send_message(channel,
f"📻 Messaggi radio **{'attivati' if radio_messages_enabled else 'disattivati'}**.")
async def queue_predownload_videos():
while True:
for index, video in enumerate(voice_queue[:int(config["YouTube"]["predownload_videos"])].copy()):
if video.downloaded:
continue
try:
with async_timeout.timeout(int(config["YouTube"]["download_timeout"])):
await video.download()
except asyncio.TimeoutError:
2018-05-31 19:43:43 +00:00
await client.send_message(client.get_channel(config["Discord"]["main_channel"]),
f"⚠️ Il download di {str(video)} ha richiesto più di"
f" {config['YouTube']['download_timeout']} secondi, pertanto è stato rimosso"
f" dalla coda.")
del voice_queue[index]
continue
except DurationError:
2018-05-31 19:43:43 +00:00
await client.send_message(client.get_channel(config["Discord"]["main_channel"]),
2018-05-29 21:50:47 +00:00
f"⚠️ {str(video)} dura più di"
f" {str(int(config['YouTube']['max_duration']) // 60)}"
f" minuti, quindi è stato rimosso dalla coda.")
del voice_queue[index]
continue
except Exception as e:
2018-05-31 19:43:43 +00:00
await client.send_message(client.get_channel(config["Discord"]["main_channel"]),
f"⚠️ E' stato incontrato un errore durante il download di {str(video)},"
f" quindi è stato rimosso dalla coda.\n\n"
f"```python\n"
f"{str(e)}"
f"```")
del voice_queue[index]
continue
await asyncio.sleep(1)
2018-05-26 08:50:54 +00:00
2018-08-01 23:11:43 +00:00
song_special_messages = {
"despacito": ":arrow_forward: this is so sad. alexa play {song}",
"faded": ":arrow_forward: Basta Garf, lasciami ascoltare {song}",
"ligma": ":arrow_forward: What is ligma? {song}!",
"sugma": ":arrow_forward: What is sugma? {song}!",
"sugondese": ":arrow_forward: What is sugondese? {song}!",
"bofa": ":arrow_forward: What is bofa? {song}!",
"updog": ":arrow_forward: What is updog? {song}!",
"sayo-nara": ":arrow_forward: I gently open the door. {song} awaits me inside.",
"monika": ":arrow_forward: Just Monika. Just Monika. Just {song}.",
"take me home": ":arrow_forward: Take me home, to {song}, the place I belong!",
"never gonna give you up": ":arrow_forward: Rickrolling in 2018. Enjoy {song}!",
"september": ":arrow_forward: Do you remember? {song}.",
"homestuck": ":arrow_forward: > Enter song name. {song}",
"undertale": ":arrow_forward: Howdy! I'm Flowey! Listen to this friendly song: {song}",
"pumped up kicks": ":arrow_forward: Non metterti mica in testa strane idee ascoltando {song}...",
"jesus": ":arrow_forward: Respawn in 3 giorni. Intanto, ascolta {song}.",
"through The fire And flames": ":arrow_forward: Fai {song} su osu!, se ne sei capace!",
"slow clap": ":arrow_forward: :clap: :clap: :clap: {song} :clap: :clap: :clap:",
"pub scrubs": ":arrow_forward: MAV COME BACK WE MISS {song}!",
"alleluia": ":arrow_forward: Wah. Waaaah. Waluigi tiime: {song}",
"wah": ":arrow_forward: Wah. Waaaah. Waluigi tiime: {song}",
"waluigi": ":arrow_forward: Wah. Waaaah. Waluigi tiime: {song}",
"nyan cat": ":arrow_forward: Meow! :3 {song}",
"dragonborn": ":arrow_forward: FUS RO {song}!",
"dovahkiin": ":arrow_forward: FUS RO {song}!",
"initial d": ":arrow_forward: Guarda mamma sto driftando sul balcone di Balu grazie a {song}!",
"persona": ":arrow_forward: You'll never see {song} comiiiiing!",
"flamingo": ":arrow_forward: How many {song} do you have to eat?",
"linkin park": ":arrow_forward: Crawling in my {song}!",
"magicite": "⚠️ Warning: {song} contiene numerosi bug.",
"papers please": ":arrow_forward: Glory to Arstotzka! {song}!",
2018-08-05 10:07:35 +00:00
"we are number one": ":arrow_forward: Now paying respect to Robbie Rotten: {song}",
"jump up superstar": ":arrow_forward: Is {song} the Tengen Toppa Guren Lagann opening?"
2018-08-01 23:11:43 +00:00
}
async def queue_play_next_video():
await client.wait_until_ready()
global voice_client
global voice_player
2018-05-29 21:50:47 +00:00
global now_playing
while True:
if voice_client is None:
await asyncio.sleep(1)
continue
if voice_player is not None and not voice_player.is_done():
await asyncio.sleep(0.5)
continue
if len(voice_queue) == 0:
await asyncio.sleep(0.5)
2018-05-29 21:50:47 +00:00
if now_playing is not None:
await client.change_presence()
now_playing = None
continue
2018-05-29 21:50:47 +00:00
now_playing = voice_queue[0]
if not now_playing.downloaded:
await asyncio.sleep(0.5)
continue
2018-05-29 21:50:47 +00:00
voice_player = await now_playing.create_player()
voice_player.start()
2018-07-25 22:08:53 +00:00
if now_playing.enqueuer is not None:
2018-07-31 21:41:05 +00:00
session = db.Session()
2018-07-26 18:18:00 +00:00
enqueuer = await loop.run_in_executor(executor, session.query(db.Discord).filter_by(discord_id=now_playing.enqueuer.id).one_or_none)
played_music = db.PlayedMusic(enqueuer=enqueuer,
2018-07-31 21:44:56 +00:00
filename=now_playing.plain_text(),
2018-07-26 17:26:03 +00:00
timestamp=datetime.datetime.now())
2018-07-25 22:08:53 +00:00
session.add(played_music)
await loop.run_in_executor(executor, session.commit)
await loop.run_in_executor(executor, session.close)
2018-05-29 21:50:47 +00:00
await client.change_presence(game=discord.Game(name=now_playing.plain_text(), type=2))
2018-08-01 23:11:43 +00:00
for key in song_special_messages:
if key in now_playing.file.lower():
await client.send_message(client.get_channel(config["Discord"]["main_channel"]),
song_special_messages[key].format(song=str(now_playing)))
2018-08-01 23:13:20 +00:00
break
2018-07-26 17:40:21 +00:00
else:
await client.send_message(client.get_channel(config["Discord"]["main_channel"]),
f":arrow_forward: Ora in riproduzione: {str(now_playing)}")
del voice_queue[0]
2018-06-13 22:10:57 +00:00
commands = {
"!ping": cmd_ping,
"!cv": cmd_cv,
2018-07-26 17:40:21 +00:00
"!summon": cmd_cv,
2018-06-13 22:10:57 +00:00
"!play": cmd_play,
"!p": cmd_play,
2018-06-27 20:15:53 +00:00
"!search": cmd_play,
"!file": cmd_play,
2018-06-13 22:10:57 +00:00
"!skip": cmd_skip,
"!s": cmd_skip,
"!remove": cmd_remove,
2018-06-20 17:46:35 +00:00
"!cancel": cmd_remove,
2018-06-13 22:10:57 +00:00
"!queue": cmd_queue,
"!q": cmd_queue,
"!shuffle": cmd_shuffle,
2018-07-15 12:41:42 +00:00
"!clear": cmd_clear,
"!dump_vp": cmd_dump_voice_player_error,
2018-07-26 17:40:21 +00:00
"!register": cmd_register,
"!forceplay": cmd_forceplay,
2018-08-05 21:24:41 +00:00
"!fp": cmd_forceplay,
"!radiomessages": cmd_radiomessages
2018-06-13 22:10:57 +00:00
}
2018-01-15 12:42:40 +00:00
def process(users_connection=None):
2017-10-30 12:45:38 +00:00
print("Discordbot starting...")
2018-01-15 12:42:40 +00:00
if users_connection is not None:
asyncio.ensure_future(update_users_pipe(users_connection))
asyncio.ensure_future(queue_predownload_videos())
asyncio.ensure_future(queue_play_next_video())
2017-11-09 19:34:18 +00:00
client.on_error = on_error
2018-05-31 19:43:43 +00:00
loop.run_until_complete(client.login(config["Discord"]["bot_token"], bot=True))
loop.run_until_complete(client.connect())
2018-01-15 12:42:40 +00:00
if __name__ == "__main__":
2018-01-25 14:30:07 +00:00
process()