#!/usr/bin/python
# -*- coding: utf8 -*-
# author: wly
# date: 2015-05-29
# see license for license details
import os
import argparse
import logging
import time

from vstf.controller.reporters.report.pdf.pdfcreator import PdfvSwitchCreator
from vstf.controller.reporters.report.html.htmlcreator import HtmlvSwitchCreator
from vstf.controller.reporters.report.data_factory import CommonData, TaskData, ScenarioData, HistoryData
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.provider.html_provider import StyleProvider
import vstf.common.constants as cst


__version__ = ''' '''
LOG = logging.getLogger(__name__)


class Report(object):
    def __init__(self, dbase, rpath):
        """

        :type dbase: object DbManage
        """
        self._dbase = dbase
        self._rpath = "."
        if os.path.exists(rpath):
            self._rpath = rpath

    def create_pdf(self, taskid):
        common_data = CommonData(taskid, self._dbase)
        scenario_list = common_data.get_scenariolist()
        history_data = HistoryData(taskid, self._dbase)
        attach_list = []
        for scenario in scenario_list:
            out_file = os.path.join(self._rpath, "vstf_report_%s_%s.pdf" % (scenario, time.strftime(cst.TIME_STR)))
            LOG.info(out_file)
            scenario_data = ScenarioData(taskid, self._dbase, scenario)
            pdf = PdfvSwitchCreator(out_file, common_data, scenario_data, history_data)
            if pdf:
                pdf.create()
                attach_list.append(out_file)
        if attach_list:
            self._mail_settings.mset_attach(attach_list)

    def create_html(self, taskid):
        task_data = TaskData(taskid, self._dbase)

        html_settings = HtmlSettings()
        LOG.info(html_settings.settings)

        provider = StyleProvider(html_settings.settings)
        out_file = os.path.join(self._rpath, "mail.html")
        LOG.info(out_file)

        html = HtmlvSwitchCreator(task_data, provider, out_file)
        content = html.create()

        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()