Fixes

classes/logger.py

@@ -0,0 +1,21 @@
+import logging
+import threading
+
+from datetime import datetime
+
+
+class Logger:
+    @staticmethod
+    def _format(message: str, severity: str) -> str:
+        thread = threading.current_thread().name
+        datestr = str(datetime.now())
+        return f"{datestr} - {thread} - {severity} - {message}"
+
+    def debug(self, message: str) -> None:
+        print(self.__class__()._format(message, "DEBUG"))
+
+    def info(self, message: str) -> None:
+        print(self.__class__()._format(message, "INFO"))
+
+    def error(self, message: str) -> None:
+        print(self.__class__()._format(message, "ERROR"))

classes/mission.py

@@ -27,14 +27,16 @@ class Mission:
 
         for attachment in self.attachments:
             try:
-                content = open(attachment, "rb").read()
-                message.add_attachment(content)
+                content = open(attachment[0], "rb").read()
+                filename = attachment[0].split("/")[-1]
+                message.add_attachment(content, *(attachment[1].split("/") if len(attachment) > 1 else ["application", "octet-stream"]), filename=filename)
 
             except:
+                raise
                 pass
 
         message["From"] = self.sender
         message["To"] = ",".join(self.recipients)
         message["Subject"] = self.subject
 
-        return message
+        return message

worker.py

@@ -5,8 +5,10 @@ import pathlib
 from classes.config import Config
 from classes.mission import Mission
 from classes.smtp import SMTP
+from classes.logger import Logger
 
 
+logger = Logger()
 config = Config.fromFile("settings.ini")
 
 tempdir = config.getTempDir()
@@ -17,14 +19,16 @@ if not tempdir.exists():
 elif not tempdir.is_dir():
     raise ValueError(f"Temp Dir {str(tempdir)} exists and is not a directory.")
 
-messages = []
-
-for f in tempdir.glob("*.mmm"):
-    mission = Mission.fromFile(f)
-    messages.append(mission.to_message())
-    f.unlink()
-
 smtp = SMTP.fromConfig(config)
 
-for message in messages:
-    smtp.send_message(message)
+for f in tempdir.glob("*.mmm"):
+    logger.debug(f"Found file {f}")
+    try:
+        mission = Mission.fromFile(f)
+        message = mission.to_message()
+        logger.debug("Sending email...")
+        smtp.send_message(message)
+        f.unlink()
+    except Exception as e:
+        logger.error(f"An error has occurred while processing {f}: {e}")
+