From a037d469d7698af5bc97f60bae6860bf616f26e7 Mon Sep 17 00:00:00 2001 From: Daniel Date: Fri, 7 Jul 2017 13:43:53 -0400 Subject: [PATCH] Add basic rewrite voice example --- examples/basic_voice.py | 132 ++++++++++++++++++++++++++++++++++++++++ 1 file changed, 132 insertions(+) create mode 100644 examples/basic_voice.py diff --git a/examples/basic_voice.py b/examples/basic_voice.py new file mode 100644 index 000000000..be9716907 --- /dev/null +++ b/examples/basic_voice.py @@ -0,0 +1,132 @@ +import asyncio + +import discord +import youtube_dl + +from discord.ext import commands + +# Suppress noise about console usage from errors +youtube_dl.utils.bug_reports_message = lambda: '' + + +ytdl_format_options = { + 'format': 'bestaudio/best', + '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' # ipv6 addresses cause issues sometimes +} + +ffmpeg_options = { + 'before_options': '-nostdin', + 'options': '-vn' +} + +ytdl = youtube_dl.YoutubeDL(ytdl_format_options) + + +class YTDLSource(discord.PCMVolumeTransformer): + def __init__(self, source, *, data, volume=0.5): + super().__init__(source, volume) + + self.data = data + + self.title = data.get('title') + self.url = data.get('url') + + @classmethod + async def from_url(cls, url, *, loop=None): + loop = loop or asyncio.get_event_loop() + data = await loop.run_in_executor(ytdl.extract_info, url) + + if 'entries' in data: + # take first item from a playlist + data = data['entries'][0] + + filename = ytdl.prepare_filename(data) + return cls(discord.FFmpegPCMAudio(filename, **ffmpeg_options), data=data) + + +class Music: + def __init__(self, bot): + self.bot = bot + + @commands.command() + async def join(self, ctx, *, channel: discord.VoiceChannel): + """Joins a voice channel""" + + if ctx.voice_client is not None: + return await ctx.voice_client.move_to(channel) + + await channel.connect() + + @commands.command() + async def play(self, ctx, *, query): + """Plays a file from the local filesystem""" + + if ctx.voice_client is None: + if ctx.author.voice.channel: + await ctx.author.voice.channel.connect() + else: + return await ctx.send("Not connected to a voice channel.") + + if ctx.voice_client.is_playing(): + ctx.voice_client.stop() + + source = discord.PCMVolumeTransformer(discord.FFmpegPCMAudio(query)) + ctx.voice_client.play(source, after=lambda e: print('Player error: %s' % e) if e else None) + + await ctx.send('Now playing: {}'.format(query)) + + @commands.command() + async def yt(self, ctx, *, url): + """Streams from a url (almost anything youtube_dl supports)""" + + if ctx.voice_client is None: + if ctx.author.voice.channel: + await ctx.author.voice.channel.connect() + else: + return await ctx.send("Not connected to a voice channel.") + + if ctx.voice_client.is_playing(): + ctx.voice_client.stop() + + player = await YoutubeSource.from_url(query, loop=self.bot.loop) + ctx.voice_client.play(player, after=lambda e: print('Player error: %s' % e) if e else None) + + await ctx.send('Now playing: {}'.format(player.title)) + + @commands.command() + async def volume(self, ctx, volume: int): + """Changes the player's volume""" + + if ctx.voice_client is None: + return await ctx.send("Not connected to a voice channel.") + + ctx.voice_client.source.volume = volume + await ctx.send("Changed volume to {}%".format(volume)) + + + @commands.command() + async def stop(self, ctx): + """Stops and disconnects the bot from voice""" + + await ctx.voice_client.disconnect() + + +bot = commands.Bot(command_prefix=commands.when_mentioned_or("!"), + description='Music bot example') + +@bot.event +async def on_ready(): + print('Logged in as {0.id}/{0}'.format(bot.user)) + print('------') + +bot.add_cog(Music(bot)) +bot.run('token')