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
|
import os
import sys
import lxml.etree
import logging
import Bcfg2.Server.Plugin
logger = logging.getLogger("Packages")
class PackagesSources(Bcfg2.Server.Plugin.SingleXMLFileBacked,
Bcfg2.Server.Plugin.StructFile):
__identifier__ = None
def __init__(self, filename, cachepath, fam, packages, config):
try:
Bcfg2.Server.Plugin.SingleXMLFileBacked.__init__(self,
filename,
fam)
except OSError:
err = sys.exc_info()[1]
msg = "Packages: Failed to read configuration file: %s" % err
if not os.path.exists(self.name):
msg += " Have you created it?"
logger.error(msg)
raise Bcfg2.Server.Plugin.PluginInitError(msg)
Bcfg2.Server.Plugin.StructFile.__init__(self, filename)
self.cachepath = cachepath
self.config = config
if not os.path.exists(self.cachepath):
# create cache directory if needed
os.makedirs(self.cachepath)
self.pkg_obj = packages
self.loaded = False
def Index(self):
Bcfg2.Server.Plugin.SingleXMLFileBacked.Index(self)
self.entries = []
for xsource in self.xdata.findall('.//Source'):
source = self.source_from_xml(xsource)
if source is not None:
self.entries.append(source)
self.pkg_obj.Reload()
self.loaded = True
def source_from_xml(self, xsource):
""" create a *Source object from its XML representation in
sources.xml """
stype = xsource.get("type")
if stype is None:
logger.error("No type specified for source, skipping")
return None
try:
module = getattr(__import__("Bcfg2.Server.Plugins.Packages.%s" %
stype.title()).Server.Plugins.Packages,
stype.title())
cls = getattr(module, "%sSource" % stype.title())
except (ImportError, AttributeError):
logger.error("Unknown source type %s" % stype)
return None
return cls(self.cachepath, xsource, self.config)
def __getitem__(self, key):
return self.entries[key]
|
