voice.py small fixes

This commit is contained in:
CramMK
2020-03-25 12:03:06 +01:00
parent 194d0f7274
commit 374a3dd5ed

View File

@@ -8,7 +8,7 @@ https://discordpy.readthedocs.io/en/latest/ext/commands/cogs.html
https://stackoverflow.com/questions/56060614/how-to-make-a-discord-bot-play-youtube-audio https://stackoverflow.com/questions/56060614/how-to-make-a-discord-bot-play-youtube-audio
""" """
# IMPORTS # IMPORTS - external
import discord import discord
from discord.ext import commands from discord.ext import commands
from discord.utils import get from discord.utils import get
@@ -17,6 +17,9 @@ import os
import glob import glob
import youtube_dl import youtube_dl
# IMPORTS - internal
import loadconfig
# COG INIT # COG INIT
class Voice(commands.Cog): class Voice(commands.Cog):
def __init__(self, bot): def __init__(self, bot):
@@ -25,7 +28,7 @@ class Voice(commands.Cog):
# COG BODY # COG BODY
@commands.command(name="join", aliases=["j"]) @commands.command(name="join", aliases=["j"])
@commands.guild_only() @commands.guild_only()
async def join(self, ctx): async def join(self, ctx, *):
""" """
Tries to join the author's current voice channel Tries to join the author's current voice channel
""" """
@@ -46,7 +49,7 @@ class Voice(commands.Cog):
@commands.command(name="leave", aliases=["quit"]) @commands.command(name="leave", aliases=["quit"])
@commands.guild_only() @commands.guild_only()
async def leave(self, ctx): async def leave(self, ctx, *):
""" """
Leaves the voice channel, if connected Leaves the voice channel, if connected
""" """
@@ -65,7 +68,7 @@ class Voice(commands.Cog):
@commands.command(name="skip") @commands.command(name="skip")
@commands.guild_only() @commands.guild_only()
async def skip(self, ctx): async def skip(self, ctx, *):
""" """
Skips the current song: WIP! Skips the current song: WIP!
""" """
@@ -82,13 +85,30 @@ class Voice(commands.Cog):
@commands.command(name="play", aliases=["p"]) @commands.command(name="play", aliases=["p"])
@commands.guild_only() @commands.guild_only()
async def play(self, ctx, *query: str): async def play(self, ctx, *query: str, *):
""" """
Plays music from YouTube Plays music from YouTube
""" """
# concat query tuple into string # concat query tuple into string
url = " ".join(query) url = " ".join(query)
# join channel
channel = ctx.message.author.voice.channel
voice = get(self.bot.voice_clients, guild=ctx.guild)
if not channel:
ctx.send("You're not connected to a voice channel!")
if voice and voice.is_connected():
warning = (
"I'm already connected to another channel!\n"
f"Use `{loadconfig.__prefix__}join` for force-join.\n"
)
await ctx.send(warning)
else:
voice = await channel.connect()
await ctx.send(f"`Joined {channel}!`")
youtube_dl.utils.bug_reports_message = lambda: '' youtube_dl.utils.bug_reports_message = lambda: ''
ytdl_format_options = { ytdl_format_options = {
'format': 'bestaudio/best', 'format': 'bestaudio/best',
@@ -146,8 +166,6 @@ class Voice(commands.Cog):
filename = data['url'] if stream else ytdl.prepare_filename(data) filename = data['url'] if stream else ytdl.prepare_filename(data)
return cls(discord.FFmpegPCMAudio(filename, **ffmpeg_options), data=data) return cls(discord.FFmpegPCMAudio(filename, **ffmpeg_options), data=data)
voice = get(self.bot.voice_clients, guild=ctx.guild)
async with ctx.typing(): async with ctx.typing():
player = await YTDLSource.from_url(url, loop=self.bot.loop) player = await YTDLSource.from_url(url, loop=self.bot.loop)
voice.play( voice.play(
@@ -155,11 +173,12 @@ class Voice(commands.Cog):
after=lambda e: print('Player error: %s' % e) if e else None) after=lambda e: print('Player error: %s' % e) if e else None)
player_embed = discord.Embed(colour=discord.Colour.blue()) player_embed = discord.Embed(colour=discord.Colour.blue())
player_embed.set_thumbal(url=player.thumbnail) player_embed.set_thumbnail(url=player.thumbnail)
player_embed.add_field(name="Song", value=player.title) player_embed.add_field(name="Song", value=player.title)
player_embed.add_field(name="Uploader", value=player.uploader) player_embed.add_field(name="Uploader", value=player.uploader)
await ctx.send("`:musical_note: Now playing: `")
await ctx.send(player_embed) await ctx.send(player_embed)
# End of YouTube Player # End of YouTube Player