From 9668d967773849427f39e80a15cb75a68bfe564a Mon Sep 17 00:00:00 2001 From: klemek Date: Mon, 11 Jan 2021 22:08:27 +0100 Subject: [PATCH] fetch answer as mention if available --- src/logs/message_log.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/src/logs/message_log.py b/src/logs/message_log.py index 7e42f90..79c86df 100644 --- a/src/logs/message_log.py +++ b/src/logs/message_log.py @@ -18,12 +18,18 @@ class MessageLog: self.pinned = message.pinned self.mention_everyone = message.mention_everyone self.tts = message.tts - self.reference = ( - message.reference.message_id if message.reference is not None else None - ) self.bot = message.author.bot or message.author.system self.content = message.content self.mentions = message.raw_mentions + if message.reference is not None: + self.reference = message.reference.message_id + if message.reference.resolved is not None: + try: + self.mentions += [message.reference.resolved.author.id] + except AttributeError: + pass + else: + self.reference = None self.role_mentions = message.raw_role_mentions self.channel_mentions = message.raw_channel_mentions self.image = False