1 ##############################################################################
2 # Copyright (c) 2015 Huawei Technologies Co.,Ltd and others.
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 ##############################################################################
11 from vstf.controller.reporters.report.pdf.pdftemplate import PdfVswitch
12 from vstf.controller.reporters.report.pdf.story import TitleStory, SpaceStory, ImageStory, TableStory, \
13 LinePlotStory, Story, TableOfContentsStory, PageBreakStory, ParagraphStory, BarChartStory
14 import vstf.common.candy_text as candy
15 from vstf.controller.reporters.report.provider.pdf_provider import PdfProvider
16 from vstf.controller.settings.template_settings import TemplateSettings
21 LOG = logging.getLogger(__name__)
24 class PdfCreator(object):
25 def __init__(self, provider):
26 self._provider = provider
31 theme = self._provider.get_theme
32 self._pdf = PdfVswitch(theme["title"],
39 def save_pdf(self, ofile):
40 self._pdf.generate(self._story, ofile)
42 def add_coverpage(self):
44 story = PageBreakStory(story)
45 self._story += story.storylist
47 def add_contents(self):
48 if self._provider.ifcontents:
50 story = TableOfContentsStory(story)
51 self._story += story.storylist
53 def create_story(self):
58 def create(self, ofile):
63 def add_context(self):
64 context = self._provider.get_context
65 self._story += self._raw_context(context)
67 def _raw_context(self, context, ci=0, si=0, ui=0, level=-1):
69 for key, value in sorted(context.items()):
72 _sn, _node, _style = candy.text2tuple(key)
73 if _node in candy.dom:
74 if _node == candy.chapter:
76 elif _node == candy.section:
80 _story += self._raw_context(value, ci, si, ui, level + 1)
84 LOG.info("node: %s %s" % (_node, candy.title))
85 if _node == candy.title:
87 if level in range(len(candy.dom)):
89 value[0] = "Chapter %s %s" % (ci, value[0])
90 story = PageBreakStory(story)
92 value[0] = "%s.%s %s" % (ci, si, value[0])
94 value[0] = "%s.%s.%s %s" % (ci, si, ui, value[0])
96 story = TitleStory(story, data=value, style=_style)
97 elif _node == candy.table:
98 story = TableStory(story, data=value, style=_style)
99 elif _node == candy.figure:
100 story = ImageStory(story, data=value, style=_style)
101 elif _node == candy.paragraph:
102 story = ParagraphStory(story, data=value, style=_style)
103 elif _node == candy.plot:
104 story = LinePlotStory(story, data=value, style=_style)
105 elif _node == candy.chart:
106 story = BarChartStory(story, data=value, style=_style)
107 elif _node == candy.space:
108 assert isinstance(value, int)
109 for i in range(value):
110 story = SpaceStory(story)
111 _story += story.storylist
116 from vstf.common.log import setup_logging
117 setup_logging(level=logging.DEBUG, log_file="/var/log/pdf-creator.log", clevel=logging.INFO)
119 out_file = "vstf_report.pdf"
121 info = TemplateSettings()
122 provider = PdfProvider(info.settings)
123 reporter = PdfCreator(provider)
124 reporter.create(out_file)
126 if __name__ == '__main__':