Last active
October 4, 2025 12:26
-
Star
(208)
You must be signed in to star a gist -
Fork
(98)
You must be signed in to fork a gist
-
-
Save vbe0201/ade9b80f2d3b64643d854938d40a0a2d to your computer and use it in GitHub Desktop.
Revisions
-
vbe0201 revised this gist
Jul 29, 2019 . 1 changed file with 227 additions and 222 deletions.There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters. Learn more about bidirectional Unicode charactersOriginal file line number Diff line number Diff line change @@ -1,24 +1,19 @@ # -*- coding: utf-8 -*- """ Copyright (c) 2019 Valentin B. A simple music bot written in discord.py using youtube-dl. Though it's a simple example, music bots are complex and require much time and knowledge until they work perfectly. Use this as an example or a base for your own bot and extend it as you want. If there are any bugs, please let me know. Requirements: Python 3.5+ pip install -U discord.py pynacl youtube-dl You also need FFmpeg in your PATH environment variable or the FFmpeg.exe binary in your bot's directory on Windows. """ import asyncio @@ -32,29 +27,20 @@ from async_timeout import timeout from discord.ext import commands # Silence useless bug reports messages youtube_dl.utils.bug_reports_message = lambda: '' class VoiceError(Exception): pass class YTDLError(Exception): pass class YTDLSource(discord.PCMVolumeTransformer): YTDL_OPTIONS = { 'format': 'bestaudio/best', 'extractaudio': True, 'audioformat': 'mp3', @@ -70,23 +56,24 @@ class YTDLSource(discord.PCMVolumeTransformer): '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') @@ -99,36 +86,36 @@ def __init__(self, message, source, *, data, volume=0.5): 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 @@ -138,134 +125,143 @@ async def create_source(cls, message, search: str, *, loop=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) class Song: __slots__ = ('source', 'requester') def __init__(self, source: YTDLSource): self.source = source self.requester = source.requester def create_embed(self): embed = (discord.Embed(title='Now playing', description='```css\n{0.source.title}\n```'.format(self), color=discord.Color.blurple()) .add_field(name='Duration', value=self.source.duration) .add_field(name='Requested by', value=self.requester.mention) .add_field(name='Uploader', value='[{0.source.uploader}]({0.source.uploader_url})'.format(self)) .add_field(name='URL', value='[Click]({0.source.url})'.format(self)) .set_thumbnail(url=self.source.thumbnail)) return embed class SongQueue(asyncio.Queue): def __getitem__(self, item): if isinstance(item, slice): return list(itertools.islice(self._queue, item.start, item.stop, item.step)) else: return self._queue[item] def __iter__(self): return self._queue.__iter__() def __len__(self): return self.qsize() def clear(self): self._queue.clear() def shuffle(self): random.shuffle(self._queue) def remove(self, index: int): del self._queue[index] class VoiceState: def __init__(self, bot: commands.Bot, ctx: commands.Context): self.bot = bot self._ctx = ctx self.current = None self.voice = None self.next = asyncio.Event() self.songs = SongQueue() self._loop = False self._volume = 0.5 self.skip_votes = set() self.audio_player = bot.loop.create_task(self.audio_player_task()) def __del__(self): self.audio_player.cancel() @property def loop(self): return self._loop @loop.setter def loop(self, value: bool): self._loop = value @property def volume(self): return self._volume @volume.setter def volume(self, value: float): self._volume = value @property def is_playing(self): return self.voice and self.current async def audio_player_task(self): while True: self.next.clear() if not self.loop: # Try to get the next song within 3 minutes. # If no song will be added to the queue in time, # the player will disconnect due to performance # reasons. try: async with timeout(180): # 3 minutes self.current = await self.songs.get() except asyncio.TimeoutError: self.bot.loop.create_task(self.stop()) return self.current.source.volume = self._volume self.voice.play(self.current.source, after=self.play_next_song) await self.current.source.channel.send(embed=self.current.create_embed()) await self.next.wait() def play_next_song(self, error=None): if error: raise VoiceError(str(error)) self.next.set() def skip(self): self.skip_votes.clear() if self.is_playing: self.voice.stop() async def stop(self): @@ -277,14 +273,13 @@ async def stop(self): class Music(commands.Cog): def __init__(self, bot: commands.Bot): self.bot = bot self.voice_states = {} def get_voice_state(self, ctx: commands.Context): state = self.voice_states.get(ctx.guild.id) if not state: state = VoiceState(self.bot, ctx) self.voice_states[ctx.guild.id] = state @@ -294,223 +289,233 @@ def cog_unload(self): for state in self.voice_states.values(): self.bot.loop.create_task(state.stop()) def cog_check(self, ctx: commands.Context): if not ctx.guild: raise commands.NoPrivateMessage('This command can\'t be used in DM channels.') return True async def cog_before_invoke(self, ctx: commands.Context): ctx.voice_state = self.get_voice_state(ctx) async def cog_command_error(self, ctx: commands.Context, error: commands.CommandError): await ctx.send('An error occurred: {}'.format(str(error))) @commands.command(name='join', invoke_without_subcommand=True) async def _join(self, ctx: commands.Context): """Joins a voice channel.""" destination = ctx.author.voice.channel if ctx.voice_state.voice: await ctx.voice_state.voice.move_to(destination) return ctx.voice_state.voice = await destination.connect() @commands.command(name='summon') @commands.has_permissions(manage_guild=True) async def _summon(self, ctx: commands.Context, *, channel: discord.VoiceChannel = None): """Summons the bot to a voice channel. If no channel was specified, it joins your channel. """ if not channel and not ctx.author.voice: raise VoiceError('You are neither connected to a voice channel nor specified a channel to join.') destination = channel or ctx.author.voice.channel if ctx.voice_state.voice: await ctx.voice_state.voice.move_to(destination) return ctx.voice_state.voice = await destination.connect() @commands.command(name='leave', aliases=['disconnect']) @commands.has_permissions(manage_guild=True) async def _leave(self, ctx: commands.Context): """Clears the queue and leaves the voice channel.""" if not ctx.voice_state.voice: return await ctx.send('Not connected to any voice channel.') await ctx.voice_state.stop() del self.voice_states[ctx.guild.id] @commands.command(name='volume') async def _volume(self, ctx: commands.Context, *, volume: int): """Sets the volume of the player.""" if not ctx.voice_state.is_playing: return await ctx.send('Nothing being played at the moment.') if 0 > volume > 100: return await ctx.send('Volume must be between 0 and 100') ctx.voice_state.volume = volume / 100 await ctx.send('Volume of the player set to {}%'.format(volume)) @commands.command(name='now', aliases=['current', 'playing']) async def _now(self, ctx: commands.Context): """Displays the currently playing song.""" await ctx.send(embed=ctx.voice_state.current.create_embed()) @commands.command(name='pause') @commands.has_permissions(manage_guild=True) async def _pause(self, ctx: commands.Context): """Pauses the currently playing song.""" if not ctx.voice_state.is_playing and ctx.voice_state.voice.is_playing(): ctx.voice_state.voice.pause() await ctx.message.add_reaction('⏯') @commands.command(name='resume') @commands.has_permissions(manage_guild=True) async def _resume(self, ctx: commands.Context): """Resumes a currently paused song.""" if not ctx.voice_state.is_playing and ctx.voice_state.voice.is_paused(): ctx.voice_state.voice.resume() await ctx.message.add_reaction('⏯') @commands.command(name='stop') @commands.has_permissions(manage_guild=True) async def _stop(self, ctx: commands.Context): """Stops playing song and clears the queue.""" ctx.voice_state.songs.clear() if not ctx.voice_state.is_playing: ctx.voice_state.voice.stop() await ctx.message.add_reaction('⏹') @commands.command(name='skip') async def _skip(self, ctx: commands.Context): """Vote to skip a song. The requester can automatically skip. 3 skip votes are needed for the song to be skipped. """ if not ctx.voice_state.is_playing: return await ctx.send('Not playing any music right now...') voter = ctx.message.author if voter == ctx.voice_state.current.requester: await ctx.message.add_reaction('⏭') ctx.voice_state.skip() elif voter.id not in ctx.voice_state.skip_votes: ctx.voice_state.skip_votes.add(voter.id) total_votes = len(ctx.voice_state.skip_votes) if total_votes >= 3: await ctx.message.add_reaction('⏭') ctx.voice_state.skip() else: await ctx.send('Skip vote added, currently at **{}/3**'.format(total_votes)) else: await ctx.send('You have already voted to skip this song.') @commands.command(name='queue') async def _queue(self, ctx: commands.Context, *, page: int = 1): """Shows the player's queue. You can optionally specify the page to show. Each page contains 10 elements. """ if len(ctx.voice_state.songs) == 0: return await ctx.send('Empty queue.') items_per_page = 10 pages = math.ceil(len(ctx.voice_state.songs) / items_per_page) start = (page - 1) * items_per_page end = start + items_per_page queue = '' for i, song in enumerate(ctx.voice_state.songs[start:end], start=start): queue += '`{0}.` [**{1.source.title}**]({1.source.url})\n'.format(i + 1, song) embed = (discord.Embed(description='**{} tracks:**\n\n{}'.format(len(ctx.voice_state.songs), queue)) .set_footer(text='Viewing page {}/{}'.format(page, pages))) await ctx.send(embed=embed) @commands.command(name='shuffle') async def _shuffle(self, ctx: commands.Context): """Shuffles the queue.""" if len(ctx.voice_state.songs) == 0: return await ctx.send('Empty queue.') ctx.voice_state.songs.shuffle() await ctx.message.add_reaction('✅') @commands.command(name='remove') async def _remove(self, ctx: commands.Context, index: int): """Removes a song from the queue at a given index.""" if len(ctx.voice_state.songs) == 0: return await ctx.send('Empty queue.') ctx.voice_state.songs.remove(index - 1) await ctx.message.add_reaction('✅') @commands.command(name='loop') async def _loop(self, ctx: commands.Context): """Loops the currently playing song. Invoke this command again to unloop the song. """ if not ctx.voice_state.is_playing: return await ctx.send('Nothing being played at the moment.') # Inverse boolean value to loop and unloop. ctx.voice_state.loop = not ctx.voice_state.loop await ctx.message.add_reaction('✅') @commands.command(name='play') async def _play(self, ctx: commands.Context, *, search: str): """Plays a song. If there are songs in the queue, this will be queued until the other songs finished playing. This command automatically searches from various sites if no URL is provided. A list of these sites can be found here: https://rg3.github.io/youtube-dl/supportedsites.html """ if not ctx.voice_state.voice: await ctx.invoke(self._join) async with ctx.typing(): try: source = await YTDLSource.create_source(ctx, search, loop=self.bot.loop) except YTDLError as e: await ctx.send('An error occurred while processing this request: {}'.format(str(e))) else: song = Song(source) await ctx.voice_state.songs.put(song) await ctx.send('Enqueued {}'.format(str(source))) @_join.before_invoke @_play.before_invoke async def ensure_voice_state(self, ctx: commands.Context): if not ctx.author.voice or not ctx.author.voice.channel: raise commands.CommandError('You are not connected to any voice channel.') if ctx.voice_client: if ctx.voice_client.channel != ctx.author.voice.channel: raise commands.CommandError('Bot is already in a voice channel.') bot = commands.Bot('music.', description='Yet another music bot.') bot.add_cog(Music(bot)) @bot.event async def on_ready(): print('Logged in as:\n{0.user.name}\n{0.user.id}'.format(bot)) bot.run('Token') -
vbe0201 revised this gist
Jul 2, 2019 . 1 changed file with 1 addition and 1 deletion.There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters. Learn more about bidirectional Unicode charactersOriginal file line number Diff line number Diff line change @@ -13,7 +13,7 @@ Python 3.6+ ---------------------- discord.py rewrite: pip3 install -U discord.py PyNaCl: pip3 install -U pynacl youtube_dl: pip3 install -U youtube_dl -
vbe0201 revised this gist
Apr 9, 2019 . 1 changed file with 6 additions and 5 deletions.There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters. Learn more about bidirectional Unicode charactersOriginal file line number Diff line number Diff line change @@ -276,7 +276,7 @@ async def stop(self): self.voice = None class Music(commands.Cog): def __init__(self, bot): self.bot = bot self.voice_states = {} @@ -290,20 +290,20 @@ def get_voice_state(self, ctx): return state def cog_unload(self): for state in self.voice_states.values(): self.bot.loop.create_task(state.stop()) def cog_check(self, ctx): if not ctx.guild: raise commands.NoPrivateMessage('This command can\'t be used in a DM channel.') return True async def cog_before_invoke(self, ctx): ctx.state = self.get_voice_state(ctx) async def cog_command_error(self, ctx, error): # This kind of error handling is not really good. It's simple and functional, but not good. # I'd recommend to extend this. await ctx.send(error) @@ -512,4 +512,5 @@ async def ensure_voice(self, ctx): async def on_ready(): print(f'=====================\nLogged in as:\n{bot.user.name}\n{bot.user.id}\n=====================') bot.run('token') -
vbe0201 revised this gist
Nov 17, 2018 . 1 changed file with 0 additions and 1 deletion.There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters. Learn more about bidirectional Unicode charactersOriginal file line number Diff line number Diff line change @@ -229,7 +229,6 @@ async def audio_player_task(self): async with timeout(300): # 5 minutes self.current = await self.songs.get() except asyncio.TimeoutError: return self.bot.loop.create_task(self.stop()) self.current.source.volume = self._volume -
vbe0201 revised this gist
Nov 16, 2018 . 1 changed file with 19 additions and 8 deletions.There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters. Learn more about bidirectional Unicode charactersOriginal file line number Diff line number Diff line change @@ -29,6 +29,7 @@ import discord import youtube_dl from async_timeout import timeout from discord.ext import commands if not discord.opus.is_loaded(): @@ -205,11 +206,12 @@ def __len__(self): class VoiceState: def __init__(self, bot, ctx): self.current = None self.voice = None self._volume = 0.5 self.bot = bot self._ctx = ctx self.next = asyncio.Event() self.songs = SongQueue() self.skip_votes = set() @@ -218,8 +220,17 @@ def __init__(self, bot): async def audio_player_task(self): while True: self.next.clear() # Try to get a song within the next few minutes. # If no song will be added to the queue in time, # the player will disconnect due to performance # reasons. try: async with timeout(300): # 5 minutes self.current = await self.songs.get() except asyncio.TimeoutError: ctx = self._ctx return self.bot.loop.create_task(self.stop()) self.current.source.volume = self._volume self.voice.play(self.current.source, after=self.play_next_song) @@ -271,12 +282,12 @@ def __init__(self, bot): self.bot = bot self.voice_states = {} def get_voice_state(self, ctx): state = self.voice_states.get(ctx.guild.id) if state is None: state = VoiceState(self.bot, ctx) self.voice_states[ctx.guild.id] = state return state @@ -291,7 +302,7 @@ def __local_check(self, ctx): return True async def __before_invoke(self, ctx): ctx.state = self.get_voice_state(ctx) async def __error(self, ctx, error): # This kind of error handling is not really good. It's simple and functional, but not good. -
vbe0201 revised this gist
Nov 12, 2018 . 1 changed file with 4 additions and 4 deletions.There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters. Learn more about bidirectional Unicode charactersOriginal file line number Diff line number Diff line change @@ -21,15 +21,15 @@ you could also place the FFmpeg.exe in your bot's directory. """ import asyncio import functools import itertools import math import random import discord import youtube_dl from discord.ext import commands if not discord.opus.is_loaded(): """ -
vbe0201 revised this gist
Nov 12, 2018 . No changes.There are no files selected for viewing
-
vbe0201 revised this gist
Nov 12, 2018 . 1 changed file with 3 additions and 0 deletions.There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters. Learn more about bidirectional Unicode charactersOriginal file line number Diff line number Diff line change @@ -39,6 +39,9 @@ On Windows this DLL is automatically provided for you. """ discord.opus.load_opus('opus') # Fuck your useless bug reports message that gets two link embeds and confuses users youtube_dl.utils.bug_reports_message = lambda: '' class YTDLError(Exception): -
vbe0201 revised this gist
Sep 17, 2018 . 1 changed file with 1 addition and 1 deletion.There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters. Learn more about bidirectional Unicode charactersOriginal file line number Diff line number Diff line change @@ -98,7 +98,7 @@ def __str__(self): return f'**{self.title}** by **{self.uploader}** *[Duration: {self.duration}]*' @classmethod async def create_source(cls, message, search: str, *, loop=None): loop = loop or asyncio.get_event_loop() partial = functools.partial(cls.ytdl.extract_info, search, download=False, process=False) -
vbe0201 revised this gist
Sep 17, 2018 . 1 changed file with 2 additions and 0 deletions.There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters. Learn more about bidirectional Unicode charactersOriginal file line number Diff line number Diff line change @@ -477,6 +477,8 @@ async def _disconnect(self, ctx): raise MusicError('Not connected to any voice channel.') await ctx.state.stop() # Clear the VoiceState object from the cache. del self.voice_states[str(ctx.guild.id)] @_join.before_invoke @_play.before_invoke -
vbe0201 revised this gist
Sep 17, 2018 . 1 changed file with 6 additions and 4 deletions.There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters. Learn more about bidirectional Unicode charactersOriginal file line number Diff line number Diff line change @@ -10,6 +10,8 @@ Otherwise you'll just struggle. I strongly recommend to improve this example for yourself and not just use it. Requirements: Python 3.6+ ---------------------- discord.py rewrite: pip3 install -U git+https://github.com/Rapptz/discord.py@rewrite PyNaCl: pip3 install -U pynacl @@ -103,7 +105,7 @@ async def create_source(cls, message, search: str, loop=None): data = await loop.run_in_executor(None, partial) if data is None: raise YTDLError(f'Couldn\'t find anything that matches the search query `{search}`') if 'entries' not in data: process_info = data @@ -115,14 +117,14 @@ async def create_source(cls, message, search: str, loop=None): break if process_info is None: raise YTDLError(f'Couldn\'t retrieve any data for the search query `{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(f'Error while trying to fetch the data for the url `{webpage_url}`') if 'entries' not in processed_info: info = processed_info @@ -132,7 +134,7 @@ async def create_source(cls, message, search: str, loop=None): try: info = processed_info['entries'].pop(0) except IndexError: raise YTDLError(f'Couldn\'t retrieve any matches for the url `{webpage_url}`') return cls(message, discord.FFmpegPCMAudio(info['url'], **cls.ffmpeg_opts), data=info) -
vbe0201 revised this gist
Sep 17, 2018 . 1 changed file with 2 additions and 2 deletions.There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters. Learn more about bidirectional Unicode charactersOriginal file line number Diff line number Diff line change @@ -93,7 +93,7 @@ def __init__(self, message, source, *, data, volume=0.5): self.stream_url = data.get('url') def __str__(self): return f'**{self.title}** by **{self.uploader}** *[Duration: {self.duration}]*' @classmethod async def create_source(cls, message, search: str, loop=None): @@ -341,7 +341,7 @@ async def _play(self, ctx, *, search: str): song = Song(ctx.state.voice, source) await ctx.state.songs.put(song) await ctx.send(f'Enqueued {str(source)}') @commands.command(name='volume') async def _volume(self, ctx, *, volume: int): -
vbe0201 revised this gist
Sep 16, 2018 . 1 changed file with 1 addition and 4 deletions.There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters. Learn more about bidirectional Unicode charactersOriginal file line number Diff line number Diff line change @@ -481,10 +481,7 @@ async def _disconnect(self, ctx): async def ensure_voice(self, ctx): if not ctx.author.voice or not ctx.author.voice.channel: raise commands.CommandError('You are not connected to any voice channel.') if ctx.voice_client is not None: if ctx.voice_client.channel != ctx.author.voice.channel: raise MusicError('Bot already in a voice channel.') -
vbe0201 revised this gist
Sep 16, 2018 . 1 changed file with 3 additions and 0 deletions.There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters. Learn more about bidirectional Unicode charactersOriginal file line number Diff line number Diff line change @@ -14,6 +14,9 @@ discord.py rewrite: pip3 install -U git+https://github.com/Rapptz/discord.py@rewrite PyNaCl: pip3 install -U pynacl youtube_dl: pip3 install -U youtube_dl And also either have FFmpeg in your PATH environment variable or on Windows, you could also place the FFmpeg.exe in your bot's directory. """ import discord -
vbe0201 created this gist
Sep 16, 2018 .There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters. Learn more about bidirectional Unicode charactersOriginal file line number Diff line number Diff line change @@ -0,0 +1,498 @@ """ (c) 2018 Vale A simple music bot written in discord.py rewrite using youtube_dl. Though it's a simple example, music bots are very complex and require much time and knowledge until they work perfectly. Use this as an example or a base for your own bot and extend it like you want. If there are any bugs, please let me know. It is important for you to understand what each part of this code does before using it. Otherwise you'll just struggle. I strongly recommend to improve this example for yourself and not just use it. Requirements: ---------------------- discord.py rewrite: pip3 install -U git+https://github.com/Rapptz/discord.py@rewrite PyNaCl: pip3 install -U pynacl youtube_dl: pip3 install -U youtube_dl """ import discord from discord.ext import commands import asyncio import random import functools import itertools import math import youtube_dl if not discord.opus.is_loaded(): """ The opus library here is opus.dll on Windows. Or libopus.so on Linux in the current directory. Replace this with the location where opus is installed and its proper filename. On Windows this DLL is automatically provided for you. """ discord.opus.load_opus('opus') class YTDLError(Exception): pass class MusicError(Exception): pass class YTDLSource(discord.PCMVolumeTransformer): ytdl_opts = { '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_opts = { 'before_options': '-reconnect 1 -reconnect_streamed 1 -reconnect_delay_max 5', 'options': '-vn', } ytdl = youtube_dl.YoutubeDL(ytdl_opts) def __init__(self, message, source, *, data, volume=0.5): super().__init__(source, volume) self.requester = message.author self.channel = message.channel self.data = data self.uploader = data.get('uploader') self.uploader_url = data.get('uploader_url') self.upload_date = f'{data.get("upload_date")[6:8]}.{data.get("upload_date")[4:6]}.{data.get("upload_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 f'**{self.title}** by **{self.uploader}**' @classmethod async def create_source(cls, message, search: str, loop=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(f"Couldn't find anything that matches the search query `{search}`") if 'entries' not in data: process_info = data else: process_info = None for entry in data['entries']: if entry is not None: process_info = entry break if process_info is None: raise YTDLError(f"Couldn't retrieve any data for the search query `{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(f"Error while trying to fetch the data for the url `{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(f"Couldn't retrieve any matches for the url `{webpage_url}`") return cls(message, discord.FFmpegPCMAudio(info['url'], **cls.ffmpeg_opts), data=info) @staticmethod def parse_duration(duration: int): minutes, seconds = divmod(duration, 60) hours, minutes = divmod(minutes, 60) days, hours = divmod(hours, 24) # Create an actual string duration = [] if days > 0: duration.append(f'{days} days') if hours > 0: duration.append(f'{hours} hours') if minutes > 0: duration.append(f'{minutes} minutes') if seconds > 0: duration.append(f'{seconds} seconds') return ', '.join(duration) class Song: def __init__(self, state, source): self.state = state self.source = source self.requester = source.requester def create_embed(self): embed = discord.Embed(title='Now playing:', description=f'```css\n{self.source.title}\n```', color=discord.Color.green()) embed.add_field(name='Duration:', value=self.source.duration) embed.add_field(name='Requested by:', value=self.requester.mention) embed.add_field(name='Uploader:', value=f'[{self.source.uploader}]({self.source.uploader_url})') embed.add_field(name='Song URL:', value=f'[Click here]({self.source.url})') embed.set_thumbnail(url=self.source.thumbnail) return embed class SongQueue(asyncio.Queue): def __iter__(self): return self._queue.__iter__() def clear(self): self._queue.clear() def shuffle(self): random.shuffle(self._queue) def remove(self, value: int): self._queue.rotate(-value) self._queue.pop() self._queue.rotate(value - 1) def __getitem__(self, index): if isinstance(index, slice): return list(itertools.islice(self._queue, index.start, index.stop, index.step)) else: return self._queue[index] def __len__(self): return len(self._queue) class VoiceState: def __init__(self, bot): self.current = None self.voice = None self._volume = 0.5 self.bot = bot self.next = asyncio.Event() self.songs = SongQueue() self.skip_votes = set() self.audio_player = bot.loop.create_task(self.audio_player_task()) async def audio_player_task(self): while True: self.next.clear() self.current = await self.songs.get() self.current.source.volume = self._volume self.voice.play(self.current.source, after=self.play_next_song) await self.current.source.channel.send(embed=self.current.create_embed()) await self.next.wait() @property def volume(self): return self._volume @volume.setter def volume(self, value): self._volume = value if self.voice: self.voice.source.volume = value def is_done(self): if self.voice is None or self.current is None: return True return not self.voice.is_playing() and not self.voice.is_paused() def play_next_song(self, error=None): fut = asyncio.run_coroutine_threadsafe(self.next.set(), self.bot.loop) try: fut.result() except: raise MusicError(error) def skip(self): self.skip_votes.clear() if not self.is_done(): self.voice.stop() async def stop(self): self.songs.clear() if self.voice: await self.voice.disconnect() self.voice = None class Music: def __init__(self, bot): self.bot = bot self.voice_states = {} def get_voice_state(self, guild): state = self.voice_states.get(str(guild.id)) if state is None: state = VoiceState(self.bot) self.voice_states[str(guild.id)] = state return state def __unload(self): for state in self.voice_states.values(): self.bot.loop.create_task(state.stop()) def __local_check(self, ctx): if not ctx.guild: raise commands.NoPrivateMessage('This command can\'t be used in a DM channel.') return True async def __before_invoke(self, ctx): ctx.state = self.get_voice_state(ctx.guild) async def __error(self, ctx, error): # This kind of error handling is not really good. It's simple and functional, but not good. # I'd recommend to extend this. await ctx.send(error) @commands.command(name='join', invoke_without_command=True) async def _join(self, ctx): """Joins a voice channel.""" destination = ctx.author.voice.channel if ctx.state.voice is not None: return await ctx.state.voice.move_to(destination) ctx.state.voice = await destination.connect() @commands.command(name='summon') @commands.has_permissions(manage_guild=True) async def _summon(self, ctx, *, channel: discord.VoiceChannel=None): """Summons the bot to a voice channel. If no channel given, it joins your channel.""" if channel is None and not ctx.author.voice: raise MusicError('You are not connected to a voice channel nor specified a channel to join.') destination = channel or ctx.author.voice.channel if ctx.state.voice is not None: return await ctx.state.voice.move_to(destination) ctx.state.voice = await destination.connect() @commands.command(name='play') async def _play(self, ctx, *, search: str): """Plays a song. If there are currently songs in the queue then it is queued until the next song is done playing. This command automatically searches from YouTube as well. A list of supported sites can be found here: https://rg3.github.io/youtube-dl/supportedsites.html """ if ctx.state.voice is None: await ctx.invoke(self._join) async with ctx.typing(): try: source = await YTDLSource.create_source(ctx.message, search, loop=self.bot.loop) except Exception as e: await ctx.send(f'An error occurred while processing this request: {e}') else: song = Song(ctx.state.voice, source) await ctx.state.songs.put(song) await ctx.send(f'Enqueued {str(source)} *[Duration: {source.duration}]*') @commands.command(name='volume') async def _volume(self, ctx, *, volume: int): """Sets the volume of the currently playing song.""" if ctx.state.is_done(): return await ctx.send('Nothing playing at the moment.') if 0 > volume > 100: return await ctx.send('Volume must be between 0 and 100.') ctx.state.volume = volume / 100 await ctx.send(f'The player\'s volume was set to {volume}%') @commands.command(name='now', aliases=['playing', 'current']) async def _now(self, ctx): """Displays the currently playing song.""" await ctx.send(embed=ctx.state.current.create_embed()) @commands.command(name='pause') @commands.has_permissions(manage_guild=True) async def _pause(self, ctx): """Pauses the currently playing song.""" if not ctx.state.is_done(): ctx.state.voice.pause() await ctx.message.add_reaction('⏯') @commands.command(name='resume') @commands.has_permissions(manage_guild=True) async def _resume(self, ctx): """Resumes a currently paused song.""" if not ctx.state.is_done(): ctx.state.voice.resume() await ctx.message.add_reaction('⏯') @commands.command(name='stop') @commands.has_permissions(manage_guild=True) async def _stop(self, ctx): """Stops playing audio and clears the queue.""" ctx.state.songs.clear() if not ctx.state.is_done(): ctx.state.voice.stop() await ctx.message.add_reaction('⏹') @commands.command(name='skip') async def _skip(self, ctx): """Vote to skip a song. The requester can automatically skip. 3 skip votes are needed for the song to be skipped. """ if ctx.state.is_done(): raise MusicError('Not playing any music right now...') voter = ctx.message.author if voter == ctx.state.current.requester: await ctx.message.add_reaction('⏭') ctx.state.skip() elif voter.id not in ctx.state.skip_votes: ctx.state.skip_votes.add(voter.id) total_votes = len(ctx.state.skip_votes) if total_votes >= 3: await ctx.message.add_reaction('⏭') ctx.state.skip() else: await ctx.send(f'Skip vote added, currently at **{total_votes}/3**') else: await ctx.send('You have already voted to skip this song.') @commands.command(name='queue') async def _queue(self, ctx, *, page: int=1): """Shows the player's queue.""" if len(ctx.state.songs) == 0: raise MusicError('Nothing in the queue.') items_per_page = 10 pages = math.ceil(len(ctx.state.songs) / items_per_page) start = (page - 1) * items_per_page end = start + items_per_page queue = '' for index, song in enumerate(ctx.state.songs[start:end], start=start): queue += f'`{index + 1}.` [**{song.source.title}**]({song.source.url})\n' embed = discord.Embed(color=discord.Color.green(), description=f'**{len(ctx.state.songs)} tracks:**\n\n{queue}') embed.set_footer(text=f'Viewing page {page}/{pages}') await ctx.send(embed=embed) @commands.command(name='shuffle') async def _shuffle(self, ctx): """Shuffles the current queue.""" if len(ctx.state.songs) == 0: raise MusicError('Nothing in the queue.') ctx.state.songs.shuffle() await ctx.message.add_reaction('✅') @commands.command(name='remove') async def _remove(self, ctx, index: int): """Removes an item from the queue with the given index.""" # Please refer from using this command as well as the remove method of SongQueue class. # I implemented this just for completeness. # But as you may have noticed, asyncio.Queue uses a dequeue to actually store the objects. # It's not the sense behind a dequeue to sort out items at specific indexes. # If this is what you want to do, you might better use a different data structure like a list. if len(ctx.state.songs) == 0: raise MusicError('Nothing in the queue.') ctx.state.songs.remove(index) await ctx.message.add_reaction('✅') @commands.command(name='disconnect') @commands.has_permissions(manage_guild=True) async def _disconnect(self, ctx): """Clears the queue and leaves the voice channel.""" if ctx.state.voice is None: raise MusicError('Not connected to any voice channel.') await ctx.state.stop() @_join.before_invoke @_play.before_invoke async def ensure_voice(self, ctx): if not ctx.author.voice or not ctx.author.voice.channel: raise commands.CommandError('You are not connected to any voice channel.') if ctx.voice_client is None: return if ctx.voice_client is not None: if ctx.voice_client.channel != ctx.author.voice.channel: raise MusicError('Bot already in a voice channel.') bot = commands.Bot(command_prefix='music.', description='A shitty, fucked up music bot. You can skid it.') bot.add_cog(Music(bot)) @bot.event async def on_ready(): print(f'=====================\nLogged in as:\n{bot.user.name}\n{bot.user.id}\n=====================') bot.run('token')