Allow to run directly from a git checkout (#1037)

Signed-off-by: Sol Jerome <sol.jerome@gmail.com>

1 files changed, 97 insertions, 0 deletions

diff --git a/src/lib/Bcfg2/Server/Plugins/Packages/PackagesSources.py b/src/lib/Bcfg2/Server/Plugins/Packages/PackagesSources.py
new file mode 100644
index 000000000..3511cfc3d
--- /dev/null
+++ b/src/lib/Bcfg2/Server/Plugins/Packages/PackagesSources.py
@@ -0,0 +1,97 @@
+import os
+import sys
+import lxml.etree
+import Bcfg2.Server.Plugin
+from Bcfg2.Server.Plugins.Packages.Source import SourceInitError
+
+class PackagesSources(Bcfg2.Server.Plugin.SingleXMLFileBacked,
+                      Bcfg2.Server.Plugin.StructFile,
+                      Bcfg2.Server.Plugin.Debuggable):
+    __identifier__ = None
+
+    def __init__(self, filename, cachepath, fam, packages, config):
+        Bcfg2.Server.Plugin.Debuggable.__init__(self)
+        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?"
+            self.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
+            try:
+                os.makedirs(self.cachepath)
+            except OSError:
+                err = sys.exc_info()[1]
+                self.logger.error("Could not create Packages cache at %s: %s" %
+                                  (self.cachepath, err))
+        self.pkg_obj = packages
+        self.parsed = set()
+        self.loaded = False
+
+    def toggle_debug(self):
+        Bcfg2.Server.Plugin.Debuggable.toggle_debug(self)
+        for source in self.entries:
+            source.toggle_debug()
+
+    def HandleEvent(self, event=None):
+        Bcfg2.Server.Plugin.SingleXMLFileBacked.HandleEvent(self, event=event)
+        if event.filename != self.name:
+            self.parsed.add(os.path.basename(event.filename))
+
+        if sorted(list(self.parsed)) == sorted(self.extras):
+            self.logger.info("Reloading Packages plugin")
+            self.pkg_obj.Reload()
+            self.loaded = True
+
+    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)
+
+    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:
+            self.logger.error("Packages: 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):
+            self.logger.error("Packages: Unknown source type %s" % stype)
+            return None
+
+        try:
+            source = cls(self.cachepath, xsource, self.config)
+        except SourceInitError:
+            err = sys.exc_info()[1]
+            self.logger.error("Packages: %s" % err)
+            source = None
+
+        return source
+
+    def __getitem__(self, key):
+        return self.entries[key]
+
+    def __repr__(self):
+        return "PackagesSources: %s" % repr(self.entries)
+
+    def __str__(self):
+        return "PackagesSources: %s" % str(self.entries)