141 lines
4.3 KiB
Python
141 lines
4.3 KiB
Python
"""
|
|
Rework the voice commands
|
|
|
|
https://discordpy.readthedocs.io/en/latest/ext/commands/cogs.html
|
|
|
|
"""
|
|
|
|
# IMPORTS - external
|
|
import discord
|
|
from discord.ext import commands
|
|
|
|
# COG INIT
|
|
class voice(commands.Cog):
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
|
|
# COG BODY
|
|
|
|
class VoiceError(Exception):
|
|
pass
|
|
|
|
class YTDLError(Exception):
|
|
pass
|
|
|
|
|
|
# YTDLSource class from:
|
|
# https://gist.github.com/vbe0201/ade9b80f2d3b64643d854938d40a0a2d
|
|
class YTDLSource(discord.PCMVolumeTransformer):
|
|
YTDL_OPTIONS = {
|
|
'format': 'bestaudio/best',
|
|
'extractaudio': True,
|
|
'audioformat': 'mp3',
|
|
'outtmpl': '%(extractor)s-%(id)s-%(title)s.%(ext)s',
|
|
'restrictfilenames': True,
|
|
'noplaylist': True,
|
|
'nocheckcertificate': True,
|
|
'ignoreerrors': False,
|
|
'logtostderr': False,
|
|
'quiet': True,
|
|
'no_warnings': True,
|
|
'default_search': 'auto',
|
|
'source_address': '0.0.0.0',
|
|
}
|
|
|
|
FFMPEG_OPTIONS = {
|
|
'before_options': '-reconnect 1 -reconnect_streamed 1 -reconnect_delay_max 5',
|
|
'options': '-vn',
|
|
}
|
|
|
|
ytdl = youtube_dl.YoutubeDL(YTDL_OPTIONS)
|
|
|
|
def __init__(self, ctx: commands.Context, source: discord.FFmpegPCMAudio, *, data: dict, volume: float = 0.5):
|
|
super().__init__(source, volume)
|
|
|
|
self.requester = ctx.author
|
|
self.channel = ctx.channel
|
|
self.data = data
|
|
|
|
self.uploader = data.get('uploader')
|
|
self.uploader_url = data.get('uploader_url')
|
|
date = data.get('upload_date')
|
|
self.upload_date = date[6:8] + '.' + date[4:6] + '.' + date[0:4]
|
|
self.title = data.get('title')
|
|
self.thumbnail = data.get('thumbnail')
|
|
self.description = data.get('description')
|
|
self.duration = self.parse_duration(int(data.get('duration')))
|
|
self.tags = data.get('tags')
|
|
self.url = data.get('webpage_url')
|
|
self.views = data.get('view_count')
|
|
self.likes = data.get('like_count')
|
|
self.dislikes = data.get('dislike_count')
|
|
self.stream_url = data.get('url')
|
|
|
|
def __str__(self):
|
|
return '**{0.title}** by **{0.uploader}**'.format(self)
|
|
|
|
@classmethod
|
|
async def create_source(cls, ctx: commands.Context, search: str, *, loop: asyncio.BaseEventLoop = None):
|
|
loop = loop or asyncio.get_event_loop()
|
|
|
|
partial = functools.partial(cls.ytdl.extract_info, search, download=False, process=False)
|
|
data = await loop.run_in_executor(None, partial)
|
|
|
|
if data is None:
|
|
raise YTDLError('Couldn\'t find anything that matches `{}`'.format(search))
|
|
|
|
if 'entries' not in data:
|
|
process_info = data
|
|
else:
|
|
process_info = None
|
|
for entry in data['entries']:
|
|
if entry:
|
|
process_info = entry
|
|
break
|
|
|
|
if process_info is None:
|
|
raise YTDLError('Couldn\'t find anything that matches `{}`'.format(search))
|
|
|
|
webpage_url = process_info['webpage_url']
|
|
partial = functools.partial(cls.ytdl.extract_info, webpage_url, download=False)
|
|
processed_info = await loop.run_in_executor(None, partial)
|
|
|
|
if processed_info is None:
|
|
raise YTDLError('Couldn\'t fetch `{}`'.format(webpage_url))
|
|
|
|
if 'entries' not in processed_info:
|
|
info = processed_info
|
|
else:
|
|
info = None
|
|
while info is None:
|
|
try:
|
|
info = processed_info['entries'].pop(0)
|
|
except IndexError:
|
|
raise YTDLError('Couldn\'t retrieve any matches for `{}`'.format(webpage_url))
|
|
|
|
return cls(ctx, discord.FFmpegPCMAudio(info['url'], **cls.FFMPEG_OPTIONS), data=info)
|
|
|
|
@staticmethod
|
|
def parse_duration(duration: int):
|
|
minutes, seconds = divmod(duration, 60)
|
|
hours, minutes = divmod(minutes, 60)
|
|
days, hours = divmod(hours, 24)
|
|
|
|
duration = []
|
|
if days > 0:
|
|
duration.append('{} days'.format(days))
|
|
if hours > 0:
|
|
duration.append('{} hours'.format(hours))
|
|
if minutes > 0:
|
|
duration.append('{} minutes'.format(minutes))
|
|
if seconds > 0:
|
|
duration.append('{} seconds'.format(seconds))
|
|
|
|
return ', '.join(duration)
|
|
# End of copy pasta
|
|
|
|
|
|
# COG ENDING
|
|
def setup(bot):
|
|
bot.add_cog(voice(bot))
|