adjust project directories

[bottlenecks.git] / testsuites / vstf / vstf_scripts / vstf / controller / reporters / reporter.py

diff --git a/testsuites/vstf/vstf_scripts/vstf/controller/reporters/reporter.py b/testsuites/vstf/vstf_scripts/vstf/controller/reporters/reporter.py
new file mode 100644 (file)
index 0000000..654c9b8
--- /dev/null
+++ b/testsuites/vstf/vstf_scripts/vstf/controller/reporters/reporter.py
@@ -0,0 +1,123 @@
+##############################################################################
+# Copyright (c) 2015 Huawei Technologies Co.,Ltd and others.
+#
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Apache License, Version 2.0
+# which accompanies this distribution, and is available at
+# http://www.apache.org/licenses/LICENSE-2.0
+##############################################################################
+
+import os
+import argparse
+import logging
+import time
+
+from vstf.controller.reporters.report.provider.html_provider import HtmlProvider
+from vstf.controller.reporters.report.provider.pdf_provider import PdfProvider
+from vstf.controller.settings.template_settings import TemplateSettings
+from vstf.controller.reporters.report.data_factory import TaskData
+from vstf.controller.reporters.report.html.htmlcreator import HtmlCreator
+from vstf.controller.reporters.report.pdf.pdfcreator import PdfCreator
+from vstf.controller.database.dbinterface import DbManage
+from vstf.controller.settings.mail_settings import MailSettings
+from vstf.controller.reporters.mail.sendmail import SendMail
+from vstf.controller.settings.html_settings import HtmlSettings
+from vstf.controller.reporters.report.candy_generator import CandyGenerator
+import vstf.common.constants as cst
+
+
+LOG = logging.getLogger(__name__)
+
+
+class Report(object):
+    def __init__(self, dbase, rpath):
+        """
+
+        :type dbase: object DbManage
+        """
+        self._dbase = dbase
+        self._rpath = "."
+        self._mail_settings = MailSettings()
+        if os.path.exists(rpath):
+            self._rpath = rpath
+
+    def create_pdf(self, taskid):
+        task = TaskData(taskid, self._dbase)
+        scenario_list = task.common.get_scenariolist()
+        creator = CandyGenerator(task)
+        attach_list = []
+        for scenario in scenario_list:
+            out_file = os.path.join(self._rpath, "vstf_report_%s_%s.pdf" % (scenario, time.strftime(cst.TIME_FORMAT3)))
+            LOG.info(out_file)
+            creator.create(scenario)
+            info = TemplateSettings()
+            provider = PdfProvider(info.settings)
+            reporter = PdfCreator(provider)
+            reporter.create(out_file)
+            attach_list.append(out_file)
+
+        if attach_list:
+            self._mail_settings.mset_attach(attach_list)
+
+    def create_html(self, taskid):
+        task = TaskData(taskid, self._dbase)
+
+        creator = CandyGenerator(task)
+        creator.create_all()
+
+        html_settings = HtmlSettings()
+        info = TemplateSettings()
+        LOG.info(html_settings.settings)
+
+        provider = HtmlProvider(info.settings, html_settings.settings)
+        out_file = os.path.join(self._rpath, "mail.html")
+        LOG.info(out_file)
+
+        html = HtmlCreator(provider)
+        content = html.create(out_file)
+
+        self._mail_settings.mset_subtype('html')
+        self._mail_settings.mset_content(content)
+
+    def report(self, taskid, mail_off):
+        self._mail_settings = MailSettings()
+        mail = SendMail(self._mail_settings.settings)
+        self.create_pdf(taskid)
+        self.create_html(taskid)
+        if not mail_off:
+            mail.send()
+
+
+def main():
+    from vstf.common.log import setup_logging
+    setup_logging(level=logging.DEBUG, log_file="/var/log/vstf/vstf-reporter.log", clevel=logging.INFO)
+
+    parser = argparse.ArgumentParser(add_help=True)
+    parser.add_argument('-rpath',
+                        action='store',
+                        default='./',
+                        type=str,
+                        help=" the path name of test results  "
+                        )
+    parser.add_argument('-mail_off',
+                        action='store_true',
+                        help="is need send mail the for the report"
+                        )
+    parser.add_argument('--taskid',
+                        action='store',
+                        default=-1,
+                        help="report depand of a history task id."
+                        )
+    args = parser.parse_args()
+    dbase = DbManage()
+
+    report = Report(dbase, args.rpath)
+    if args.taskid == -1:
+        taskid = dbase.get_last_taskid()
+    else:
+        taskid = args.taskid
+    report.report(taskid, args.mail_off)
+
+
+if __name__ == '__main__':
+    main()