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
|
"""IncludeHelper makes it easier to include group- and host-specific files in a template.
Synopsis:
{% python
import os
include = metadata.TemplateHelper['include'].IncludeHelper
custom = include(metadata, path).files(os.path.basename(name))
%}\
{% for file in custom %}\
########## Start ${include.specificity(file)} ##########
{% include ${file} %}
########## End ${include.specificity(file)} ##########
{% end %}\
This would let you include files with the same base name; e.g. in a
template for ''foo.conf'', the include files would be called
''foo.conf.G_<group>.genshi_include''. If a template needs to include
different files in different places, you can do that like so:
inc = metadata.TemplateHelper['include'].IncludeHelper(metadata, path)
custom_bar = inc.files("bar")
custom_baz = inc.files("baz")
This would result in two different sets of custom files being used,
one drawn from ''bar.conf.G_<group>.genshi_include'' and the other
from ''baz.conf.G_<group>.genshi_include''.
==== Methods ====
=== files ===
Usage:
"""
import os
import re
import Bcfg2.Options
__export__ = ["IncludeHelper"]
class IncludeHelper (object):
def __init__(self, metadata, path):
""" Constructor.
The template path can be found in the ''path'' variable that is set for all Genshi templates."""
self.metadata = metadata
self.path = path
def _get_basedir(self):
setup = Bcfg2.Options.OptionParser({'repo':
Bcfg2.Options.SERVER_REPOSITORY})
setup.parse('--')
return os.path.join(setup['repo'], os.path.dirname(self.path))
def files(self, fname):
""" Return a list of files to include for this host. Files
are found in the template directory based on the following
patterns:
* ''<prefix>.H_<hostname>.genshi_include'': Host-specific files
* ''<prefix>.G_<group>.genshi_include'': Group-specific files
Note that there is no numeric priority on the group-specific
files. All matching files are returned by
''IncludeHelper.files()''. """
files = []
hostfile = os.path.join(self._get_basedir(),
"%s.H_%s.genshi_include" %
(fname, self.metadata.hostname))
if os.path.isfile(hostfile):
files.append(hostfile)
for group in self.metadata.groups:
filename = os.path.join(self._get_basedir(),
"%s.G_%s.genshi_include" % (fname, group))
if os.path.isfile(filename):
files.append(filename)
return sorted(files)
@staticmethod
def specificity(fname):
""" Get a string describing the specificity of the given file """
match = re.search(r'(G|H)_(.*)\.genshi_include', fname)
if match:
if match.group(1) == "G":
stype = "group"
else:
stype = "host"
return "%s-specific configs for %s" % (stype, match.group(2))
return "Unknown specificity"
|