diff options
Diffstat (limited to 'src/sbin/GenerateHostInfo')
-rw-r--r-- | src/sbin/GenerateHostInfo | 59 |
1 files changed, 25 insertions, 34 deletions
diff --git a/src/sbin/GenerateHostInfo b/src/sbin/GenerateHostInfo index e6d166877..4b34ade89 100644 --- a/src/sbin/GenerateHostInfo +++ b/src/sbin/GenerateHostInfo @@ -6,42 +6,24 @@ __revision__ = '$Revision$' from ConfigParser import ConfigParser -from lxml.etree import Element, SubElement, parse +from lxml.etree import Element, SubElement, parse, dump, tostring from os import fork, execl, dup2, wait import sys -def pretty_print(element, level=0): - '''Produce a pretty-printed text representation of element''' - if element.text: - fmt = "%s<%%s %%s>%%s</%%s>" % (level*" ") - data = (element.tag, (" ".join(["%s='%s'" % keyval for keyval in element.attrib.iteritems()])), - element.text, element.tag) - children = element.getchildren() - if children: - fmt = "%s<%%s %%s>\n" % (level*" ",) + (len(children) * "%s") + "%s</%%s>\n" % (level*" ") - data = (element.tag, ) + (" ".join(["%s='%s'" % (key, element.attrib[key]) for key in element.attrib]),) - data += tuple([pretty_print(entry, level+2) for entry in children]) + (element.tag, ) - else: - fmt = "%s<%%s %%s/>\n" % (level * " ") - data = (element.tag, " ".join(["%s='%s'" % (key, element.attrib[key]) for key in element.attrib])) - return fmt % data - - if __name__ == '__main__': c = ConfigParser() c.read(['/etc/bcfg2.conf']) configpath = "%s/etc/report-configuration.xml" % c.get('server', 'repository') - #hostinfopath = "%s/etc/hostinfo.xml" % c.get('server', 'repository') clientdatapath = "%s/Metadata/clients.xml" % c.get('server', 'repository') sendmailpath = c.get('statistics','sendmailpath') clientElement = parse(clientdatapath) hostlist = [client.get('name') for client in clientElement.findall("Client")] - #HostInfo = Element("HostInformation") pids = {} fullnames = {} null = open('/dev/null', 'w+') + while hostlist or pids: if hostlist and len(pids.keys()) < 15: host = hostlist.pop() @@ -57,22 +39,31 @@ if __name__ == '__main__': else: try: (cpid, status) = wait() - chost = pids[cpid] - del pids[cpid] - if status == 0: - #SubElement(HostInfo, "HostInfo", name=chost, fqdn=chost, pingable='Y') - clientElement.xpath("/Clients/Client[@name=%s]"%chost).set("pingable",'Y') + except OSError: + continue + + chost = pids[cpid] + del pids[cpid] + if status == 0: + #if '-v' in sys.argv: + print "Alive: %s (fullname: %s):"%(chost,fullnames[chost]) + #SubElement(HostInfo, "HostInfo", name=chost, fqdn=chost, pingable='Y') + clientElement.xpath("//Client[@name='%s']"%(fullnames[chost]))[0].set("pingable",'Y') + #also set pingtime, if you can get it + else: + if chost.count('.') > 0: + fullnames[chost.split('.')[0]] = chost + hostlist.append(chost.split('.')[0]) else: - if chost.count('.') > 0: - fullnames[chost.split('.')[0]] = chost - hostlist.append(chost.split('.')[0]) - else: - #SubElement(HostInfo, "HostInfo", name=fullnames[chost], fqdn=fullnames[chost], pingable='N') - clientElement.xpath("/Clients/Client[@name=%s]"%chost).set("pingable",'N') - except: - pass + print "Dead: %s (fullname: %s):"%(chost,fullnames[chost]) + + #SubElement(HostInfo, "HostInfo", name=fullnames[chost], fqdn=fullnames[chost], pingable='N') + clientElement.xpath("//Client[@name='%s']"%(fullnames[chost]))[0].set("pingable",'N') + #also set pingtime if you can get it + + dump(clientElement.getroot()) fout = open(clientdatapath, 'w') - fout.write(pretty_print(clientElement)) + fout.write(tostring(clientElement.getroot())) fout.close() |