From 69c451d3aa2069b96c77c147c15fdce2a93211be Mon Sep 17 00:00:00 2001
From: klemek <klemek@outlook.fr>
Date: Thu, 7 Jan 2021 18:08:08 +0100
Subject: [PATCH] reading if messages have images

---
 log_manager.py | 17 +++++++++++++++--
 utils.py       | 11 ++++++++++-
 2 files changed, 25 insertions(+), 3 deletions(-)

diff --git a/log_manager.py b/log_manager.py
index 897ac56..93c1ba5 100644
--- a/log_manager.py
+++ b/log_manager.py
@@ -6,7 +6,7 @@ import gzip
 from datetime import datetime
 import logging
 
-from utils import code_message
+from utils import code_message, is_extension
 
 LOG_DIR = "logs"
 
@@ -15,7 +15,9 @@ if not os.path.exists(LOG_DIR):
 
 
 CHUNK_SIZE = 1000
-FORMAT = 2
+FORMAT = 3
+IMAGE_FORMAT = ["gif", "gifv", "png", "jpg", "jpeg", "bmp"]
+EMBED_IMAGES = ["image", "gifv"]
 
 current_analysis = []
 
@@ -43,6 +45,16 @@ class MessageLog:
             self.mentions = message.raw_mentions
             self.role_mentions = message.raw_role_mentions
             self.channel_mentions = message.raw_channel_mentions
+            self.image = False
+            for attachment in message.attachment:
+                if is_extension(attachment.filename, IMAGE_FORMAT):
+                    self.image = True
+                    break
+            if not self.image:
+                for embed in message.embeds:
+                    if embed.type in EMBED_IMAGES:
+                        self.image = True
+                        break
             self.reactions = {}
         elif isinstance(message, dict):
             self.id = int(message["id"])
@@ -64,6 +76,7 @@ class MessageLog:
             self.mentions = [int(m) for m in message["mentions"]]
             self.role_mentions = [int(m) for m in message["role_mentions"]]
             self.channel_mentions = [int(m) for m in message["channel_mentions"]]
+            self.image = message["image"]
             self.reactions = message["reactions"]
 
     async def load(self, message: discord.Message):
diff --git a/utils.py b/utils.py
index 6bed476..b96e8eb 100644
--- a/utils.py
+++ b/utils.py
@@ -1,4 +1,5 @@
 from typing import List, Dict
+import os
 import logging
 import discord
 
@@ -17,6 +18,14 @@ def mention(member_id: int) -> str:
     return f"<@{member_id}>"
 
 
+# FILE
+
+
+def is_extension(filepath: str, ext_list: List[str]) -> bool:
+    filename, file_extension = os.path.splitext(filepath.lower())
+    return file_extension in ext_list
+
+
 # LISTS
 
 
@@ -67,7 +76,7 @@ def day_interval(interval: int) -> str:
 
 def get_intro(
     subject: str,
-    emotes: Dict[str],
+    emotes: Dict[str, object],
     full: bool,
     channels: List[discord.TextChannel],
     members: List[discord.Member],