umi/cogs/unpin.py
2022-06-23 00:09:34 -05:00

39 lines
1.2 KiB
Python

import discord
from discord.ext.commands import Cog
"""Cog setup function"""
def setup(bot: discord.Bot):
bot.add_cog(Unpin(bot))
class Unpin(Cog):
"""Reply to unpinned messages in the channel they were posted."""
def __init__(self, bot: discord.Bot):
self.bot: discord.Bot = bot
print("Initialized Unpin cog")
@Cog.listener()
async def on_guild_channel_pins_update(
self, channel: discord.TextChannel | discord.Thread, last_pin
):
# get last audit log entry
guild = channel.guild
entries = await guild.audit_logs(limit=1).flatten()
entry = entries[0]
# check it is an unpin event
if entry.action != discord.AuditLogAction.message_unpin:
return
# get the unpinned message
message_id = entry.extra.message_id
message = await channel.fetch_message(message_id)
# make sure the message is actually unpinned
# - quickly pinning after unpinning can cause issues otherwise
if message.pinned:
return
# construct and send message as a reply
url = f"https://discord.com/channels/{guild.id}/{channel.id}/{message.id}"
msg = await channel.send(
embed=discord.Embed(title="Message unpinned", url=url),
reference=message,
mention_author=False
)