summaryrefslogtreecommitdiffstats
path: root/src/sbin
diff options
context:
space:
mode:
Diffstat (limited to 'src/sbin')
-rw-r--r--src/sbin/StatReports152
1 files changed, 30 insertions, 122 deletions
diff --git a/src/sbin/StatReports b/src/sbin/StatReports
index 8539eb4de..283063b04 100644
--- a/src/sbin/StatReports
+++ b/src/sbin/StatReports
@@ -8,75 +8,56 @@ for bcfg2'''
__revision__ = '$Revision$'
from ConfigParser import ConfigParser, NoSectionError, NoOptionError
-from lxml.etree import XML, Element, SubElement, tostring
+from lxml.etree import XML, XSLT, parse, Element, ElementTree, SubElement, tostring
from xml.parsers.expat import ExpatError
from time import asctime, strptime, time
from socket import getfqdn
from sys import argv
from getopt import getopt, GetoptError
-import re, os, libxml2, libxslt
-from tempfile import mktemp
+import re, os
from copy import deepcopy
-def generatereport(rs, nr):
+def generatereport(rspec, nrpt):
'''generatereport creates and returns an ElementTree representation
of a report adhering to the XML spec for intermediate reports'''
- reportspec = deepcopy(rs)
- nodereprt = deepcopy(nr)
+ reportspec = deepcopy(rspec)
+ nodereprt = deepcopy(nrpt)
reportgood = reportspec.get("good", default = 'Y')
reportmodified = reportspec.get("modified", default = 'Y')
-
current_date = asctime()[:10]
'''build regex of all the nodes we are reporting about'''
- regex = '|'.join([x.get("name") for x in \
- reportspec.findall('Machine')])
- pattern = re.compile(regex)
-
+ pattern = re.compile( '|'.join([item.get("name") for item in reportspec.findall('Machine')]))
for node in nodereprt.findall('Node'):
-
- if node.findall('HostInfo') == [] or \
- not pattern.match(node.get("name")) or \
- node.findall('Statistics') == [] or \
- node.find("HostInfo").get("fqdn") == "":#maybe issue a warning instead?
+ if not (node.findall("HostInfo") and node.findall("Statistics") and
+ node.find("HostInfo").get("fqdn") and pattern.match(node.get('name'))):
+ # don't know enough about node
nodereprt.remove(node)
continue
#reduce to most recent Statistics entry
statisticslist = node.findall('Statistics')
#this line actually sorts from most recent to oldest
- statisticslist.sort(lambda y, x: cmp(strptime(x.get("time")), \
- strptime(y.get("time"))))
+ stats = statisticslist.sort(lambda y, x: cmp(strptime(x.get("time")), strptime(y.get("time"))))[0]
- stats = statisticslist[0]
-
- [node.remove(x) for x in node.findall('Statistics')]
-
-
+ [node.remove(item) for item in node.findall('Statistics')]
+
#add a good tag if node is good and we wnat to report such
if reportgood == 'Y' and stats.get('state') == 'clean':
SubElement(stats,"Good")
- for x in stats.findall('Modified'):
- if reportmodified == 'N' or x.getchildren() == None:
- stats.remove(x)
-
- for x in stats.findall('Bad'):
- if x.getchildren() == None:
- stats.remove(x)
+ [stats.remove(item) for item in stats.findall("Bad") + stats.findall("Modified") if \
+ item.getchildren() == None]
+ [stats.remove(item) for item in stats.findall("Modified") if reportmodified == 'N']
#test for staleness -if stale add Stale tag
if stats.get("time").find(current_date) == -1:
SubElement(stats,"Stale")
-
node.append(stats)
-
return nodereprt
-
-
def mail(mailbody, confi):
'''mail mails a previously generated report'''
@@ -273,109 +254,36 @@ if __name__ == '__main__':
#IMPORTANT to add some error checking here-parseerrors
#this might be sufficient
try:
- styledoc = libxml2.parseFile(transformpath+transform)
- style = libxslt.parseStylesheetDoc(styledoc)
+ stylesheet = XSLT(parse(transformpath + transform))
+ #styledoc = libxml2.parseFile(transformpath+transform)
+ #style = libxslt.parseStylesheetDoc(styledoc)
except:
print("StatReports: invalid XSLT transform file.")
raise SystemExit, 1
if deliverymechanism == 'mail':
if delivtype == 'nodes-individual':
- p2noderep = deepcopy(procnodereport)
+ reportdata = deepcopy(procnodereport)
for noden in procnodereport.findall("Node"):
- [p2noderep.remove(y) for y in p2noderep.findall("Node")]
- p2noderep.append(noden)
- tempfilename = mktemp()
- tempr = open(tempfilename, "w+")
- tempr.write(tostring(p2noderep))
- tempr.seek(0)
- doc = libxml2.parseFile(tempfilename)
- result = style.applyStylesheet(doc, None)
- tempr.close()
- os.unlink(tempfilename)
- del tempr
- try:
- tempfilename = mktemp()
- tempr = open(tempfilename, "w+")
- style.saveResultToFile(tempr, result)
- tempr.seek(0)
- outputstring = tempr.read()
- except:
- outputstring = None#this is a nasty hack. When the xslt transform breaks-- just blank it out
- #this is done due to a bug in libxslt
- #This needs to be fixed in future releases
-
- tempr.close()
- del tempr
- os.unlink(tempfilename)
-
- if not outputstring == None:
- toastring = ''
- for desti in deliv.findall("Destination"):
- toastring = "%s%s " % \
- (toastring, desti.get('address'))
- #prepend To: and From:
- outputstring = "To: %s\nFrom: root@%s\n%s"% \
- (toastring, getfqdn(), outputstring)
- mail(outputstring, c) #call function to send
- doc.freeDoc()
- result.freeDoc()
- style.freeStylesheet()
+ [reportdata.remove(y) for y in reportdata.findall("Node")]
+ reportdata.append(noden)
else:
- tempfilename = mktemp()
- tempr = open(tempfilename, "w+")
- tempr.write(tostring(procnodereport))
- tempr.seek(0)
- doc = libxml2.parseFile(tempfilename)
- result = style.applyStylesheet(doc, None)
- tempr.close()
- del tempr
- os.unlink(tempfilename)
- tempfilename = mktemp()
- tempr = open(tempfilename, "w+")
- style.saveResultToFile(tempr, result)
- tempr.seek(0)
- outputstring = tempr.read()
- tempr.close()
- del tempr
- os.unlink(tempfilename)
-
- if not outputstring == None:
- toastring = ''
- for desti in deliv.findall("Destination"):
- toastring = "%s%s " % \
- (toastring, desti.get('address'))
+ reportdata = procnodereport
+ outputstring = tostring(stylesheet.apply(reportdata).getroot())
+
+ if not outputstring == None:
+ toastring = ''
+ for desti in deliv.findall("Destination"):
+ toastring = "%s%s " % \
+ (toastring, desti.get('address'))
#prepend To: and From:
outputstring = "To: %s\nFrom: root@%s\n%s"% \
(toastring, getfqdn(), outputstring)
mail(outputstring, c) #call function to send
- style.freeStylesheet()
- doc.freeDoc()
- result.freeDoc()
else:
- tempfilename = mktemp()
- tempr = open(tempfilename, "w+")
- tempr.write(tostring(procnodereport))
- tempr.seek(0)
- doc = libxml2.parseFile(tempfilename)
- result = style.applyStylesheet(doc, None)
- tempr.close()
- del tempr
- os.unlink(tempfilename)
- tempfilename = mktemp()
- tempr = open(tempfilename, "w+")
- style.saveResultToFile(tempr, result)
- tempr.seek(0)
- outputstring = tempr.read()
- tempr.close()
- del tempr
- os.unlink(tempfilename)
-
+ outputstring = tostring(stylesheet.apply(ElementTree(procnodereport)).getroot())
if deliverymechanism == 'rss':
rss(outputstring, deliv, reprt)
else: # must be deliverymechanism == 'www':
www(outputstring, deliv)
- style.freeStylesheet()
- doc.freeDoc()
- result.freeDoc()