testsuites/vstf/vstf_scripts/vstf/controller/reporters/reporter.py

   1 ##############################################################################
   2 # Copyright (c) 2015 Huawei Technologies Co.,Ltd and others.
   3 #
   4 # All rights reserved. This program and the accompanying materials
   5 # are made available under the terms of the Apache License, Version 2.0
   6 # which accompanies this distribution, and is available at
   7 # http://www.apache.org/licenses/LICENSE-2.0
   8 ##############################################################################
   9
  10 import os
  11 import argparse
  12 import logging
  13 import time
  14
  15 from vstf.controller.reporters.report.provider.html_provider import HtmlProvider
  16 from vstf.controller.reporters.report.provider.pdf_provider import PdfProvider
  17 from vstf.controller.settings.template_settings import TemplateSettings
  18 from vstf.controller.reporters.report.data_factory import TaskData
  19 from vstf.controller.reporters.report.html.htmlcreator import HtmlCreator
  20 from vstf.controller.reporters.report.pdf.pdfcreator import PdfCreator
  21 from vstf.controller.database.dbinterface import DbManage
  22 from vstf.controller.settings.mail_settings import MailSettings
  23 from vstf.controller.reporters.mail.sendmail import SendMail
  24 from vstf.controller.settings.html_settings import HtmlSettings
  25 from vstf.controller.reporters.report.candy_generator import CandyGenerator
  26 import vstf.common.constants as cst
  27
  28
  29 LOG = logging.getLogger(__name__)
  30
  31
  32 class Report(object):
  33
  34     def __init__(self, dbase, rpath):
  35         """
  36
  37         :type dbase: object DbManage
  38         """
  39         self._dbase = dbase
  40         self._rpath = "."
  41         self._mail_settings = MailSettings()
  42         if os.path.exists(rpath):
  43             self._rpath = rpath
  44
  45     def create_pdf(self, taskid):
  46         task = TaskData(taskid, self._dbase)
  47         scenario_list = task.common.get_scenariolist()
  48         creator = CandyGenerator(task)
  49         attach_list = []
  50         for scenario in scenario_list:
  51             out_file = os.path.join(
  52                 self._rpath, "vstf_report_%s_%s.pdf" %
  53                 (scenario, time.strftime(
  54                     cst.TIME_FORMAT3)))
  55             LOG.info(out_file)
  56             creator.create(scenario)
  57             info = TemplateSettings()
  58             provider = PdfProvider(info.settings)
  59             reporter = PdfCreator(provider)
  60             reporter.create(out_file)
  61             attach_list.append(out_file)
  62
  63         if attach_list:
  64             self._mail_settings.mset_attach(attach_list)
  65
  66     def create_html(self, taskid):
  67         task = TaskData(taskid, self._dbase)
  68
  69         creator = CandyGenerator(task)
  70         creator.create_all()
  71
  72         html_settings = HtmlSettings()
  73         info = TemplateSettings()
  74         LOG.info(html_settings.settings)
  75
  76         provider = HtmlProvider(info.settings, html_settings.settings)
  77         out_file = os.path.join(self._rpath, "mail.html")
  78         LOG.info(out_file)
  79
  80         html = HtmlCreator(provider)
  81         content = html.create(out_file)
  82
  83         self._mail_settings.mset_subtype('html')
  84         self._mail_settings.mset_content(content)
  85
  86     def report(self, taskid, mail_off):
  87         self._mail_settings = MailSettings()
  88         mail = SendMail(self._mail_settings.settings)
  89         self.create_pdf(taskid)
  90         self.create_html(taskid)
  91         if not mail_off:
  92             mail.send()
  93
  94
  95 def main():
  96     from vstf.common.log import setup_logging
  97     setup_logging(
  98         level=logging.DEBUG,
  99         log_file="/var/log/vstf/vstf-reporter.log",
 100         clevel=logging.INFO)
 101
 102     parser = argparse.ArgumentParser(add_help=True)
 103     parser.add_argument('-rpath',
 104                         action='store',
 105                         default='./',
 106                         type=str,
 107                         help=" the path name of test results  "
 108                         )
 109     parser.add_argument('-mail_off',
 110                         action='store_true',
 111                         help="is need send mail the for the report"
 112                         )
 113     parser.add_argument('--taskid',
 114                         action='store',
 115                         default=-1,
 116                         help="report depand of a history task id."
 117                         )
 118     args = parser.parse_args()
 119     dbase = DbManage()
 120
 121     report = Report(dbase, args.rpath)
 122     if args.taskid == -1:
 123         taskid = dbase.get_last_taskid()
 124     else:
 125         taskid = args.taskid
 126     report.report(taskid, args.mail_off)
 127
 128
 129 if __name__ == '__main__':
 130     main()