1 files changed, 127 insertions, 0 deletions
diff --git a/testsuites/vstf/vstf_scripts/vstf/controller/reporters/report/pdf/pdfcreator.py b/testsuites/vstf/vstf_scripts/vstf/controller/reporters/report/pdf/pdfcreator.py
new file mode 100644
index 00000000..c33974ec
--- /dev/null
+++ b/testsuites/vstf/vstf_scripts/vstf/controller/reporters/report/pdf/pdfcreator.py
@@ -0,0 +1,127 @@
+##############################################################################
+# 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.reporters.report.pdf.pdftemplate import PdfVswitch
+from vstf.controller.reporters.report.pdf.story import TitleStory, SpaceStory, ImageStory, TableStory, \
+    LinePlotStory, Story, TableOfContentsStory, PageBreakStory, ParagraphStory, BarChartStory
+import vstf.common.candy_text as candy
+from vstf.controller.reporters.report.provider.pdf_provider import PdfProvider
+from vstf.controller.settings.template_settings import TemplateSettings
+
+import os
+import logging
+
+LOG = logging.getLogger(__name__)
+
+
+class PdfCreator(object):
+    def __init__(self, provider):
+        self._provider = provider
+        self._story = []
+        self._pdf = None
+
+    def create_pdf(self):
+        theme = self._provider.get_theme
+        self._pdf = PdfVswitch(theme["title"],
+                               theme["logo"],
+                               theme["header"],
+                               theme["footer"],
+                               theme["note"],
+                               theme["style"])
+
+    def save_pdf(self, ofile):
+        self._pdf.generate(self._story, ofile)
+
+    def add_coverpage(self):
+        story = Story()
+        story = PageBreakStory(story)
+        self._story += story.storylist
+
+    def add_contents(self):
+        if self._provider.ifcontents:
+            story = Story()
+            story = TableOfContentsStory(story)
+            self._story += story.storylist
+
+    def create_story(self):
+        self.add_coverpage()
+        self.add_contents()
+        self.add_context()
+
+    def create(self, ofile):
+        self.create_pdf()
+        self.create_story()
+        self.save_pdf(ofile)
+
+    def add_context(self):
+        context = self._provider.get_context
+        self._story += self._raw_context(context)
+
+    def _raw_context(self, context, ci=0, si=0, ui=0, level=-1):
+        _story = []
+        for key, value in sorted(context.items()):
+            LOG.info(key)
+            LOG.info(value)
+            _sn, _node, _style = candy.text2tuple(key)
+            if _node in candy.dom:
+                if _node == candy.chapter:
+                    ci = _style
+                elif _node == candy.section:
+                    si = _style
+                else:
+                    ui = _style
+                _story += self._raw_context(value, ci, si, ui, level + 1)
+
+            else:
+                story = Story()
+                LOG.info("node: %s  %s" % (_node, candy.title))
+                if _node == candy.title:
+                    assert value
+                    if level in range(len(candy.dom)):
+                        if level == 0:
+                            value[0] = "Chapter %s %s" % (ci, value[0])
+                            story = PageBreakStory(story)
+                        elif level == 1:
+                            value[0] = "%s.%s %s" % (ci, si, value[0])
+                        else:
+                            value[0] = "%s.%s.%s %s" % (ci, si, ui, value[0])
+                        LOG.info(value)
+                        story = TitleStory(story, data=value, style=_style)
+                elif _node == candy.table:
+                    story = TableStory(story, data=value, style=_style)
+                elif _node == candy.figure:
+                    story = ImageStory(story, data=value, style=_style)
+                elif _node == candy.paragraph:
+                    story = ParagraphStory(story, data=value, style=_style)
+                elif _node == candy.plot:
+                    story = LinePlotStory(story, data=value, style=_style)
+                elif _node == candy.chart:
+                    story = BarChartStory(story, data=value, style=_style)
+                elif _node == candy.space:
+                    assert isinstance(value, int)
+                    for i in range(value):
+                        story = SpaceStory(story)
+                _story += story.storylist
+        return _story
+
+
+def main():
+    from vstf.common.log import setup_logging
+    setup_logging(level=logging.DEBUG, log_file="/var/log/pdf-creator.log", clevel=logging.INFO)
+
+    out_file = "vstf_report.pdf"
+
+    info = TemplateSettings()
+    provider = PdfProvider(info.settings)
+    reporter = PdfCreator(provider)
+    reporter.create(out_file)
+
+if __name__ == '__main__':
+    main()