From b17b6aff9aefdc8054953929bdb250a51b4c9ad0 Mon Sep 17 00:00:00 2001 From: jsl12 <32917998+jsl12@users.noreply.github.com> Date: Wed, 11 May 2022 23:39:18 -0500 Subject: [PATCH] added the ability to filter with mentions --- src/kwaylon/kwaylon.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/kwaylon/kwaylon.py b/src/kwaylon/kwaylon.py index b98f82c..234dec5 100644 --- a/src/kwaylon/kwaylon.py +++ b/src/kwaylon/kwaylon.py @@ -87,7 +87,8 @@ class Kwaylon(Client): if days >= 1000: await message.reply( - f'https://tenor.com/view/i-hate-you-anakin-darth-vader-vader-star-wars-gif-13071041') + f'https://tenor.com/view/i-hate-you-anakin-darth-vader-vader-star-wars-gif-13071041' + ) return now = datetime.today().astimezone() @@ -105,6 +106,9 @@ class Kwaylon(Client): await message.reply(res) else: + if len(message.mentions) > 0: + df = df[df['auth_id'].isin([m.id for m in message.mentions])] + most = df.sort_values('count').iloc[-1] msg = await self.fetch_message(most) await message.reply(f'Most {emoji_ref} in past {days} days\n{msg.jump_url}')