Refactor media type detection and log when we miss it

LogBot/logbot.py

@@ -7,6 +7,28 @@ from urllib.parse import urlparse
 import jinja2
 from xbotlib import Bot
 
+IMAGE_TYPES = (".jpg", "jpeg", "png", ".gif", ".bmp", ".svg", "eps")
+
+AUDIO_TYPES = (".mp3", ".ogg", ".oga", ".mogg", ".wav", ".m4a", ".webm")
+
+FILE_TYPES = ".pdf"
+
+VIDEO_TYPES = (
+    ".mp4",
+    ".webm",
+    ".flv",
+    ".vob",
+    ".avi",
+    ".mov",
+    ".qt",
+    ".mpg",
+    ".mpeg",
+    ".mp4",
+    ".m2v",
+    ".mpe",
+    ".3gp",
+)
+
 
 def add_to_db(self, message, media_post=None):
     keys = [x for x in self.db[message.room]["messages"].keys()]
@@ -39,36 +61,17 @@ def write_log(self, message, logger):
 
 def download(self, message, logger):
     # define media_type
-    if message.url.lower().endswith(
+    if message.url.lower().endswith(IMAGE_TYPES):
-        (".jpg", "jpeg", "png", ".gif", ".bmp", ".svg", "eps")
-    ):
         media_type = "images"
-    elif message.url.lower().endswith(".pdf"):
+    elif message.url.lower().endswith(FILE_TYPES):
         media_type = "pdf"
-    elif message.url.lower().endswith(
+    elif message.url.lower().endswith(AUDIO_TYPES):
-        (".mp3", ".ogg", ".oga", ".mogg", ".wav", ".m4a", ".webm")
-    ):
         media_type = "audio"
-    elif message.url.lower().endswith(
+    elif message.url.lower().endswith(VIDEO_TYPES):
-        (
-            ".mp4",
-            ".webm",
-            ".flv",
-            ".vob",
-            ".avi",
-            ".mov",
-            ".qt",
-            ".mpg",
-            ".mpeg",
-            ".mp4",
-            ".m2v",
-            ".mpe",
-            ".3gp",
-        )
-    ):
         media_type = "video"
     else:
         media_type = None
+        logger.info(f"Unable to determine media type of {message.url.lower()}")
 
     # download file
     data = urllib.request.urlopen(message.url).read()