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

caching anyone

Browse source
This commit is contained in:
Steffo 2018-02-25 20:47:12 +01:00
parent 2e01daaafc
commit 9daa86d0d4
WARNING! Although there is a key with this ID in the database it does not verify this commit! This commit is SUSPICIOUS.
GPG key ID: C27544372FBB445D
1 changed files with 32 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

48
discordbot.py
View file

@ -1,4 +1,3 @@
import datetime
import random import random
import discord import discord
import discord.opus import discord.opus
@ -11,8 +10,8 @@ import youtube_dl
import concurrent.futures import concurrent.futures
import stagismo import stagismo
import platform import platform
import uuid
import typing import typing
import os
# Init the event loop # Init the event loop
import asyncio import asyncio
@ -23,6 +22,9 @@ import configparser
config = configparser.ConfigParser() config = configparser.ConfigParser()
config.read("config.ini") config.read("config.ini")
class DurationError(Exception):
pass
class Video: class Video:
def __init__(self): def __init__(self):
self.user = None self.user = None
@ -40,18 +42,27 @@ class Video:
self.ytdl_url = ytdl_url self.ytdl_url = ytdl_url
return self return self
async def download(self) -> bool: async def download(self):
# Don't download already existent files # Retrieve info before downloading
if self.filename is not None: try:
return True with youtube_dl.YoutubeDL() as ytdl:
file_id = uuid.uuid4() info = await loop.run_in_executor(executor, functools.partial(ytdl.extract_info, self.ytdl_url, download=False))
file_id = info["entries"][0].get("title", hash(self.ytdl_url))
except Exception as e:
print(e)
raise e
if os.path.exists(f"opusfiles/{file_id}.opus"):
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, ytdl_args = {"noplaylist": True,
"format": "bestaudio", "format": "best",
"postprocessors": [{ "postprocessors": [{
"key": 'FFmpegExtractAudio', "key": 'FFmpegExtractAudio',
"preferredcodec": 'opus' "preferredcodec": 'opus'
}], }],
"outtmpl": f"opusfiles/{file_id}.opus"} "outtmpl": f"opusfiles/{file_id}.opus",
"quiet": True}
if "youtu" in self.ytdl_url: if "youtu" in self.ytdl_url:
ytdl_args["username"] = config["YouTube"]["username"] ytdl_args["username"] = config["YouTube"]["username"]
ytdl_args["password"] = config["YouTube"]["password"] ytdl_args["password"] = config["YouTube"]["password"]
@ -61,17 +72,15 @@ class Video:
await loop.run_in_executor(executor, functools.partial(ytdl.download, [self.ytdl_url])) await loop.run_in_executor(executor, functools.partial(ytdl.download, [self.ytdl_url]))
except Exception as e: except Exception as e:
print(e) print(e)
return False raise e
# Set the filename to the downloaded video # Set the filename to the downloaded video
self.filename = f"opusfiles/{file_id}.opus" self.filename = f"opusfiles/{file_id}.opus"
return True
if __debug__: if __debug__:
version = "Dev" version = "Dev"
else: else:
# Find the latest git tag # Find the latest git tag
import subprocess import subprocess
import os
old_wd = os.getcwd() old_wd = os.getcwd()
try: try:
os.chdir(os.path.dirname(__file__)) os.chdir(os.path.dirname(__file__))
@ -327,15 +336,22 @@ async def update_music_queue():
video = voice_queue.pop() video = voice_queue.pop()
if video.ytdl_url: if video.ytdl_url:
await client.send_message(client.get_channel(config["Discord"]["main_channel"]), f"ℹ E' iniziato il download di `{video.ytdl_url}`.") await client.send_message(client.get_channel(config["Discord"]["main_channel"]), f"ℹ E' iniziato il download di `{video.ytdl_url}`.")
success = await video.download() try:
if not success: await video.download()
await client.send_message(client.get_channel(config["Discord"]["main_channel"]), f"⚠️ C'è stato un errore durante il download di `{video.ytdl_url}`") except DurationError:
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).")
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 continue
voice_player = voice_client.create_ffmpeg_player(video.filename) voice_player = voice_client.create_ffmpeg_player(video.filename)
voice_player.start() voice_player.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(client.get_channel(config["Discord"]["main_channel"]), f"▶ Ora in riproduzione in <#{voice_client.channel.id}>:\n"
f"`{video.filename}`") f"`{video.filename}`")
await client.change_presence(game=discord.Game(name="YouTube", type=2)) await client.change_presence(game=discord.Game(name="youtube-dl", type=2))
def process(users_connection=None): def process(users_connection=None):