Upload the contribution of vstf as bottleneck network framework.

[bottlenecks.git] / vstf / vstf / controller / settings / mail_settings.py

diff --git a/vstf/vstf/controller/settings/mail_settings.py b/vstf/vstf/controller/settings/mail_settings.py
new file mode 100755 (executable)
index 0000000..fd66b5c
--- /dev/null
+++ b/vstf/vstf/controller/settings/mail_settings.py
@@ -0,0 +1,121 @@
+#!/usr/bin/env python
+# -*- coding: utf8 -*-
+# author: wly
+# date: 2015-09-06
+# see license for license details
+
+import logging
+import pprint
+
+import vstf.controller.settings.settings as sets
+import vstf.common.decorator as deco
+from vstf.common.input import raw_choice
+
+LOG = logging.getLogger(__name__)
+
+
+class MailSettings(sets.Settings):
+    def __init__(self, path="/etc/vstf", filename="reporters.mail.mail-settings", mode=sets.SETS_DEFAULT):
+        super(MailSettings, self).__init__(path, filename, mode)
+
+    def _register_func(self):
+        super(MailSettings, self)._register_func()
+        body = set(
+            self._fset['body'].keys()
+        )
+        LOG.debug(body)
+        for item in body:
+            item = item.encode()
+            func_name = "set_%s" % item
+            setattr(self, func_name, self._setting_file(func_name, self._mset['body'], self._fset['body'], item))
+        other = {"attach", "content", "subtype"}
+        for item in other:
+            func_name = "mset_%s" % item
+            setattr(self, func_name, self._setting_memory(func_name, self._mset['body'], item))
+
+        LOG.debug(self.__dict__)
+
+    def sinput(self):
+        if raw_choice("if set mail server"):
+            server = self.raw_server()
+            self.set_server(server)
+
+        if raw_choice("if set mail body"):
+            body = self.raw_body()
+            self.set_body(body)
+        print "%s set finish: " % (self._filename)
+        print "+++++++++++++++++++++++++++++++++++++++++"
+        pprint.pprint(self.settings, indent=4)
+        print "+++++++++++++++++++++++++++++++++++++++++"
+
+    @deco.vstf_input("password", types=str)
+    @deco.vstf_input("username", types=str)
+    @deco.vstf_input('host', types=str)
+    def raw_server(self):
+        print "---------------------------------------"
+        print "Please vstf set mail server info like:"
+        print "    'host': 'localhost',"
+        print "    'username': 'user',['\\n' = None]"
+        print "    'password': '******',['\\n' = None]"
+        print "---------------------------------------"
+
+    @deco.vstf_input("subject", types=str, default='vstf mail')
+    @deco.vstf_input("bcc", types=list, default=[])
+    @deco.vstf_input("cc", types=list, default=[])
+    @deco.vstf_input("to", types=list, default=[])
+    @deco.vstf_input('from', types=list, default=['vstf_from@vstf.com'])
+    def raw_body(self):
+        print "----------------------------------------------------"
+        print "Please vstf set mail server info like:"
+        print "    'from': ['vstf_from@vstf.com'],"
+        print "    'to': ['vstf_to@vstf.com'],"
+        print "    'cc': ['vstf_cc@vstf.com']"
+        print "    'bcc': ['vstf_bcc@vstf.com']"
+        print "    'subject': Vstf Performance Test Report"
+        print "----------------------------------------------------"
+
+
+def unit_test():
+    from vstf.common.log import setup_logging
+    setup_logging(level=logging.DEBUG, log_file="/var/log/vstf/vstf-mail-settings.log", clevel=logging.INFO)
+
+    mail_settings = MailSettings()
+    mail_settings.sinput()
+
+    return
+
+    mail_server = {
+        "host": "localhost",
+        "username": None,
+        "password": None
+    }
+    mail_settings.set_server(mail_server)
+
+    from_list = ['vstf_from@vstf.com']
+    mail_settings.set_from(from_list)
+    to_list = ['wangli11@huawei.com']
+    mail_settings.set_to(to_list)
+    cc_list = ['wangli11@huawei.com']
+    mail_settings.set_cc(cc_list)
+    bcc_list = ['wangli11@huawei.com']
+    mail_settings.set_bcc(bcc_list)
+    bcc_list = ['wangli11@huawei.com']
+    mail_settings.set_bcc(bcc_list)
+
+    subject = "Virtual Switching Performance Test Report"
+    mail_settings.set_subject(subject)
+
+    subtype = "plain"
+    mail_settings.mset_subtype(subtype)
+
+    attach_list = []
+    mail_settings.mset_attach(attach_list)
+
+    content = "this is a test"
+    mail_settings.mset_content(content)
+
+    LOG.info(mail_settings.settings)
+
+
+if __name__ == '__main__':
+    unit_test()