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
|
'''This module implements a templating generator based on Genshi'''
__revision__ = '$Revision$'
from genshi.template import TemplateLoader, \
TextTemplate, MarkupTemplate, TemplateError
try:
from genshi.template import NewTextTemplate
have_ntt = True
except:
have_ntt = False
import logging
import Bcfg2.Server.Plugin
import genshi.core, genshi.input
logger = logging.getLogger('Bcfg2.Plugins.TGenshi')
def removecomment(stream):
"""A genshi filter that removes comments from the stream."""
for kind, data, pos in stream:
if kind is genshi.core.COMMENT:
continue
yield kind, data, pos
class TemplateFile:
'''Template file creates Genshi template structures for the loaded file'''
def __init__(self, name, specific, encoding):
self.name = name
self.specific = specific
self.encoding = encoding
if self.specific.all:
matchname = self.name
elif self.specific.group:
matchname = self.name[:self.name.find('.G')]
else:
matchname = self.name[:self.name.find('.H')]
if matchname.endswith('.txt'):
self.template_cls = TextTemplate
elif matchname.endswith('.newtxt'):
if not have_ntt:
logger.error("Genshi NewTextTemplates not supported by this version of Genshi")
else:
self.template_cls = NewTextTemplate
else:
self.template_cls = MarkupTemplate
def handle_event(self, event):
'''Handle all fs events for this template'''
if event.code2str() == 'deleted':
return
try:
loader = TemplateLoader()
self.template = loader.load(self.name, cls=self.template_cls,
encoding=self.encoding)
except TemplateError, terror:
logger.error('Genshi template error: %s' % terror)
except genshi.input.ParseError, perror:
logger.error('Genshi parse error: %s' % perror)
def bind_entry(self, entry, metadata):
'''Build literal file information'''
fname = entry.get('realname', entry.get('name'))
try:
stream = self.template.generate( \
name=fname, metadata=metadata,
path=self.name).filter(removecomment)
if have_ntt:
ttypes = [TextTemplate, NewTextTemplate]
else:
ttypes = [TextTemplate]
if True in [isinstance(self.template, t) for t in ttypes]:
textdata = stream.render('text')
if type(textdata) == unicode:
entry.text = textdata
else:
if self.encoding != 'ascii':
logger.debug("Override encoding of %s TGenshi template to %s" % (self.name, self.encoding))
entry.text = unicode(textdata, self.encoding)
else:
xmldata = stream.render('xml')
if type(xmldata) == unicode:
entry.text = xmldata
else:
if self.encoding != 'ascii':
logger.debug("Override encoding of %s TGenshi template to %s" % (self.name, self.encoding))
entry.text = unicode(xmldata, self.encoding)
except TemplateError, terror:
logger.error('Genshi template error: %s' % terror)
raise Bcfg2.Server.Plugin.PluginExecutionError
except AttributeError, err:
logger.error('Genshi template loading error: %s' % err)
raise Bcfg2.Server.Plugin.PluginExecutionError
class TGenshi(Bcfg2.Server.Plugin.GroupSpool):
'''
The TGenshi generator implements a templating
mechanism for configuration files
'''
name = 'TGenshi'
__version__ = '$Id$'
__author__ = 'jeff@ocjtech.us'
filename_pattern = 'template\.(txt|newtxt|xml)'
es_child_cls = TemplateFile
|
