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