fix sanity check

Klemek · Sep 2, 2022 · cb37b05 · cb37b05
1 parent d8e9e48
commit cb37b05
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/logs/channel_logs.py b/src/logs/channel_logs.py
@@ -66,8 +66,8 @@ async def load(
         is_empty = self.last_message_id is None
         try:
             if is_empty:
-                sanity_check = len(await channel.history(limit=1).flatten())
-                if sanity_check != 1:
+                sanity_check = len([message async for message in channel.history(limit=1)])
+                if sanity_check < 1:
                     yield len(self.messages), True
                     return
             # load backward

diff --git a/src/utils/utils.py b/src/utils/utils.py
@@ -116,7 +116,7 @@ def escape_text(text: str) -> str:
     return discord.utils.escape_markdown(discord.utils.escape_mentions(text))
 
 
-class FakeMessage:
+class FakeMessage(discord.abc.Snowflake):
     def __init__(self, id: int):
         self.id = id