email log check

Author: sourovamin

myapp/email.py

@@ -3,16 +3,52 @@
 from myapp import app, mail
 from flask_mail import Message
 from werkzeug.utils import secure_filename
+from datetime import datetime
 import io
+import os
 
 APP_NAME=app.config['APP_NAME']
 APP_URL=app.config['APP_URL']
 APP_TITLE=app.config['APP_TITLE']
 
+EMAIL_LOG_DIR=os.path.join(app.config["USERS_DATA"], 'email_logs')
+SEND_FAILED_FILE=os.path.join(EMAIL_LOG_DIR, 'failed.log')
+SEND_SUCCESS_FILE=os.path.join(EMAIL_LOG_DIR, 'success.log')
+
+def write_email_log(file_path, msg, e = None):
+    try:
+        os.makedirs(EMAIL_LOG_DIR, exist_ok=True)
+        current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        with open(file_path, "a") as file:
+            file.write(f"Time:" + str(current_time) + "\n")
+            file.write("Subject: " + str(msg.subject) + "\n")
+            file.write("Sender: " + str(msg.sender) + "\n")
+            file.write("Recipients: " + str(msg.recipients) + "\n")
+            file.write("Body: " + str(msg.body) + "\n")
+            file.write("HTML Body: " + str(msg.html) + "\n")
+            file.write("Reply-To: " + str(msg.reply_to) + "\n")
+            if e is not None:
+                file.write(f"Exception: {e} \n")
+            file.write("-" * 50 + "\n")
+
+        # keep the log file size in check, max 500kb
+        if os.path.exists(file_path) and os.path.getsize(file_path) > 512000:
+            with open(file_path, "r+") as file:
+                lines = file.readlines()
+                if len(lines) > 500:
+                    file.seek(0)
+                    file.writelines(lines[500:])
+                    file.truncate()
+    except Exception as e:
+        print(f"Failed to write email log to file: {e}")
 
 def send_async_email(app, msg):
     with app.app_context():
-        mail.send(msg)
+        try:
+            mail.send(msg)
+            write_email_log(SEND_SUCCESS_FILE, msg)
+        except Exception as e: 
+            write_email_log(SEND_FAILED_FILE, msg, e)
 
 def send_email(subject, sender, recipients, text_body, html_body, reply_to, attachment=None, attachment_path=None, attachment_type=None, open_type="rb"):
     msg = Message(subject, sender=sender, recipients=recipients, reply_to = reply_to)