1 files changed, 138 insertions, 0 deletions
diff --git a/vstf/vstf/controller/reporters/report/candy_generator.py b/vstf/vstf/controller/reporters/report/candy_generator.py
new file mode 100755
index 00000000..ea296550
--- /dev/null
+++ b/vstf/vstf/controller/reporters/report/candy_generator.py
@@ -0,0 +1,138 @@
+##############################################################################
+# 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
+##############################################################################
+
+from vstf.controller.settings.template_settings import TemplateSettings
+from vstf.controller.reporters.report.data_factory import TaskData
+from vstf.controller.database.dbinterface import DbManage
+import vstf.common.candy_text as candy
+import logging
+LOG = logging.getLogger(__name__)
+
+
+class CandyGenerator(object):
+    def __init__(self, task):
+        self._task = task
+
+    def create(self, scenario):
+        context = {}
+
+        sn = 1
+        chapterid = 1
+        name = candy.tuple2text(sn, candy.chapter, chapterid)
+        context[name] = self.create_env()
+
+        sn += 1
+        chapterid += 1
+        name = candy.tuple2text(sn, candy.chapter, chapterid)
+        context[name] = self.create_scenario(scenario)
+
+        template = TemplateSettings()
+        template.set_context(context)
+        LOG.info(template.settings)
+
+    def create_all(self):
+        context = {}
+
+        sn = 1
+        chapterid = 1
+        name = candy.tuple2text(sn, candy.chapter, chapterid)
+        context[name] = self.create_env()
+
+        scenarios = self._task.common.get_scenariolist()
+        for scenario in scenarios:
+            sn += 1
+            chapterid += 1
+            name = candy.tuple2text(sn, candy.chapter, chapterid)
+            context[name] = self.create_scenario(scenario)
+
+        template = TemplateSettings()
+        template.set_context(context)
+        LOG.info(template.settings)
+
+    def create_env(self):
+        env = {
+            "01##title#1": ["System Environment"],
+            "02##table#2": self._task.common.get_systeminfo()
+        }
+        return env
+
+    def create_scenario(self, scenario):
+        scenario_chapter = {
+            "01##title#1": ["Scenario Result"]
+        }
+        scenario_data = getattr(self._task, scenario)
+        test_list = scenario_data.get_testlist()
+        sectionid = 0
+        sn = 1
+        for test in test_list:
+            sn += 1
+            sectionid += 1
+            name = candy.tuple2text(sn, candy.section, sectionid)
+            testid = test.TestID
+            case = test.CaseTag.decode()
+            ttype = test.Type.decode()
+
+            params_info = [
+                "  Case: " + case,
+                "  Test tool: " + test.Tools.decode(),
+                "  vSwitch: " + test.Switch.decode(),
+                "  Protocol: " + test.Protocol.decode(),
+                "  Type: " + ttype
+            ]
+            if ttype in ["frameloss", "throughput"]:
+                draw = {
+                    "style": 1,
+                    "node": candy.plot,
+                    "data": scenario_data.get_framerate_chartdata(case, ttype)
+                }
+                table = scenario_data.get_ratedata(testid, ttype)
+            else:
+                draw = {
+                    "style": 1,
+                    "node": candy.chart,
+                    "data": scenario_data.get_latency_bardata(case)
+                }
+                table = scenario_data.get_latency_tabledata(case)
+            test_section = self.create_test(sectionid, params_info, table, draw)
+            scenario_chapter[name] = test_section
+
+        return scenario_chapter
+
+    def create_test(self, section, info, table, draw):
+        """
+
+        :rtype : dict
+        """
+        sn = 7
+        draw_name = candy.tuple2text(sn, draw["node"], draw["style"])
+        case_section = {
+            "01##title#2": ["Test ID: %s" % section],
+            "02##paragraph#2": ["Parameter"],
+            "03##paragraph#3": info,
+            "04##paragraph#2": ["Result"],
+            "05##table#2": table,
+            "06##space#2": 2,
+            draw_name: draw["data"]
+        }
+        return case_section
+
+
+def main():
+    from vstf.common.log import setup_logging
+    setup_logging(level=logging.DEBUG, log_file="/var/log/vstf/vstf-candy.log", clevel=logging.INFO)
+
+    dbase = DbManage()
+    taskid = dbase.get_last_taskid()
+    task = TaskData(taskid, dbase)
+    creator = CandyGenerator(task)
+
+    creator.create("Tn")
+if __name__ == '__main__':
+    main()
+