1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
##############################################################################
# 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()
|
