gigidsss/Plex-Bot-Music
1
0
Fork 0
mirror of https://github.com/jarulsamy/Plex-Bot.git synced 2024-08-19 15:01:55 +02:00
Code Issues Projects Releases Wiki Activity

✨ Add auto message deletion

Browse Source 
Now playing messages now auto delete on song completion.
This commit is contained in:
Joshua Arulsamy 2020-08-08 18:08:51 -06:00
parent 81e0b50620
commit 21c6bcc746
1 changed files with 26 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
PlexBot/bot.py
View File

@ -20,11 +20,29 @@ class General(commands.Cog):
self.bot = bot self.bot = bot
@command() @command()
async def kill(self, ctx): async def kill(self, ctx, *args):
await ctx.send(f"Stopping upon the request of {ctx.author.mention}") if "silent" not in args:
await ctx.send(f"Stopping upon the request of {ctx.author.mention}")
await self.bot.close() await self.bot.close()
logger.info(f"Stopping upon the request of {ctx.author.mention}") logger.info(f"Stopping upon the request of {ctx.author.mention}")
@command()
async def cleanup(self, ctx, limit=250):
channel = ctx.message.channel
try:
async for i in channel.history(limit=limit):
# Only delete messages sent by self
if i.author == self.bot.user:
try:
await i.delete()
except (discord.Forbidden, discord.NotFound, discord.HTTPException):
pass
except discord.Forbidden:
await ctx.send("I don't have the necessary permissions to delete messages.")
class Plex(commands.Cog): class Plex(commands.Cog):
def __init__(self, bot, base_url, plex_token, lib_name, bot_prefix) -> None: def __init__(self, bot, base_url, plex_token, lib_name, bot_prefix) -> None:
@ -44,6 +62,8 @@ class Plex(commands.Cog):
self.vc = None self.vc = None
self.current_track = None self.current_track = None
self.np_message_id = None
self.play_queue = asyncio.Queue() self.play_queue = asyncio.Queue()
self.play_next_event = asyncio.Event() self.play_next_event = asyncio.Event()
@ -64,11 +84,6 @@ class Plex(commands.Cog):
return score[0] return score[0]
@command()
async def hello(self, ctx, *, member: discord.member = None):
member = member or ctx.author
await ctx.send(f"Hello {member}")
async def _play(self): async def _play(self):
track_url = self.current_track.getStreamURL() track_url = self.current_track.getStreamURL()
audio_stream = FFmpegPCMAudio(track_url) audio_stream = FFmpegPCMAudio(track_url)
@ -82,7 +97,7 @@ class Plex(commands.Cog):
logger.debug(f"URL: {track_url}") logger.debug(f"URL: {track_url}")
embed, f = self._build_embed(self.current_track) embed, f = self._build_embed(self.current_track)
await self.ctx.send(embed=embed, file=f) self.np_message_id = await self.ctx.send(embed=embed, file=f)
async def _audio_player_task(self): async def _audio_player_task(self):
while True: while True:
@ -101,6 +116,7 @@ class Plex(commands.Cog):
await self._play() await self._play()
await self.play_next_event.wait() await self.play_next_event.wait()
await self.np_message_id.delete()
def _toggle_next(self, error=None): def _toggle_next(self, error=None):
self.current_track = None self.current_track = None
@ -138,6 +154,8 @@ class Plex(commands.Cog):
@command() @command()
async def play(self, ctx, *args): async def play(self, ctx, *args):
# Save the context to use with async callbacks
self.ctx = ctx
if not len(args): if not len(args):
await ctx.send(f"Usage: {self.bot_prefix}play TITLE_OF_SONG") await ctx.send(f"Usage: {self.bot_prefix}play TITLE_OF_SONG")
@ -165,8 +183,6 @@ class Plex(commands.Cog):
embed, f = self._build_embed(track, t="queue") embed, f = self._build_embed(track, t="queue")
await ctx.send(embed=embed, file=f) await ctx.send(embed=embed, file=f)
# Save the context to use with async callbacks
self.ctx = ctx
# Add the song to the async queue # Add the song to the async queue
await self.play_queue.put(track) await self.play_queue.put(track)