summaryrefslogtreecommitdiffstats
path: root/tools/nagiosgen-convert.py
diff options
context:
space:
mode:
Diffstat (limited to 'tools/nagiosgen-convert.py')
-rwxr-xr-xtools/nagiosgen-convert.py75
1 files changed, 75 insertions, 0 deletions
diff --git a/tools/nagiosgen-convert.py b/tools/nagiosgen-convert.py
new file mode 100755
index 000000000..2c2142735
--- /dev/null
+++ b/tools/nagiosgen-convert.py
@@ -0,0 +1,75 @@
+#!/usr/bin/env python
+
+import os
+import sys
+import lxml.etree
+
+import Bcfg2.Options
+
+def main():
+ opts = {'repo': Bcfg2.Options.SERVER_REPOSITORY}
+ setup = Bcfg2.Options.OptionParser(opts)
+ setup.parse(sys.argv[1:])
+ repo = setup['repo']
+ oldconfigfile = os.path.join(repo, 'Properties', 'NagiosGen.xml')
+ newconfigpath = os.path.join(repo, 'NagiosGen')
+ newconfigfile = os.path.join(newconfigpath, 'config.xml')
+ parentsfile = os.path.join(newconfigpath, 'parents.xml')
+
+ if not os.path.exists(oldconfigfile):
+ print("%s does not exist, nothing to do" % oldconfigfile)
+ return 1
+
+ if not os.path.exists(newconfigpath):
+ print("%s does not exist, cannot write %s" %
+ (newconfigpath, newconfigfile))
+ return 2
+
+ newconfig = lxml.etree.XML("<NagiosGen/>")
+
+ oldconfig = lxml.etree.parse(oldconfigfile)
+ for host in oldconfig.getroot().getchildren():
+ if host.tag == lxml.etree.Comment:
+ # skip comments
+ continue
+
+ if host.tag == 'default':
+ print("default tag will not be converted; use a suitable Group tag instead")
+ continue
+
+ newhost = lxml.etree.Element("Client", name=host.tag)
+ for opt in host:
+ newopt = lxml.etree.Element("Option", name=opt.tag)
+ newopt.text = opt.text
+ newhost.append(newopt)
+ newconfig.append(newhost)
+
+ # parse the parents config, if it exists
+ if os.path.exists(parentsfile):
+ parentsconfig = lxml.etree.parse(parentsfile)
+ for el in parentsconfig.xpath("//Depend"):
+ newhost = newconfig.find("Client[@name='%s']" % el.get("name"))
+ if newhost is not None:
+ newparents = newhost.find("Option[@name='parents']")
+ if newparents is not None:
+ newparents.text += "," + el.get("on")
+ else:
+ newparents = lxml.etree.Element("Option", name="parents")
+ newparents.text = el.get("on")
+ newhost.append(newparents)
+ else:
+ newhost = lxml.etree.Element("Client", name=el.get("name"))
+ newparents = lxml.etree.Element("Option", name="parents")
+ newparents.text = el.get("on")
+ newhost.append(newparents)
+ newconfig.append(newhost)
+
+ try:
+ open(newconfigfile, 'w').write(lxml.etree.tostring(newconfig,
+ pretty_print=True))
+ print("%s written" % newconfigfile)
+ except IOError:
+ print("Failed to write %s" % newconfigfile)
+
+if __name__ == '__main__':
+ sys.exit(main())