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

Merge pull request #1 from Steffo99/local

Browse source 
stuff
This commit is contained in:
Steffo 2018-02-25 20:48:06 +01:00 committed by GitHub
commit bf7404c521
Signed by: github
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 145 additions and 162 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
.gitignore vendored
View file

@ -3,4 +3,5 @@ config.ini
__pycache__ __pycache__
diario.json diario.json
libopus-0.dll libopus-0.dll
music.opus music.opus
opusfiles/

304
discordbot.py
View file

@ -1,7 +1,7 @@
import datetime
import random import random
import discord import discord
import discord.opus import discord.opus
import discord.voice_client
import functools import functools
import sys import sys
import db import db
@ -9,6 +9,9 @@ import errors
import youtube_dl import youtube_dl
import concurrent.futures import concurrent.futures
import stagismo import stagismo
import platform
import typing
import os
# Init the event loop # Init the event loop
import asyncio import asyncio
@ -19,88 +22,87 @@ import configparser
config = configparser.ConfigParser() config = configparser.ConfigParser()
config.read("config.ini") config.read("config.ini")
# Find the latest git tag class DurationError(Exception):
import subprocess pass
import os
old_wd = os.getcwd()
try:
os.chdir(os.path.dirname(__file__))
version = str(subprocess.check_output(["git", "describe", "--tags"]), encoding="utf8").strip()
except:
version = "v???"
finally:
os.chdir(old_wd)
# Init the discord bot
client = discord.Client()
discord.opus.load_opus("/usr/lib/x86_64-linux-gnu/libopus.so")
voice_client = None
voice_player = None
voice_queue = []
voice_playing = None
# Init the executor
executor = concurrent.futures.ThreadPoolExecutor(max_workers=3)
class Video: class Video:
def __init__(self): def __init__(self):
self.user = None self.user = None
self.info = None self.filename = None
self.enqueued = None self.ytdl_url = None
self.channel = None
@staticmethod @staticmethod
async def init(author, info, enqueued, channel): async def init(user, filename=None, ytdl_url=None):
if filename is None and ytdl_url is None:
raise Exception("Filename or url must be specified")
self = Video() self = Video()
discord_user = await find_user(author) discord_user = await find_user(user)
self.user = discord_user.royal if discord_user is not None else None self.user = discord_user.royal if discord_user is not None else None
self.info = info self.filename = filename
self.enqueued = enqueued self.ytdl_url = ytdl_url
self.channel = channel
return self return self
def create_embed(self):
embed = discord.Embed(type="rich",
title=self.info.get("title"),
url=self.info.get("webpage_url"),
colour=discord.Colour(13375518))
# Uploader
if self.info.get("uploader"):
embed.set_author(name=self.info["uploader"],
url=self.info.get("uploader_url"))
# Thumbnail
if "thumbnail" in self.info:
embed.set_thumbnail(url=self.info["thumbnail"])
# Duration
embed.add_field(name="Durata", value=str(datetime.timedelta(seconds=self.info["duration"])))
# Views
if "view_count" in self.info and self.info["view_count"] is not None:
embed.add_field(name="Visualizzazioni", value="{:_}".format(self.info["view_count"]).replace("_", " "))
# Likes
if "like_count" in self.info and self.info["like_count"] is not None:
embed.add_field(name="Mi piace", value="{:_}".format(self.info["like_count"]).replace("_", " "))
# Dislikes
if "dislike_count" in self.info and self.info["dislike_count"] is not None:
embed.add_field(name="Non mi piace", value="{:_}".format(self.info["dislike_count"]).replace("_", " "))
return embed
async def download(self): async def download(self):
# Retrieve info before downloading
try: try:
with youtube_dl.YoutubeDL({"noplaylist": True, with youtube_dl.YoutubeDL() as ytdl:
"format": "bestaudio", info = await loop.run_in_executor(executor, functools.partial(ytdl.extract_info, self.ytdl_url, download=False))
"postprocessors": [{ file_id = info["entries"][0].get("title", hash(self.ytdl_url))
"key": 'FFmpegExtractAudio',
"preferredcodec": 'opus'
}],
"outtmpl": "music.%(ext)s",
"quiet": True}) as ytdl:
info = await loop.run_in_executor(executor, functools.partial(ytdl.extract_info, self.info["webpage_url"]))
except Exception as e: except Exception as e:
client.send_message(self.channel, f"⚠ Errore durante il download del video:\n" print(e)
f"```" raise e
f"{e}" if os.path.exists(f"opusfiles/{file_id}.opus"):
f"```", embed=self.create_embed()) return
if info["entries"][0]["duration"] > int(config["YouTube"]["max_duration"]):
raise DurationError(f"File duration is over the limit set in the config ({config['YouTube']['max_duration']}).")
ytdl_args = {"noplaylist": True,
"format": "best",
"postprocessors": [{
"key": 'FFmpegExtractAudio',
"preferredcodec": 'opus'
}],
"outtmpl": f"opusfiles/{file_id}.opus",
"quiet": True}
if "youtu" in self.ytdl_url:
ytdl_args["username"] = config["YouTube"]["username"]
ytdl_args["password"] = config["YouTube"]["password"]
# Download the video
try:
with youtube_dl.YoutubeDL(ytdl_args) as ytdl:
await loop.run_in_executor(executor, functools.partial(ytdl.download, [self.ytdl_url]))
except Exception as e:
print(e)
raise e
# Set the filename to the downloaded video
self.filename = f"opusfiles/{file_id}.opus"
if __debug__:
version = "Dev"
else:
# Find the latest git tag
import subprocess
old_wd = os.getcwd()
try:
os.chdir(os.path.dirname(__file__))
version = str(subprocess.check_output(["git", "describe", "--tags"]), encoding="utf8").strip()
except:
version = "v???"
finally:
os.chdir(old_wd)
# 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":
discord.opus.load_opus("libopus-0.dll")
voice_client: typing.Optional[discord.VoiceClient] = None
voice_player: typing.Optional[discord.voice_client.StreamPlayer] = None
voice_queue: typing.List[Video] = []
# Init the executor
executor = concurrent.futures.ThreadPoolExecutor(max_workers=3)
async def find_user(user: discord.User): async def find_user(user: discord.User):
session = await loop.run_in_executor(executor, db.Session) session = await loop.run_in_executor(executor, db.Session)
@ -186,47 +188,10 @@ async def on_message(message: discord.Message):
if "playlist" in url: if "playlist" in url:
await client.send_message(message.channel, f"ℹ️ Hai inviato una playlist al bot.\n" await client.send_message(message.channel, f"ℹ️ Hai inviato una playlist al bot.\n"
f"L'elaborazione potrebbe richiedere un po' di tempo.") f"L'elaborazione potrebbe richiedere un po' di tempo.")
# Extract the info from the url # If target is a single video
try: video = await Video.init(user=message.author, ytdl_url=url)
with youtube_dl.YoutubeDL({"quiet": True, "skip_download": True, "noplaylist": True, "format": "webm[abr>0]/bestaudio/best"}) as ytdl: await client.send_message(message.channel, f"✅ Aggiunto alla coda: `{url}`")
info = await loop.run_in_executor(executor, functools.partial(ytdl.extract_info, url)) voice_queue.append(video)
except youtube_dl.utils.DownloadError as e:
if "is not a valid URL" in str(e) or "Unsupported URL" in str(e):
await client.send_message(message.channel, f"⚠️ Il link inserito non è valido.\n"
f"Se vuoi cercare un video su YouTube, usa `!search <query>`")
else:
await client.send_message(message.channel, f"⚠ Errore:\n"
f"```\n"
f"{e}"
f"```")
return
if "_type" not in info:
# If target is a single video
video = await Video.init(author=message.author, info=info, enqueued=datetime.datetime.now(), channel=message.channel)
await client.send_message(message.channel, f"✅ Aggiunto alla coda:", embed=video.create_embed())
voice_queue.append(video)
elif info["_type"] == "playlist":
# If target is a playlist
if len(info["entries"]) < 20:
for single_info in info["entries"]:
video = await Video.init(author=message.author, info=single_info, enqueued=datetime.datetime.now(), channel=message.channel)
await client.send_message(message.channel, f"✅ Aggiunto alla coda:", embed=video.create_embed())
voice_queue.append(video)
else:
await client.send_message(message.channel, f"ℹ La playlist contiene {len(info['entries'])} video.\n"
f"Sei sicuro di volerli aggiungere alla coda?\n"
f"Rispondi **sì** o **no**.\n"
f"_(Il bot potrebbe crashare.)_")
answer = await client.wait_for_message(timeout=60, author=message.author, channel=message.channel)
if "sì" in answer.content.lower() or "si" in answer.content.lower():
for single_info in info["entries"]:
video = await Video.init(author=message.author, info=single_info,
enqueued=datetime.datetime.now(), channel=message.channel)
await client.send_message(message.channel, f"✅ Aggiunto alla coda:", embed=video.create_embed())
voice_queue.append(video)
elif "no" in answer.content.lower():
await client.send_message(message.channel, f"ℹ Operazione annullata.")
return
elif message.content.startswith("!search"): elif message.content.startswith("!search"):
await client.send_typing(message.channel) await client.send_typing(message.channel)
# The bot should be in voice chat # The bot should be in voice chat
@ -241,22 +206,35 @@ async def on_message(message: discord.Message):
await client.send_message(message.channel, "⚠️ Non hai specificato il titolo!\n" await client.send_message(message.channel, "⚠️ Non hai specificato il titolo!\n"
"Sintassi corretta: `!search <titolo>`") "Sintassi corretta: `!search <titolo>`")
return return
# Extract the info from the url # If target is a single video
video = await Video.init(user=message.author, ytdl_url=f"ytsearch:{text}")
await client.send_message(message.channel, f"✅ Aggiunto alla coda: `ytsearch:{text}`")
voice_queue.append(video)
elif message.content.startswith("!file"):
await client.send_typing(message.channel)
# The bot should be in voice chat
if voice_client is None:
await client.send_message(message.channel, "⚠️ Non sono connesso alla cv!\n"
"Fammi entrare scrivendo `!cv` mentre sei in chat vocale.")
return
# Find the sent text
try: try:
with youtube_dl.YoutubeDL({"quiet": True, "skip_download": True, "noplaylist": True, "format": "webm[abr>0]/bestaudio/best"}) as ytdl: text:str = message.content.split(" ", 1)[1]
info = await loop.run_in_executor(executor, functools.partial(ytdl.extract_info, f"ytsearch:{text}")) except IndexError:
except youtube_dl.utils.DownloadError as e: await client.send_message(message.channel, "⚠️ Non hai specificato il nome del file!\n"
if "is not a valid URL" in str(e) or "Unsupported URL" in str(e): "Sintassi corretta: `!file <nomefile>`")
await client.send_message(message.channel, f"⚠️ Il video ottenuto dalla ricerca non è valido. Prova a cercare qualcos'altro...") return
# Ensure the filename ends with .opus
if not text.endswith(".opus"):
await client.send_message(message.channel, "⚠️ Il nome file specificato non è valido.")
return return
# If target is a single video # If target is a single video
video = await Video.init(author=message.author, info=info["entries"][0], enqueued=datetime.datetime.now(), channel=message.channel) video = await Video.init(user=message.author, filename=text)
await client.send_message(message.channel, f"✅ Aggiunto alla coda:", embed=video.create_embed()) await client.send_message(message.channel, f"✅ Aggiunto alla coda: `{text}`")
voice_queue.append(video) voice_queue.append(video)
elif message.content.startswith("!skip"): elif message.content.startswith("!skip"):
global voice_player global voice_player
voice_player.stop() voice_player.stop()
voice_player = None
await client.send_message(message.channel, f"⏩ Video saltato.") await client.send_message(message.channel, f"⏩ Video saltato.")
elif message.content.startswith("!pause"): elif message.content.startswith("!pause"):
if voice_player is None or not voice_player.is_playing(): if voice_player is None or not voice_player.is_playing():
@ -272,12 +250,11 @@ async def on_message(message: discord.Message):
voice_player.resume() voice_player.resume()
await client.send_message(message.channel, f"▶️ Riproduzione ripresa.") await client.send_message(message.channel, f"▶️ Riproduzione ripresa.")
elif message.content.startswith("!cancel"): elif message.content.startswith("!cancel"):
try: if not len(voice_queue) > 1:
video = voice_queue.pop() await client.send_message(message.channel, f"⚠ Non ci sono video da annullare.")
except IndexError:
await client.send_message(message.channel, f"⚠ La playlist è vuota.")
return return
await client.send_message(message.channel, f"❌ Rimosso dalla playlist:", embed=video.create_embed()) video = voice_queue.pop()
await client.send_message(message.channel, f"❌ L'ultimo video aggiunto alla playlist è stato rimosso.")
elif message.content.startswith("!stop"): elif message.content.startswith("!stop"):
if voice_player is None: if voice_player is None:
await client.send_message(message.channel, f"⚠ Non c'è nulla da interrompere!") await client.send_message(message.channel, f"⚠ Non c'è nulla da interrompere!")
@ -286,23 +263,23 @@ async def on_message(message: discord.Message):
voice_player.stop() voice_player.stop()
voice_player = None voice_player = None
await client.send_message(message.channel, f"⏹ Riproduzione interrotta e playlist svuotata.") await client.send_message(message.channel, f"⏹ Riproduzione interrotta e playlist svuotata.")
elif message.content.startswith("!np"): #elif message.content.startswith("!np"):
if voice_player is None or not voice_player.is_playing(): # if voice_player is None or not voice_player.is_playing():
await client.send_message(message.channel, f"ℹ Non c'è nulla in riproduzione al momento.") # await client.send_message(message.channel, f"ℹ Non c'è nulla in riproduzione al momento.")
return # return
await client.send_message(message.channel, f"▶️ Ora in riproduzione in <#{voice_client.channel.id}>:", embed=voice_playing.create_embed()) # await client.send_message(message.channel, f"▶️ Ora in riproduzione in <#{voice_client.channel.id}>:", embed=voice_playing.create_embed())
elif message.content.startswith("!queue"): #elif message.content.startswith("!queue"):
if voice_player is None: # if voice_player is None:
await client.send_message(message.channel, f"ℹ Non c'è nulla in riproduzione al momento.") # await client.send_message(message.channel, f"ℹ Non c'è nulla in riproduzione al momento.")
return # return
to_send = "" # to_send = ""
to_send += f"0. {voice_playing.info['title'] if voice_playing.info['title'] is not None else '_Senza titolo_'} - <{voice_playing.info['webpage_url'] if voice_playing.info['webpage_url'] is not None else ''}>\n" # to_send += f"0. {voice_playing.info['title'] if voice_playing.info['title'] is not None else '_Senza titolo_'} - <{voice_playing.info['webpage_url'] if voice_playing.info['webpage_url'] is not None else ''}>\n"
for n, video in enumerate(voice_queue): # for n, video in enumerate(voice_queue):
to_send += f"{n+1}. {video.info['title'] if video.info['title'] is not None else '_Senza titolo_'} - <{video.info['webpage_url'] if video.info['webpage_url'] is not None else ''}>\n" # to_send += f"{n+1}. {video.info['title'] if video.info['title'] is not None else '_Senza titolo_'} - <{video.info['webpage_url'] if video.info['webpage_url'] is not None else ''}>\n"
if len(to_send) >= 2000: # if len(to_send) >= 2000:
to_send = to_send[0:1997] + "..." # to_send = to_send[0:1997] + "..."
break # break
await client.send_message(message.channel, to_send) # await client.send_message(message.channel, to_send)
elif message.content.startswith("!cast"): elif message.content.startswith("!cast"):
try: try:
spell = message.content.split(" ", 1)[1] spell = message.content.split(" ", 1)[1]
@ -342,34 +319,39 @@ async def update_users_pipe(users_connection):
async def update_music_queue(): async def update_music_queue():
await client.wait_until_ready() await client.wait_until_ready()
global voice_client
global voice_player global voice_player
global voice_playing global voice_queue
while True: while True:
# Wait until there is nothing playing if voice_client is None:
if voice_client is not None and voice_player is not None and (voice_player.is_playing() and not voice_player.is_done()): await asyncio.sleep(5)
continue
if voice_player is not None and not voice_player._end.is_set():
await asyncio.sleep(1) await asyncio.sleep(1)
continue continue
if len(voice_queue) == 0: if len(voice_queue) == 0:
if voice_playing is not None: await client.change_presence()
# Set the playing status
voice_playing = None
await client.change_presence()
await asyncio.sleep(1) await asyncio.sleep(1)
continue continue
# Get the last video in the queue video = voice_queue.pop()
video = voice_queue.pop(0) if video.ytdl_url:
# Notify the chat of the download await client.send_message(client.get_channel(config["Discord"]["main_channel"]), f"ℹ E' iniziato il download di `{video.ytdl_url}`.")
await client.send_message(video.channel, f"ℹ E' iniziato il download della prossima canzone.") try:
# Download the video await video.download()
await video.download() except DurationError:
# Play the video await client.send_message(client.get_channel(config["Discord"]["main_channel"]), f"⚠ Il file supera il limite di durata impostato in config.ini (`{config['YouTube']['max_duration']}` secondi).")
voice_player = voice_client.create_ffmpeg_player(f"music.opus") continue
except Exception as e:
await client.send_message(client.get_channel(config["Discord"]["main_channel"]), f"⚠️ C'è stato un errore durante il download di `{video.ytdl_url}`:\n"
f"```\n"
f"{e}\n"
f"```")
continue
voice_player = voice_client.create_ffmpeg_player(video.filename)
voice_player.start() voice_player.start()
# Notify the chat of the start await client.send_message(client.get_channel(config["Discord"]["main_channel"]), f"▶ Ora in riproduzione in <#{voice_client.channel.id}>:\n"
await client.send_message(video.channel, f"▶ Ora in riproduzione in <#{voice_client.channel.id}>:", embed=video.create_embed()) f"`{video.filename}`")
# Set the playing status await client.change_presence(game=discord.Game(name="youtube-dl", type=2))
voice_playing = video
await client.change_presence(game=discord.Game(name=video.info.get("title"), type=2))
def process(users_connection=None): def process(users_connection=None):