Code Review / bottlenecks.git / blobdiff
? search:
summary | shortlog | log | commit | commitdiff | review | tree
raw | inline | side by side
Upload the contribution of vstf as bottleneck network framework.
[bottlenecks.git] / vstf / vstf / controller / reporters / mail / mail.py
diff --git a/vstf/vstf/controller/reporters/mail/mail.py b/vstf/vstf/controller/reporters/mail/mail.py
new file mode 100755 (executable)
index 0000000..42d60b1
--- /dev/null
+++ b/vstf/vstf/controller/reporters/mail/mail.py
@@ -0,0 +1,117 @@
+import smtplib
+import logging
+import os
+from email.mime.application import MIMEApplication
+from email.mime.text import MIMEText
+from email.mime.multipart import MIMEMultipart
+
+LOG = logging.getLogger(__name__)
+SRV = 'localhost'
+USER = None
+PASSWD = None
+
+
+class Mail(object):
+    def __init__(self, srv=SRV, user=USER, passwd=PASSWD):
+        self.srv = srv
+        self.user = USER
+        self.passwd = PASSWD
+        self._msg = MIMEMultipart('mixed')
+
+        # addr type
+        self.TO = "To"
+        self.FROM = "From"
+        self.CC = "Cc"
+        self.BCC = "Bcc"
+        self.__addr_choice = [self.TO, self.FROM, self.CC, self.BCC]
+
+        # text mode
+        self.HTML = "html"
+        self.PLAIN = "plain"
+        self.__mode = [self.HTML, self.PLAIN]
+        # self._charset = 'gb2312'
+
+        # timeout
+        self.timeout = 10
+
+    def attach_addr(self, addr, addr_type):
+        """
+        :param addr: a list of email address.
+        :param addr_type: must be one of [to, from, cc, bcc]
+        """
+        if not addr or not isinstance(addr, list):
+            LOG.error("The addr must be a list")
+            return False
+
+        if addr_type not in self.__addr_choice:
+            LOG.error("Not support addr type")
+            return False
+
+        if not self._msg[addr_type]:
+            self._msg[addr_type] = ','.join(addr)
+        self._msg[addr_type].join(addr)
+
+    def attach_title(self, title):
+        """Notice:
+        each time attach title, the old title will be covered.
+        """
+        if title:
+            self._msg["Subject"] = str(title)
+
+    def attach_text(self, text, mode):
+        if mode not in self.__mode:
+            LOG.error("The text mode not support.")
+            return False
+
+        msg_alternative = MIMEMultipart('alternative')
+        msg_text = MIMEText(text, mode)
+        msg_alternative.attach(msg_text)
+
+        return self._msg.attach(msg_alternative)
+
+    def attach_files(self, files):
+        for _file in files:
+            part = MIMEApplication(open(_file, "rb").read())
+            part.add_header('Content-Disposition', 'attachment', filename=os.path.basename(_file))
+            self._msg.attach(part)
+
+    def send(self):
+        server = smtplib.SMTP(self.srv, timeout=self.timeout)
+        if self.user:
+            server.ehlo()
+            server.starttls()
+            server.ehlo()
+            server.login(self.user, self.passwd)
+        maillist = []
+        if self._msg[self.TO]:
+            maillist += self._msg[self.TO].split(',')
+        if self._msg[self.CC]:
+            maillist += self._msg[self.CC].split(',')
+        if self._msg[self.BCC]:
+            maillist += self._msg[self.BCC].split(',')
+        ret = server.sendmail(self._msg[self.FROM].split(','),
+                              maillist, self._msg.as_string())
+        LOG.info("send mail ret:%s", ret)
+        server.close()
+
+
+if __name__ == "__main__":
+    m = Mail()
+    m.attach_addr(["vstf_server@vstf.com"], m.FROM)
+    m.attach_addr(["wangli11@huawei.com"], m.TO)
+    context = """
+        <!DOCTYPE html>
+        <html>
+        <head>
+        <title>vstf</title>
+        </head>
+        
+        <body>
+            hello vstf
+        </body>
+        
+        </html>
+    """
+    m.attach_text(context, m.HTML)
+    m.attach_title("Email from xeson Check")
+    m.send()