From 7b04127e5955ca34c6631c2ec83d895a5a709ee1 Mon Sep 17 00:00:00 2001
From: Raiden Sakura <raiden@project-mei.xyz>
Date: Wed, 24 Apr 2024 20:07:35 +0800
Subject: [PATCH] Don't cache messages

---
 logviewer/core/models.py | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/logviewer/core/models.py b/logviewer/core/models.py
index 39a51d9..f6ff9ed 100644
--- a/logviewer/core/models.py
+++ b/logviewer/core/models.py
@@ -23,7 +23,7 @@
         MessagePayload,
     )
 
-cache = {"users": {}, "dm_channels": {}, "messages": {}}
+cache = {"users": {}, "dm_channels": {}}
 
 
 class LogEntry:
@@ -261,7 +261,6 @@ async def refresh_attachment_url(self, attachments: Attachment):
             return attachments
         user = cache["users"][self.author.id] if self.author.id in cache["users"] else None
         dm_channel = cache["dm_channels"][self.author.id] if self.author.id in cache["dm_channels"] else None
-        discord_message = cache["messages"][self.id] if self.id in cache["messages"] else None
         update_attachments, to_save = False, []
         for i, attachment in enumerate(attachments):
             if not attachment.is_attachment_expired:
@@ -273,9 +272,7 @@ async def refresh_attachment_url(self, attachments: Attachment):
                 dm_channel: DMChannel = user.dm_channel or await user.create_dm()
                 cache["dm_channels"][self.author.id] = dm_channel
             try:
-                if not discord_message:
-                    discord_message = await user.dm_channel.fetch_message(self.id)
-                    cache["messages"][self.id] = discord_message
+                discord_message = await user.dm_channel.fetch_message(self.id)
                 attachment.url = discord_message.attachments[i].url
                 logger.debug(f"Refreshed Attachment#{i+1} for Message ID {self.id}")
                 if not update_attachments: