From 10dd5af3ead590a3bf920a772eb744d4d784b9ed Mon Sep 17 00:00:00 2001 From: wlinator Date: Wed, 4 Sep 2024 03:09:39 -0400 Subject: [PATCH] fix(afk.py): Ignore AFK removal if entry is recent --- tux/cogs/utility/afk.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/tux/cogs/utility/afk.py b/tux/cogs/utility/afk.py index 5807f1b..457646d 100644 --- a/tux/cogs/utility/afk.py +++ b/tux/cogs/utility/afk.py @@ -1,3 +1,6 @@ +from datetime import datetime, timedelta +from zoneinfo import ZoneInfo + import discord from discord.ext import commands @@ -44,16 +47,16 @@ class AFK(commands.Cog): @commands.Cog.listener("on_message") async def remove_afk(self, message: discord.Message): - if not message.guild: - return - - if message.author.bot: + if not message.guild or message.author.bot: return entry = await self.db.get_afk_member(message.author.id, guild_id=message.guild.id) if not entry: return + if entry.since + timedelta(seconds=10) > datetime.now(ZoneInfo("UTC")): + return + assert isinstance(message.author, discord.Member) await self.db.remove_afk(message.author.id)