fixed images not counted

2021-03-14 22:21:57 +01:00
parent 22af84d826
commit a0b9a0865a
1 changed files with 2 additions and 2 deletions
@@ -4,7 +4,7 @@ from datetime import datetime

 from utils import is_extension

-IMAGE_FORMAT = ["gif", "gifv", "png", "jpg", "jpeg", "bmp"]
+IMAGE_FORMAT = [".gif", ".gifv", ".png", ".jpg", ".jpeg", ".bmp"]
 EMBED_IMAGES = ["image", "gifv"]


@@ -39,7 +39,7 @@ class MessageLog:
                if is_extension(attachment.filename, IMAGE_FORMAT):
                    self.image = True
                    break
-            if not self.image:
+            else:
                for embed in message.embeds:
                    if embed.type in EMBED_IMAGES:
                        self.image = True