X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=pcubad.py;h=e13accc2f5a0321398f3494d9881492983891307;hb=da913fbd1629fc4669b186915df8ff3a340482d3;hp=c2886f519b7e5b1a0137c1b50ad13a0480d50f22;hpb=c3f2afdc81c6711c3825c82e2cd4970671575438;p=monitor.git diff --git a/pcubad.py b/pcubad.py index c2886f5..e13accc 100755 --- a/pcubad.py +++ b/pcubad.py @@ -4,41 +4,34 @@ import os import sys import string import time +from datetime import datetime,timedelta -from reboot import pcu_name +from monitor import database +from monitor.pcu import reboot +from monitor import parser as parsermodule +from monitor import config +from monitor.database import HistoryPCURecord, FindbadPCURecord +from monitor.wrapper import plc,plccache +from monitor.const import MINUP -import database -import comon -import threadpool -import syncplcdb -from nodequery import verify,query_to_dict,node_select -import parser as parsermodule from nodecommon import * +from nodequery import verify,query_to_dict,node_select +from monitor.model import * -import plc api = plc.getAuthAPI() -from unified_model import * -from monitor_policy import MINUP - -round = 1 -externalState = {'round': round, 'nodes': {}} -count = 0 def main(config): - global externalState - externalState = database.if_cached_else(1, config.dbname, lambda : externalState) - if config.increment: - # update global round number to force refreshes across all pcus - externalState['round'] += 1 - l_plcpcus = database.if_cached_else_refresh(1, 1, "pculist", lambda : plc.GetPCUs()) + #l_plcpcus = database.if_cached_else_refresh(1, 1, "pculist", lambda : plc.GetPCUs()) + l_plcpcus = plccache.l_pcus - l_pcu = None + l_pcus = None if config.pcu: for pcu in l_plcpcus: - if pcu['hostname'] == config.pcu or pcu['ip'] == config.pcu: + if ( pcu['hostname'] is not None and config.pcu in pcu['hostname'] ) or \ + ( pcu['ip'] is not None and config.pcu in pcu['ip'] ): l_pcus = [pcu['pcu_id']] - if not l_pcu: + if not l_pcus: print "ERROR: could not find pcu %s" % config.pcu sys.exit(1) else: @@ -46,108 +39,72 @@ def main(config): checkAndRecordState(l_pcus, l_plcpcus) -def checkAndRecordState(l_pcus, l_plcpcus): - global externalState - global count - global_round = externalState['round'] +hn2lb = plccache.plcdb_hn2lb +def checkAndRecordState(l_pcus, l_plcpcus): + count = 0 for pcuname in l_pcus: - if pcuname not in externalState['nodes']: - externalState['nodes'][pcuname] = {'round': 0, 'values': []} - - pcu_round = externalState['nodes'][pcuname]['round'] - if pcu_round < global_round: - # do work - values = collectStatusAndState(pcuname, l_plcpcus) - global_round = externalState['round'] - externalState['nodes'][pcuname]['values'] = values - externalState['nodes'][pcuname]['round'] = global_round - else: - count += 1 - if count % 20 == 0: - database.dbDump(config.dbname, externalState) - - database.dbDump(config.dbname, externalState) - -fbpcu = database.dbLoad('findbadpcus') -hn2lb = database.dbLoad("plcdb_hn2lb") - -def get(fb, path): - indexes = path.split("/") - values = fb - for index in indexes: - if index in values: - values = values[index] + d_pcu = None + for pcu in l_plcpcus: + if pcu['pcu_id'] == pcuname: + d_pcu = pcu + break + if not d_pcu: + continue + + pf = HistoryPCURecord.findby_or_create(plc_pcuid=d_pcu['pcu_id']) + pf.last_checked = datetime.now() + + try: + # Find the most recent record + pcurec = FindbadPCURecord.query.filter(FindbadPCURecord.plc_pcuid==pcuname).order_by(FindbadPCURecord.date_checked.desc()).first() + print "NODEREC: ", pcurec.date_checked + except: + print "COULD NOT FIND FB record for %s" % reboot.pcu_name(pcu) + import traceback + print traceback.print_exc() + # don't have the info to create a new entry right now, so continue. + continue + + pcu_state = pcurec.reboot_trial_status + current_state = pcu_state + + if current_state == 0 or current_state == "0": + if pf.status != "good": + pf.last_changed = datetime.now() + pf.status = "good" + elif current_state == 'NetDown': + if pf.status != "netdown": + pf.last_changed = datetime.now() + pf.status = "netdown" + elif current_state == 'Not_Run': + if pf.status != "badconfig": + pf.last_changed = datetime.now() + pf.status = "badconfig" else: - return None - return values - -def collectStatusAndState(pcuname, l_plcpcus): - global count - - d_pcu = None - for pcu in l_plcpcus: - if pcu['pcu_id'] == pcuname: - d_pcu = pcu - break - if not d_pcu: - return None - - pf = PersistFlags(pcuname, 1, db='pcu_persistflags') - - if not pf.checkattr('last_changed'): - pf.last_changed = time.time() - - pf.last_checked = time.time() - - if not pf.checkattr('valid'): - pf.valid = "unknown" - pf.last_valid = 0 - - if not pf.checkattr('status'): - pf.status = "unknown" - - state_path = "nodes/id_" + str(pcuname) + "/values/reboot" - bootstate_path = "nodes/id_" + str(pcuname) + "/values/plcpcu/boot_state" - - current_state = get(fbpcu, state_path) - if current_state == 0: - if pf.status != "good": pf.last_changed = time.time() - pf.status = "good" - elif current_state == 'NetDown': - if pf.status != "netdown": pf.last_changed = time.time() - pf.status = "netdown" - elif current_state == 'Not_Run': - if pf.status != "badconfig": pf.last_changed = time.time() - pf.status = "badconfig" - else: - if pf.status != "error": pf.last_changed = time.time() - pf.status = "error" + if pf.status != "error": + pf.last_changed = datetime.now() + pf.status = "error" - count += 1 - print "%d %35s %s since(%s)" % (count, pcu_name(d_pcu), pf.status, diff_time(pf.last_changed)) - # updated by other modules - #pf.enabled = - #pf.suspended = + count += 1 + print "%d %35s %s since(%s)" % (count, reboot.pcu_name(d_pcu), pf.status, diff_time(time.mktime(pf.last_changed.timetuple()))) - pf.save() + # NOTE: this commits all pending operations to the DB. Do not remove, or + # replace with another operations that also commits all pending ops, such + # as session.commit() or flush() or something + print HistoryPCURecord.query.count() return True if __name__ == '__main__': parser = parsermodule.getParser() - parser.set_defaults(filename=None, pcu=None, pcuselect=False, pcugroup=None, - increment=False, dbname="pcubad", cachepcus=False) + parser.set_defaults(filename=None, pcu=None, pcuselect=False, pcugroup=None, cachepcus=False) parser.add_option("", "--pcu", dest="pcu", metavar="hostname", help="Provide a single pcu to operate on") parser.add_option("", "--pculist", dest="pculist", metavar="file.list", help="Provide a list of files to operate on") - parser.add_option("", "--dbname", dest="dbname", metavar="FILE", - help="Specify the name of the database to which the information is saved") - parser.add_option("-i", "--increment", action="store_true", dest="increment", - help="Increment round number to force refresh or retry") config = parsermodule.parse_args(parser) try: @@ -156,6 +113,4 @@ if __name__ == '__main__': import traceback print traceback.print_exc() print "Exception: %s" % err - print "Saving data... exitting." - database.dbDump(config.dbname, externalState) sys.exit(0)