From 5e6f850f28991f2d37a01a3594ccb97e38b6aa78 Mon Sep 17 00:00:00 2001
From: dcore94 <m.lettere@gmail.com>
Date: Mon, 22 Mar 2021 11:28:05 +0100
Subject: [PATCH] added worker for sending emails and example configuration
 file

---
 mail-test-config.cfg  |  14 +++++
 pyexecplugins/Mail.py | 119 ++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 133 insertions(+)
 create mode 100644 mail-test-config.cfg
 create mode 100644 pyexecplugins/Mail.py

diff --git a/mail-test-config.cfg b/mail-test-config.cfg
new file mode 100644
index 0000000..8e8d5b0
--- /dev/null
+++ b/mail-test-config.cfg
@@ -0,0 +1,14 @@
+[common]
+loglevel = debug 
+server = http://conductor-dev.int.d4science.net/api
+#server = https://conductor.pre.d4science.org/api
+#server = http://localhost:8080/api
+threads = 1 
+pollrate = 1
+
+[pymail]
+server=smtp-relay.d4science.org
+user=conductor_dev
+password=d20d6ea975b01bc
+protocol=starttls
+port=587
diff --git a/pyexecplugins/Mail.py b/pyexecplugins/Mail.py
new file mode 100644
index 0000000..6e9664b
--- /dev/null
+++ b/pyexecplugins/Mail.py
@@ -0,0 +1,119 @@
+import json
+import logging
+import smtplib
+from email.message import EmailMessage
+from email.mime.text import MIMEText
+from email.mime.multipart import MIMEMultipart
+
+from pyexecplugins.pyexecplugins import PyExecPlugin
+
+class Plugin(PyExecPlugin):
+    name = "Mail"
+    
+    taskdef = {
+        "name" : "pymail",
+        "description" : "Send mail with pymail worker",
+        "inputKeys" : ["from", "to", "bcc", "cc", "text", "html", "extra_vars"],
+        "outputKeys" : [],
+        "ownerEmail" : "m.lettere@gmail.com"
+    }
+    
+    def getProtocol(self, config):
+        if config is not None and config["pymail"] is not None:
+            return (config["pymail"].get("protocol") or "plain")
+        else:
+            return "plain"
+
+    def getConfig(self, config, key):
+        if config is not None and config["pymail"] is not None:
+            return (config["pymail"].get(key) or self.defaults[key])
+        else:
+            return self.defaults[key]
+
+    def getDefaults(self, protocol):
+        defaults = {"server" : "localhost", "user" : None, "password" : None}
+        if protocol == "ssl":
+            defaults["port"] = 465
+        elif protocol == "starttls":
+            defaults["port"] = 587
+        else:
+            defaults["port"] = 25
+        return defaults
+
+    def __init__(self, data, config):
+        super().__init__(data)
+        if config is not None:
+            self.protocol = self.getProtocol(config)
+            self.defaults = self.getDefaults(self.protocol)
+            self.server = self.getConfig(config, "server")
+            self.port = self.getConfig(config, "port")
+            self.user = self.getConfig(config, "user")
+            self.password = self.getConfig(config, "password")
+
+        logging.getLogger("pyexec").debug("Parsed config")
+
+        self.frm = data.get("from") or "noreply@noreply.org"
+        self.to = data.get("to") or []
+        self.to = [self.to] if isinstance(self.to, str) else self.to
+        self.bcc = data.get("bcc") or []
+        self.bcc = [self.bcc] if isinstance(self.bcc, str) else self.bcc
+        self.cc = data.get("cc") or []
+        self.cc = [self.cc] if isinstance(self.cc, str) else self.cc
+        self.subject = (data.get("subject") or "No subject").format_map(data.get("extra_vars") or {})
+        self.extra_vars = data.get("extra_vars") or {}
+        self.text = data.get("text")
+        if self.text is not None:
+            self.text = self.text.format_map(data.get("extra_vars") or {})
+        self.html = data.get("html")
+        if self.html is not None:
+            self.html = self.html.format_map(data.get("extra_vars") or {})
+
+   
+    def sendMail(self):
+        logging.getLogger("pyexec").debug("Sending mail from %s to %s subj %s ",self.frm, self.to, self.subject)
+        m = self.buildMessage()
+        if self.protocol == "ssl":
+            self.sendSSLMail(m)
+        elif self.protocol == "starttls":
+            self.sendSTARTTLSMail(m)
+        else:
+            self.sendPlainMail(m)
+        logging.getLogger("pyexec").debug("Mail sent.")        
+    
+    def buildMessage(self):
+        if self.html is not None:
+            m = MIMEMultipart("alternative")
+            if self.text is not None:
+                m.attach(MIMEText(self.text, "plain"))
+            else:
+                m.preamble = "You won't be reading this email without multipart/html support"
+            m.attach(MIMEText(self.html, "html"))
+        else:
+            m = EmailMessage()
+            m.set_content(self.text)
+        m["Subject"] = self.subject
+        m["From"] = self.frm
+        m["To"] = ', '.join(self.to)
+        logging.getLogger("pyexec").debug("Message built")
+        return m
+     
+
+    def sendPlainMail(self, m):
+        logging.getLogger("pyexec").debug("Sending mail %s - %s - %s",self.server, self.port, self.user)
+        s = smtplib.SMTP(host=self.server, port=self.port)
+        s.send_message(m)
+        s.quit()
+
+    def sendSSLMail(self):
+        raise Exception("Not yet supported")
+
+    def sendSTARTTLSMail(self, m):
+        logging.getLogger("pyexec").debug("Sending STARTTLS mail %s - %s - %s",self.server, self.port, self.user)
+        s = smtplib.SMTP(host=self.server, port=self.port)
+        s.starttls()
+        s.login(self.user, self.password)
+        s.send_message(m)
+        s.quit()
+    
+    def execute(self):
+        return self.sendMail()