allow xinclude files to be missing if xi:fallback is provided

3 files changed, 21 insertions, 13 deletions

diff --git a/src/lib/Bcfg2/Server/Lint/Comments.py b/src/lib/Bcfg2/Server/Lint/Comments.py
index f5d0e265f..59d18fc57 100644
--- a/src/lib/Bcfg2/Server/Lint/Comments.py
+++ b/src/lib/Bcfg2/Server/Lint/Comments.py
@@ -1,6 +1,7 @@
-import os.path
+import os
 import lxml.etree
 import Bcfg2.Server.Lint
+from Bcfg2.Server import XI, XI_NAMESPACE
 from Bcfg2.Server.Plugins.Cfg.CfgPlaintextGenerator import CfgPlaintextGenerator
 from Bcfg2.Server.Plugins.Cfg.CfgGenshiGenerator import CfgGenshiGenerator
 from Bcfg2.Server.Plugins.Cfg.CfgCheetahGenerator import CfgCheetahGenerator
@@ -186,7 +187,7 @@ class Comments(Bcfg2.Server.Lint.ServerPlugin):
             path = os.path.join(self.metadata.data, mfile)
             if path in self.files:
                 xdata = lxml.etree.parse(path)
-                for el in xdata.findall('./{http://www.w3.org/2001/XInclude}include'):
+                for el in xdata.findall('./%sinclude' % XI_NAMESPACE):
                     if not self.has_all_xincludes(el.get('href')):
                         self.LintError("broken-xinclude-chain",
                                        "Broken XInclude chain: could not include %s" % path)
diff --git a/src/lib/Bcfg2/Server/Lint/Duplicates.py b/src/lib/Bcfg2/Server/Lint/Duplicates.py
index ee6b7a2e6..60a02ffb9 100644
--- a/src/lib/Bcfg2/Server/Lint/Duplicates.py
+++ b/src/lib/Bcfg2/Server/Lint/Duplicates.py
@@ -1,6 +1,7 @@
-import os.path
+import os
 import lxml.etree
 import Bcfg2.Server.Lint
+from Bcfg2.Server import XI, XI_NAMESPACE
 
 class Duplicates(Bcfg2.Server.Lint.ServerPlugin):
     """ Find duplicate clients, groups, etc. """
@@ -80,7 +81,7 @@ class Duplicates(Bcfg2.Server.Lint.ServerPlugin):
             path = os.path.join(self.metadata.data, mfile)
             if path in self.files:
                 xdata = lxml.etree.parse(path)
-                for el in xdata.findall('./{http://www.w3.org/2001/XInclude}include'):
+                for el in xdata.findall('./%sinclude' % XI_NAMESPACE):
                     if not self.has_all_xincludes(el.get('href')):
                         self.LintError("broken-xinclude-chain",
                                        "Broken XInclude chain: could not include %s" % path)
diff --git a/src/lib/Bcfg2/Server/Lint/Validate.py b/src/lib/Bcfg2/Server/Lint/Validate.py
index aa5cfe62c..5499e75c2 100644
--- a/src/lib/Bcfg2/Server/Lint/Validate.py
+++ b/src/lib/Bcfg2/Server/Lint/Validate.py
@@ -1,10 +1,10 @@
-import fnmatch
+import os
+import sys
 import glob
+import fnmatch
 import lxml.etree
-import os
 from subprocess import Popen, PIPE, STDOUT
-import sys
-
+from Bcfg2.Server import XI, XI_NAMESPACE
 import Bcfg2.Server.Lint
 
 class Validate(Bcfg2.Server.Lint.ServerlessPlugin):
@@ -174,23 +174,29 @@ class Validate(Bcfg2.Server.Lint.ServerlessPlugin):
     def follow_xinclude(self, xfile):
         """ follow xincludes in the given file """
         xdata = lxml.etree.parse(xfile)
-        included = set([ent.get('href') for ent in
-                        xdata.findall('./{http://www.w3.org/2001/XInclude}include')])
+        included = set([el
+                        for el in xdata.findall('./%sinclude' % XI_NAMESPACE)])
         rv = []
 
         while included:
             try:
-                filename = included.pop()
+                el = included.pop()
             except KeyError:
                 continue
+            filename = el.get("href")
 
             path = os.path.join(os.path.dirname(xfile), filename)
-            if self.HandlesFile(path):
+            if not os.path.exists(path):
+                if not el.findall('./%sfallback' % XI_NAMESPACE):
+                    self.LintError("broken-xinclude-chain",
+                                   "XInclude %s does not exist in %s: %s" %
+                                   (filename, xfile, self.RenderXML(el)))
+            elif self.HandlesFile(path):
                 rv.append(path)
                 groupdata = lxml.etree.parse(path)
                 [included.add(el.get('href'))
                  for el in
-                 groupdata.findall('./{http://www.w3.org/2001/XInclude}include')]
+                 groupdata.findall('./%sinclude' % XI_NAMESPACE)]
                 included.discard(filename)
 
         return rv