- fix UP/DOWN summary on nodes page.
[monitor.git] / nodequery.py
index 87aad9f..85ff153 100755 (executable)
 #!/usr/bin/python
 
-import plc
-api = plc.getAuthAPI()
 
 import sys
-import database
-from nodecommon import *
-from policy import Diagnose
+from monitor import database
+from monitor.common import *
+from monitor.query import *
+from monitor.model import Record
 import glob
 import os
-from reboot import pcu_name
-import util.file
+import traceback
 
 import time
 import re
+import string
+
+from monitor.wrapper import plc
+api = plc.getAuthAPI()
 
-#fb = {}
-fb = database.dbLoad("findbad")
-fbpcu = {}
+from monitor.database.info.model import HistoryNodeRecord, FindbadNodeRecord, FindbadPCURecord, session
+from monitor.util import file as utilfile
+from monitor import config
 
-class NoKeyException(Exception): pass
 
 def daysdown_print_nodeinfo(fbnode, hostname):
        fbnode['hostname'] = hostname
-       fbnode['daysdown'] = Diagnose.getStrDaysDown(fbnode)
-       fbnode['intdaysdown'] = Diagnose.getDaysDown(fbnode)
+       fbnode['daysdown'] = Record.getStrDaysDown(fbnode)
+       fbnode['intdaysdown'] = Record.getDaysDown(fbnode)
 
        print "%(intdaysdown)5s %(hostname)-44s | %(state)10.10s | %(daysdown)s" % fbnode
 
 def fb_print_nodeinfo(fbnode, hostname, fields=None):
-       fbnode['hostname'] = hostname
-       fbnode['checked'] = diff_time(fbnode['checked'])
-       if fbnode['bootcd']:
-               fbnode['bootcd'] = fbnode['bootcd'].split()[-1]
+       #fbnode['hostname'] = hostname
+       #fbnode['checked'] = diff_time(fbnode['checked'])
+       if fbnode['bootcd_version']:
+               fbnode['bootcd_version'] = fbnode['bootcd_version'].split()[-1]
        else:
-               fbnode['bootcd'] = "unknown"
+               fbnode['bootcd_version'] = "unknown"
+        if not fbnode['boot_server']:
+                fbnode['boot_server'] = "unknown"
        fbnode['pcu'] = color_pcu_state(fbnode)
 
        if not fields:
-               if 'ERROR' in fbnode['category']:
-                       fbnode['kernel'] = ""
+               if ( fbnode['observed_status'] is not None and \
+                  'DOWN' in fbnode['observed_status'] ) or \
+                  fbnode['kernel_version'] is None:
+                       fbnode['kernel_version'] = ""
                else:
-                       fbnode['kernel'] = fbnode['kernel'].split()[2]
-               fbnode['boot_state'] = fbnode['plcnode']['boot_state']
+                       fbnode['kernel_version'] = fbnode['kernel_version'].split()[2]
 
-               print "%(hostname)-39s | %(checked)11.11s | %(boot_state)5.5s| %(state)8.8s | %(ssh)5.5s | %(pcu)6.6s | %(bootcd)6.6s | %(category)8.8s | %(kernel)s" % fbnode
+               if fbnode['plc_node_stats'] is not None:
+                       fbnode['boot_state'] = fbnode['plc_node_stats']['boot_state']
+               else:
+                       fbnode['boot_state'] = "unknown"
+
+               try:
+                       if len(fbnode['nodegroups']) > 0:
+                               fbnode['category'] = fbnode['nodegroups'][0]
+               except:
+                       #print "ERROR!!!!!!!!!!!!!!!!!!!!!"
+                       pass
+
+               print "%(hostname)-45s | %(date_checked)11.11s | %(boot_state)5.5s| %(observed_status)8.8s | %(ssh_status)5.5s | %(pcu)6.6s | %(bootcd_version)6.6s | %(boot_server)s | %(kernel_version)s" % fbnode
        else:
                format = ""
                for f in fields:
                        format += "%%(%s)s " % f
                print format % fbnode
 
-def get(fb, path):
-    indexes = path.split("/")
-    values = fb
-    for index in indexes:
-        if index in values:
-            values = values[index]
-        else:
-            raise NoKeyException(index)
-    return values
-
-def verifyType(constraints, data):
-       """
-               constraints is a list of key, value pairs.
-               # [ {... : ...}==AND , ... , ... , ] == OR
-       """
-       con_or_true = False
-       for con in constraints:
-               #print "con: %s" % con
-               if len(con.keys()) == 0:
-                       con_and_true = False
-               else:
-                       con_and_true = True
-
-               for key in con.keys():
-                       #print "looking at key: %s" % key
-                       if data is None:
-                               con_and_true = False
-                               break
-
-                       try:
-                               get(data,key)
-                               o = con[key]
-                               if o.name() == "Match":
-                                       if get(data,key) is not None:
-                                               value_re = re.compile(o.value)
-                                               con_and_true = con_and_true & (value_re.search(get(data,key)) is not None)
-                                       else:
-                                               con_and_true = False
-                               elif o.name() == "ListMatch":
-                                       if get(data,key) is not None:
-                                               match = False
-                                               for listitem in get(data,key):
-                                                       value_re = re.compile(o.value)
-                                                       if value_re.search(listitem) is not None:
-                                                               match = True
-                                                               break
-                                               con_and_true = con_and_true & match
-                                       else:
-                                               con_and_true = False
-                               elif o.name() == "Is":
-                                       con_and_true = con_and_true & (get(data,key) == o.value)
-                               elif o.name() == "FilledIn":
-                                       con_and_true = con_and_true & (len(get(data,key)) > 0)
-                               elif o.name() == "PortOpen":
-                                       if get(data,key) is not None:
-                                               v = get(data,key)
-                                               con_and_true = con_and_true & (v[str(o.value)] == "open")
-                                       else:
-                                               con_and_true = False
-                               else:
-                                       value_re = re.compile(o.value)
-                                       con_and_true = con_and_true & (value_re.search(get(data,key)) is not None)
-
-                       except NoKeyException, key:
-                               print "missing key %s" % key,
-                               pass
-                               #print "missing key %s" % key
-                               #con_and_true = False
-
-               con_or_true = con_or_true | con_and_true
-
-       return con_or_true
-
-def verify(constraints, data):
-       """
-               constraints is a list of key, value pairs.
-               # [ {... : ...}==AND , ... , ... , ] == OR
-       """
-       con_or_true = False
-       for con in constraints:
-               #print "con: %s" % con
-               if len(con.keys()) == 0:
-                       con_and_true = False
-               else:
-                       con_and_true = True
-
-               for key in con.keys():
-                       #print "looking at key: %s" % key
-                       if key in data: 
-                               value_re = re.compile(con[key])
-                               con_and_true = con_and_true & (value_re.search(data[key]) is not None)
-                       elif key not in data:
-                               print "missing key %s" % key,
-                               pass
-                               #print "missing key %s" % key
-                               #con_and_true = False
-
-               con_or_true = con_or_true | con_and_true
-
-       return con_or_true
-
-def query_to_dict(query):
-       
-       ad = []
-
-       or_queries = query.split('||')
-       for or_query in or_queries:
-               and_queries = or_query.split('&&')
-
-               d = {}
-
-               for and_query in and_queries:
-                       (key, value) = and_query.split('=')
-                       d[key] = value
-
-               ad.append(d)
-       
-       return ad
-
-def pcu_in(fbdata):
-       if 'plcnode' in fbdata:
-               if 'pcu_ids' in fbdata['plcnode']:
-                       if len(fbdata['plcnode']['pcu_ids']) > 0:
-                               return True
-       return False
-
-def pcu_select(str_query, nodelist=None):
-       pcunames = []
-       nodenames = []
-       if str_query is None: return (nodenames, pcunames)
-
-       #print str_query
-       dict_query = query_to_dict(str_query)
-       #print dict_query
-
-       for node in fb['nodes'].keys():
-               if nodelist is not None: 
-                       if node not in nodelist: continue
-       
-               fb_nodeinfo  = fb['nodes'][node]['values']
-               if pcu_in(fb_nodeinfo):
-                       pcuinfo = fbpcu['nodes']['id_%s' % fb_nodeinfo['plcnode']['pcu_ids'][0]]['values']
-                       if verify(dict_query, pcuinfo):
-                               nodenames.append(node)
-                               str = "cmdhttps/locfg.pl -s %s -f iloxml/License.xml -u %s -p '%s' | grep MESSAGE" % \
-                                                       (pcu_name(pcuinfo), pcuinfo['username'], pcuinfo['password'])
-                               pcunames.append(str)
-       return (nodenames, pcunames)
-
-def node_select(str_query, nodelist=None, fbdb=None):
-       hostnames = []
-       if str_query is None: return hostnames
-
-       #print str_query
-       dict_query = query_to_dict(str_query)
-       #print dict_query
-       global fb
-
-       if fbdb is not None:
-               fb = fbdb
-
-       for node in fb['nodes'].keys():
-               if nodelist is not None: 
-                       if node not in nodelist: continue
-       
-               fb_nodeinfo  = fb['nodes'][node]['values']
-
-               if fb_nodeinfo == []:
-                       #print node, "has lost values"
-                       continue
-                       #sys.exit(1)
-               fb_nodeinfo['pcu'] = color_pcu_state(fb_nodeinfo)
-               fb_nodeinfo['hostname'] = node
-               if 'plcnode' in fb_nodeinfo:
-                       fb_nodeinfo.update(fb_nodeinfo['plcnode'])
-
-               if verify(dict_query, fb_nodeinfo):
-                       #print node #fb_nodeinfo
-                       hostnames.append(node)
-               else:
-                       #print "NO MATCH", node
-                       pass
-       
-       return hostnames
-
-
 def main():
-       global fb
-       global fbpcu
 
-       import parser as parsermodule
+       from monitor import parser as parsermodule
        parser = parsermodule.getParser()
 
-       parser.set_defaults(node=None, fromtime=None, select=None, list=None, 
+       parser.set_defaults(node=None, fromtime=None, select=None, list=None, listkeys=False,
                                                pcuselect=None, nodelist=None, daysdown=None, fields=None)
        parser.add_option("", "--daysdown", dest="daysdown", action="store_true",
                                                help="List the node state and days down...")
@@ -258,6 +86,8 @@ def main():
                                                help="List all nodes with the given key=value pattern")
        parser.add_option("", "--nodelist", dest="nodelist", metavar="nodelist.txt", 
                                                help="A list of nodes to bring out of debug mode.")
+       parser.add_option("", "--listkeys", dest="listkeys", action="store_true",
+                                               help="A list of nodes to bring out of debug mode.")
        parser.add_option("", "--fromtime", dest="fromtime", metavar="YYYY-MM-DD",
                                        help="Specify a starting date from which to begin the query.")
 
@@ -276,35 +106,53 @@ def main():
                os.chdir("..")
                fb = archive.load(file[:-4])
        else:
-               fb = database.dbLoad("findbad")
-
-       fbpcu = database.dbLoad("findbadpcus")
+               #fbnodes = FindbadNodeRecord.select(FindbadNodeRecord.q.hostname, orderBy='date_checked',distinct=True).reversed()
+               fb = None
 
        if config.nodelist:
-               nodelist = util.file.getListFromFile(config.nodelist)
+               nodelist = utilfile.getListFromFile(config.nodelist)
        else:
-               nodelist = fb['nodes'].keys()
+               # NOTE: list of nodes should come from findbad db.   Otherwise, we
+               # don't know for sure that there's a record in the db..
+               fbquery = HistoryNodeRecord.query.all()
+               nodelist = [ n.hostname for n in fbquery ]
 
        pculist = None
        if config.select is not None and config.pcuselect is not None:
-               nodelist = node_select(config.select, nodelist)
+               nodelist = node_select(config.select, nodelist, fb)
                nodelist, pculist = pcu_select(config.pcuselect, nodelist)
        elif config.select is not None:
-               nodelist = node_select(config.select, nodelist)
+               nodelist = node_select(config.select, nodelist, fb)
        elif config.pcuselect is not None:
+               print "thirhd node select"
                nodelist, pculist = pcu_select(config.pcuselect, nodelist)
 
-       if pculist:
-               for pcu in pculist:
-                       print pcu
+       #if pculist:
+       #       for pcu in pculist:
+       #               print pcu
 
+       print "len: %s" % len(nodelist)
        for node in nodelist:
                config.node = node
 
-               if node not in fb['nodes']:
+               try:
+                       # Find the most recent record
+                       fb_noderec = FindbadNodeRecord.get_latest_by(hostname=node) 
+                       if not fb_noderec: continue
+                       fb_nodeinfo = fb_noderec.to_dict()
+               except KeyboardInterrupt:
+                       print "Exiting at user request: Ctrl-C"
+                       sys.exit(1)
+               except:
+                       print traceback.print_exc()
                        continue
 
-               fb_nodeinfo  = fb['nodes'][node]['values']
+               if config.listkeys:
+                       print "Primary keys available in the findbad object:"
+                       for key in fb_nodeinfo.keys():
+                               print "\t",key
+                       sys.exit(0)
+                       
 
                if config.list:
                        print node
@@ -319,9 +167,10 @@ def main():
                                                fields = None
 
                                        fb_print_nodeinfo(fb_nodeinfo, node, fields)
-                               elif not config.select and 'state' in fb_nodeinfo:
+                               elif not config.select and 'observed_status' in fb_nodeinfo:
                                        fb_print_nodeinfo(fb_nodeinfo, node)
                                else:
+                                       print "passing..."
                                        pass
                
 if __name__ == "__main__":