- fix UP/DOWN summary on nodes page.
[monitor.git] / nodequery.py
index 806fab0..85ff153 100755 (executable)
@@ -1,37 +1,26 @@
 #!/usr/bin/python
 
-import plc
-api = plc.getAuthAPI()
 
 import sys
-import database
-from nodecommon import *
-#from policy import Diagnose
-from unified_model import Record
+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 reboot
-from monitor import util
 import traceback
 
 import time
 import re
+import string
 
-import config
-
-from monitor.database import FindbadNodeRecord, FindbadNodeRecordSync
-#from sqlobject import connectionForURI,sqlhub
-#connection = connectionForURI(config.sqlobjecturi)
-#sqlhub.processConnection = connection
-#from infovacuum.model.findbadrecord import *
+from monitor.wrapper import plc
+api = plc.getAuthAPI()
 
-#fb = {}
-fb = None
-fbpcu = None
-import string
+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
@@ -47,6 +36,8 @@ def fb_print_nodeinfo(fbnode, hostname, fields=None):
                fbnode['bootcd_version'] = fbnode['bootcd_version'].split()[-1]
        else:
                fbnode['bootcd_version'] = "unknown"
+        if not fbnode['boot_server']:
+                fbnode['boot_server'] = "unknown"
        fbnode['pcu'] = color_pcu_state(fbnode)
 
        if not fields:
@@ -69,283 +60,19 @@ def fb_print_nodeinfo(fbnode, hostname, fields=None):
                        #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 | %(kernel_version)s" % fbnode
+               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 verifyDBrecord(constraints, record):
-       """
-               constraints is a list of key, value pairs.
-               # [ {... : ...}==AND , ... , ... , ] == OR
-       """
-       def has_key(obj, key):
-               try:
-                       x = obj.__getattribute__(key)
-                       return True
-               except:
-                       return False
-
-       def get_val(obj, key):
-               try:
-                       return obj.__getattribute__(key)
-               except:
-                       return None
-
-       def get(obj, path):
-               indexes = path.split("/")
-               value = get_val(obj,indexes[0])
-               if value is not None and len(indexes) > 1:
-                       for key in indexes[1:]:
-                               if key in value:
-                                       value = value[key]
-                               else:
-                                       raise NoKeyException(key)
-               return value
-
-       #print constraints, record
-
-       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 has_key(record, key):
-                               value_re = re.compile(con[key])
-                               if type([]) == type(get(record,key)):
-                                       local_or_true = False
-                                       for val in get(record,key):
-                                               local_or_true = local_or_true | (value_re.search(val) is not None)
-                                       con_and_true = con_and_true & local_or_true
-                               else:
-                                       if get(record,key) is not None:
-                                               con_and_true = con_and_true & (value_re.search(get(record,key)) is not None)
-                       else:
-                               print "missing key %s" % key,
-                               pass
-
-               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])
-                               if type([]) == type(data[key]):
-                                       local_or_true = False
-                                       for val in data[key]:
-                                               local_or_true = local_or_true | (value_re.search(val) is not None)
-                                       con_and_true = con_and_true & local_or_true
-                               else:
-                                       if data[key] is not None:
-                                               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
-
-               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):
-       global fb
-       global fbpcu
-       pcunames = []
-       nodenames = []
-       if str_query is None: return (nodenames, pcunames)
-
-       if fb is None:
-               fb = database.dbLoad("findbad")
-       if fbpcu is None:
-               fbpcu = database.dbLoad("findbadpcus")
-
-       #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)
-                               pcunames.append(pcuinfo['pcu_id'])
-       return (nodenames, pcunames)
-
-def node_select(str_query, nodelist=None, fbdb=None):
-       global fb
-
-       hostnames = []
-       if str_query is None: return hostnames
-
-       #print str_query
-       dict_query = query_to_dict(str_query)
-       #print dict_query
-
-       if fbdb is not None:
-               fb = fbdb
-
-       for node in fb['nodes'].keys():
-               if nodelist is not None: 
-                       if node not in nodelist: continue
-
-               try:
-                       fb_noderec = FindbadNodeRecord.select(FindbadNodeRecord.q.hostname==node, 
-                                                                                          orderBy='date_checked').reversed()[0]
-               except:
-                       continue
-
-               
-               fb_nodeinfo = fb_noderec.toDict()
-
-               #fb_nodeinfo['pcu'] = color_pcu_state(fb_nodeinfo)
-               #if 'plcnode' in fb_nodeinfo:
-               #       fb_nodeinfo.update(fb_nodeinfo['plcnode'])
-
-               #if verifyDBrecord(dict_query, fb_nodeinfo):
-               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...")
@@ -359,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.")
 
@@ -377,16 +106,16 @@ def main():
                os.chdir("..")
                fb = archive.load(file[:-4])
        else:
-               fbnodes = FindbadNodeRecord.select(FindbadNodeRecord.q.hostname, orderBy='date_checked',distinct=True).reversed()
-               fb = database.dbLoad("findbad")
-
-       fbpcu = database.dbLoad("findbadpcus")
-       reboot.fb = fbpcu
+               #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:
@@ -395,25 +124,35 @@ def main():
        elif config.select is not None:
                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']:
-                       continue
-
                try:
                        # Find the most recent record
-                       fb_noderec = FindbadNodeRecord.select(FindbadNodeRecord.q.hostname==node, 
-                                                                                          orderBy='date_checked').reversed()[0]
+                       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()
-                       pass #fb_nodeinfo  = fb['nodes'][node]['values']
+                       continue
+
+               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
@@ -421,7 +160,6 @@ def main():
                        if config.daysdown:
                                daysdown_print_nodeinfo(fb_nodeinfo, node)
                        else:
-                               fb_nodeinfo = fb_noderec.toDict()
                                if config.select:
                                        if config.fields:
                                                fields = config.fields.split(",")
@@ -429,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__":