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
|
"""This provides bundle clauses with translation functionality."""
__revision__ = '$Revision$'
import copy
import lxml.etree
import re
import Bcfg2.Server.Plugin
try:
import genshi.template
import genshi.template.base
import Bcfg2.Server.Plugins.SGenshi
have_genshi = True
except:
have_genshi = False
class BundleFile(Bcfg2.Server.Plugin.StructFile):
def get_xml_value(self, metadata):
bundlename = self.name.split('/')[-1][:-4]
bundle = lxml.etree.Element('Bundle', name=bundlename)
[bundle.append(copy.deepcopy(item)) for item in self.Match(metadata)]
return bundle
class Bundler(Bcfg2.Server.Plugin.Plugin,
Bcfg2.Server.Plugin.Structure,
Bcfg2.Server.Plugin.XMLDirectoryBacked):
"""The bundler creates dependent clauses based on the
bundle/translation scheme from Bcfg1.
"""
name = 'Bundler'
__version__ = '$Id$'
__author__ = 'bcfg-dev@mcs.anl.gov'
patterns = re.compile('^(?P<name>.*)\.(xml|genshi)$')
def __init__(self, core, datastore):
Bcfg2.Server.Plugin.Plugin.__init__(self, core, datastore)
Bcfg2.Server.Plugin.Structure.__init__(self)
self.encoding = core.encoding
self.__child__ = self.template_dispatch
try:
Bcfg2.Server.Plugin.XMLDirectoryBacked.__init__(self,
self.data,
self.core.fam)
except OSError:
self.logger.error("Failed to load Bundle repository")
raise Bcfg2.Server.Plugin.PluginInitError
def template_dispatch(self, name):
bundle = lxml.etree.parse(name)
nsmap = bundle.getroot().nsmap
if name.endswith('.xml'):
if have_genshi and \
(nsmap == {'py': 'http://genshi.edgewall.org/'}):
# allow for genshi bundles with .xml extensions
spec = Bcfg2.Server.Plugin.Specificity()
return Bcfg2.Server.Plugins.SGenshi.SGenshiTemplateFile(name,
spec,
self.encoding)
else:
return BundleFile(name)
elif name.endswith('.genshi'):
if have_genshi:
spec = Bcfg2.Server.Plugin.Specificity()
return Bcfg2.Server.Plugins.SGenshi.SGenshiTemplateFile(name,
spec,
self.encoding)
def BuildStructures(self, metadata):
"""Build all structures for client (metadata)."""
bundleset = []
for bundlename in metadata.bundles:
entries = [item for (key, item) in list(self.entries.items()) if \
self.patterns.match(key).group('name') == bundlename]
if len(entries) == 0:
continue
elif len(entries) == 1:
try:
bundleset.append(entries[0].get_xml_value(metadata))
except genshi.template.base.TemplateError, t:
self.logger.error("Bundler: Failed to template genshi bundle %s" \
% (bundlename))
self.logger.error(t)
except:
self.logger.error("Bundler: Unexpected bundler error for %s" \
% (bundlename), exc_info=1)
else:
self.logger.error("Got multiple matches for bundle %s" \
% (bundlename))
return bundleset
|
