pawel02 / image_bot Goto Github PK

General commands:
/help - displays all the available commands
/clear amount - will delete the past messages with the amount specified
Image commands:
/search <keywords> - will change the search to the keyword
/get - will get the image based on the current search
Music commands:
/p <keywords> - finds the song on youtube and plays it in your current channel
/q - displays the current music queue
/skip - skips the current song being played

    #all the music related stuff
    self.is_playing = False

    # 2d array containing [song, channel]
    self.music_queue = []
    self.YDL_OPTIONS = {'format': 'bestaudio', 'noplaylist':'True'}
    self.FFMPEG_OPTIONS = {'before_options': '-reconnect 1 -reconnect_streamed 1 -reconnect_delay_max 5', 'options': '-vn'}

    self.vc = ""

 #searching the item on youtube
def search_yt(self, item):
    with YoutubeDL(self.YDL_OPTIONS) as ydl:
        try: 
            info = ydl.extract_info("ytsearch:%s" % item, download=False)['entries'][0]
        except Exception: 
            return False

    return {'source': info['formats'][0]['url'], 'title': info['title']}

def play_next(self):
    if len(self.music_queue) > 0:
        self.is_playing = True

        #get the first url
        m_url = self.music_queue[0][0]['source']

        #remove the first element as you are currently playing it
        self.music_queue.pop(0)

        self.vc.play(discord.FFmpegPCMAudio(m_url, **self.FFMPEG_OPTIONS), after=lambda e: self.play_next())
    else:
        self.is_playing = False

# infinite loop checking 
async def play_music(self):
    if len(self.music_queue) > 0:
        self.is_playing = True

        m_url = self.music_queue[0][0]['source']
        
        #try to connect to voice channel if you are not already connected

        if self.vc == "" or not self.vc.is_connected():
            self.vc = await self.music_queue[0][1].connect()
        else:
            self.vc = await self.bot.move_to(self.music_queue[0][1])
        
        print(self.music_queue)
        #remove the first element as you are currently playing it
        self.music_queue.pop(0)

        self.vc.play(discord.FFmpegPCMAudio(m_url, **self.FFMPEG_OPTIONS), after=lambda e: self.play_next())
    else:
        self.is_playing = False

@commands.command(name="play", help="Plays a selected song from youtube")
async def p(self, ctx, *args):
    query = " ".join(args)
    
    voice_channel = ctx.author.voice.channel
    if voice_channel is None:
        #you need to be connected so that the bot knows where to go
        await ctx.send("Connect to a voice channel!")
    else:
        song = self.search_yt(query)
        if type(song) == type(True):
            await ctx.send("Could not download the song. Incorrect format try another keyword. This could be due to playlist or a livestream format.")
        else:
            await ctx.send("Song added to the queue")
            self.music_queue.append([song, voice_channel])
            
            if self.is_playing == False:
                await self.play_music()

@commands.command(name="queue", help="Displays the current songs in queue")
async def q(self, ctx):
    retval = ""
    for i in range(0, len(self.music_queue)):
        retval += self.music_queue[i][0]['title'] + "\n"

    print(retval)
    if retval != "":
        await ctx.send(retval)
    else:
        await ctx.send("No music in queue")

@commands.command(name="skip", help="Skips the current song being played")
async def skip(self, ctx):
    if self.vc != "":
        self.vc.stop()
        #try to play next in the queue if it exists
        await self.play_music