vstf/vstf/controller/reporters/report/pdf/pdfcreator.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
  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
  17
  18 import os
  19 import logging
  20
  21 LOG = logging.getLogger(__name__)
  22
  23
  24 class PdfCreator(object):
  25     def __init__(self, provider):
  26         self._provider = provider
  27         self._story = []
  28         self._pdf = None
  29
  30     def create_pdf(self):
  31         theme = self._provider.get_theme
  32         self._pdf = PdfVswitch(theme["title"],
  33                                theme["logo"],
  34                                theme["header"],
  35                                theme["footer"],
  36                                theme["note"],
  37                                theme["style"])
  38
  39     def save_pdf(self, ofile):
  40         self._pdf.generate(self._story, ofile)
  41
  42     def add_coverpage(self):
  43         story = Story()
  44         story = PageBreakStory(story)
  45         self._story += story.storylist
  46
  47     def add_contents(self):
  48         if self._provider.ifcontents:
  49             story = Story()
  50             story = TableOfContentsStory(story)
  51             self._story += story.storylist
  52
  53     def create_story(self):
  54         self.add_coverpage()
  55         self.add_contents()
  56         self.add_context()
  57
  58     def create(self, ofile):
  59         self.create_pdf()
  60         self.create_story()
  61         self.save_pdf(ofile)
  62
  63     def add_context(self):
  64         context = self._provider.get_context
  65         self._story += self._raw_context(context)
  66
  67     def _raw_context(self, context, ci=0, si=0, ui=0, level=-1):
  68         _story = []
  69         for key, value in sorted(context.items()):
  70             LOG.info(key)
  71             LOG.info(value)
  72             _sn, _node, _style = candy.text2tuple(key)
  73             if _node in candy.dom:
  74                 if _node == candy.chapter:
  75                     ci = _style
  76                 elif _node == candy.section:
  77                     si = _style
  78                 else:
  79                     ui = _style
  80                 _story += self._raw_context(value, ci, si, ui, level + 1)
  81
  82             else:
  83                 story = Story()
  84                 LOG.info("node: %s  %s" % (_node, candy.title))
  85                 if _node == candy.title:
  86                     assert value
  87                     if level in range(len(candy.dom)):
  88                         if level == 0:
  89                             value[0] = "Chapter %s %s" % (ci, value[0])
  90                             story = PageBreakStory(story)
  91                         elif level == 1:
  92                             value[0] = "%s.%s %s" % (ci, si, value[0])
  93                         else:
  94                             value[0] = "%s.%s.%s %s" % (ci, si, ui, value[0])
  95                         LOG.info(value)
  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
 112         return _story
 113
 114
 115 def main():
 116     from vstf.common.log import setup_logging
 117     setup_logging(level=logging.DEBUG, log_file="/var/log/pdf-creator.log", clevel=logging.INFO)
 118
 119     out_file = "vstf_report.pdf"
 120
 121     info = TemplateSettings()
 122     provider = PdfProvider(info.settings)
 123     reporter = PdfCreator(provider)
 124     reporter.create(out_file)
 125
 126 if __name__ == '__main__':
 127     main()