more files refactoring

src/logs/channel_logs.py

@@ -0,0 +1,74 @@
+from typing import Union, Tuple
+import discord
+
+from . import MessageLog
+from utils import FakeMessage
+
+CHUNK_SIZE = 1000
+FORMAT = 3
+
+
+class ChannelLogs:
+    def __init__(self, channel: Union[discord.TextChannel, dict]):
+        if isinstance(channel, discord.TextChannel):
+            self.id = channel.id
+            self.name = channel.name
+            self.last_message_id = None
+            self.format = FORMAT
+            self.messages = []
+        elif isinstance(channel, dict):
+            self.format = channel["format"] if "format" in channel else None
+            if not self.is_format():
+                return
+            self.id = int(channel["id"])
+            self.name = channel["name"]
+            self.last_message_id = int(channel["last_message_id"])
+            self.messages = [MessageLog(message) for message in channel["messages"]]
+
+    def is_format(self):
+        return self.format == FORMAT
+
+    async def load(self, channel: discord.TextChannel) -> Tuple[int, int]:
+        self.name = channel.name
+        self.channel = channel
+        try:
+            if self.last_message_id is not None:  # append
+                while self.last_message_id != channel.last_message_id:
+                    async for message in channel.history(
+                        limit=CHUNK_SIZE,
+                        after=FakeMessage(self.last_message_id),
+                        oldest_first=True,
+                    ):
+                        self.last_message_id = message.id
+                        m = MessageLog(message)
+                        await m.load(message)
+                        self.messages.insert(0, m)
+                    yield len(self.messages), False
+            else:  # first load
+                last_message_id = None
+                done = 0
+                while done >= CHUNK_SIZE or last_message_id is None:
+                    done = 0
+                    async for message in channel.history(
+                        limit=CHUNK_SIZE,
+                        before=FakeMessage(last_message_id)
+                        if last_message_id is not None
+                        else None,
+                        oldest_first=False,
+                    ):
+                        done += 1
+                        last_message_id = message.id
+                        m = MessageLog(message)
+                        await m.load(message)
+                        self.messages += [m]
+                    yield len(self.messages), False
+                self.last_message_id = channel.last_message_id
+        except discord.errors.HTTPException:
+            return  # When an exception occurs (like Forbidden)
+        yield len(self.messages), True
+
+    def dict(self) -> dict:
+        channel = dict(self.__dict__)
+        channel.pop("channel", None)
+        channel["messages"] = [message.dict() for message in self.messages]
+        return channel