Send log also via mail

Author: cekk

src/redturtle/rsync/adapters/adapter.py

@@ -8,6 +8,11 @@
 from plone import api
 from datetime import datetime
 from redturtle.rsync.scripts.rsync import logger
+from email.message import EmailMessage
+from email.utils import formataddr
+from plone.registry.interfaces import IRegistry
+from Products.CMFPlone.interfaces.controlpanel import IMailSchema
+from zope.component import getUtility
 
 import json
 import requests
@@ -51,6 +56,7 @@ def __init__(self, context, request):
         self.sync_uids = set()
         self.start = datetime.now()
         self.end = None
+        self.send_log_template = None
 
     def requests_retry_session(
         self,
@@ -113,11 +119,13 @@ def log_info(self, msg, type="info", force_sys_log=False):
         """
         style = ""
         if type == "error":
-            style = "padding:5px;background-color:red;color:#fff"
+            style = "padding:2px;background-color:red;color:#fff"
         if type == "warning":
-            style = "padding:5px;background-color:#ff9d00;color:#fff"
-        msg = f"[{datetime.now().strftime('%d-%m-%Y %H:%M:%S')}] {msg}"
-        self.logdata.append(f'<p style="{style}">{self.autolink(msg)}</p>')
+            style = "padding:2px;background-color:#ff9d00;color:#fff"
+        # msg = f"[{datetime.now().strftime('%d-%m-%Y %H:%M:%S')}] {msg}"
+        self.logdata.append(
+            f'<p><span style="{style}">[{datetime.now().strftime("%d-%m-%Y %H:%M:%S")}]</span>&nbsp;{self.autolink(msg)}</p>'
+        )
 
         # print the message on standard output
         if type == "error":
@@ -166,6 +174,46 @@ def write_log(self):
             },
         )
 
+    def send_log(self):
+        """
+        Send the log by email.
+        """
+
+        send_to_email = getattr(self.options, "send_to_email", None)
+        if not send_to_email:
+            return
+        if not self.send_log_template:
+            logger.warning("No email template found, skipping log send by email.")
+            return
+        mailhost = api.portal.get_tool(name="MailHost")
+        if not mailhost:
+            logger.warning("No MailHost found, skipping log send by email.")
+            return
+
+        body_view = api.content.get_view(
+            name=self.send_log_template, context=self.context, request=self.request
+        )
+        body = body_view(logs=self.logdata)
+        encoding = api.portal.get_registry_record(
+            "plone.email_charset", default="utf-8"
+        )
+
+        registry = getUtility(IRegistry)
+        mail_settings = registry.forInterface(IMailSchema, prefix="plone")
+        email_from_address = mail_settings.email_from_address
+        email_from_name = mail_settings.email_from_name
+        mfrom = formataddr((email_from_name, email_from_address))
+
+        msg = EmailMessage()
+        msg.set_content(body)
+        msg["Subject"] = self.log_item_title(start=self.start)
+        msg["From"] = mfrom
+        msg["Reply-To"] = mfrom
+        msg["To"] = send_to_email
+        msg.replace_header("Content-Type", 'text/html; charset="utf-8"')
+
+        mailhost.send(msg, charset=encoding)
+
     def set_args(self, parser):
         """
         Set some additional arguments for the rsync command.

src/redturtle/rsync/scripts/rsync.py

@@ -50,6 +50,13 @@ def get_args(self, args):
             help="Log destination path (relative to Plone site)",
         )
 
+        # email to send the log to
+        parser.add_argument(
+            "--send-to-email",
+            default=None,
+            help="Email address to send the log to",
+        )
+
         # set data source
         group = parser.add_mutually_exclusive_group(required=True)
         group.add_argument("--source-path", help="Local source path")
@@ -87,6 +94,8 @@ def rsync(self):
 
         # finish, write log
         self.adapter.write_log()
+        # send log by email
+        self.adapter.send_log()
         end = datetime.now()
         delta = end - start
         total_seconds = int(delta.total_seconds())
@@ -106,6 +115,9 @@ def _main(args):
         runner.rsync()
         if not getattr(runner.options, "dry_run", False):
             print(f"[{datetime.now()}] COMMIT")
+            transaction.get().note(
+                runner.adapter.log_item_title(start=runner.adapter.start)
+            )
             transaction.commit()