diff options
-rw-r--r-- | bot.py | 25 | ||||
-rw-r--r-- | cogs/logs.py | 43 |
2 files changed, 52 insertions, 16 deletions
@@ -13,17 +13,20 @@ extensions = [ "cogs.help", "cogs.trans", "cogs.player", - "cogs.general" + "cogs.general", + "cogs.logs" ] + + def get_prefix(bot, message): """A callable Prefix for our bot. This could be edited to allow per server prefixes.""" - prefixes = ['steve ', 'STEVE ', '/', '!', '@','Steve '] + prefixes = ['steve ', 'STEVE ', '/', '!', '@', 'Steve '] # Check to see if we are outside of a guild. e.g DM's etc. - #if not message.guild: - # Only allow ? to be used in DMs - # return '?' + # if not message.guild: + # Only allow ? to be used in DMs + # return '?' # If we are in a guild, we allow for the user to mention us or use any of the prefixes in our list. return commands.when_mentioned_or(*prefixes)(bot, message) @@ -55,23 +58,13 @@ class BedrockBot(commands.Bot): await self.process_commands(message) try: - command = message.content.split()[0] + command = message.content.split()[0] except IndexError: pass if command in self.custom_commands: await message.channel.send(self.custom_commands[command]) return - - async def on_message_delete(self, message): - channel = self.get_channel(718187032869994686) - embed = discord.Embed( - title=message.author, - description=message.content, - color=message.author.color, - timestamp=message.created_at - ) - await channel.send(embed=embed) def run(self): super().run(config.token, reconnect=True) diff --git a/cogs/logs.py b/cogs/logs.py new file mode 100644 index 0000000..cd55c36 --- /dev/null +++ b/cogs/logs.py @@ -0,0 +1,43 @@ +from discord.ext import commands +import discord + + +class Logs(commands.Cog): + def __init__(self, bot): + self.bot = bot + + @commands.Cog.listener() + async def on_message_delete(self, message): + channel = self.bot.get_channel(718187032869994686) + embed = discord.Embed( + title='Deleted Message', + color=message.author.color, + timestamp=message.created_at + ) + embed.add_field( + name='User', value=message.author.mention, inline=True) + embed.add_field( + name='Channel', value=message.channel.name, inline=True) + embed.add_field(name='Message', value=message.content, inline=False) + await channel.send(embed=embed) + + @commands.Cog.listener() + async def on_message_edit(self, before, after): + channel = self.bot.get_channel(718187032869994686) + embed = discord.Embed( + title='Edited Message', + color=before.author.color, + timestamp=after.edited_at + ) + embed.add_field( + name='User', value=before.author.mention, inline=True) + embed.add_field( + name='Channel', value=before.channel.name, inline=True) + embed.add_field(name='Original Message', + value=before.content, inline=False) + embed.add_field(name='New Message', value=after.content, inline=False) + await channel.send(embed=embed) + + +def setup(bot): + bot.add_cog(Logs(bot)) |