+++ /dev/null
-#!/usr/bin/python
-
-import os
-import sys
-import string
-import time
-import sets
-from datetime import datetime,timedelta
-
-from monitor import database
-from monitor import reboot
-from monitor import parser as parsermodule
-from monitor import config
-from monitor.database.info.model import HistoryPCURecord, FindbadPCURecord
-from monitor.database.dborm import mon_session as session
-from monitor.wrapper import plc,plccache
-from monitor.const import MINUP
-
-from monitor.common import *
-from monitor.query import verify,query_to_dict,node_select
-from monitor.model import *
-
-api = plc.getAuthAPI()
-
-def main():
- main2(config)
-
-def main2(config):
-
- l_plcpcus = plccache.l_pcus
-
- l_pcus = None
- if config.site is not None:
- site = plccache.GetSitesByName([config.site])
- l_nodes = plccache.GetNodesByIds(site[0]['node_ids'])
- pcus = []
- for node in l_nodes:
- pcus += node['pcu_ids']
- # clear out dups.
- l_pcus = [pcu for pcu in sets.Set(pcus)]
-
- elif config.node:
- node = plccache.GetNodeByName(config.node)
- pcus = node['pcu_ids']
- # clear out dups.
- l_pcus = [pcu for pcu in sets.Set(pcus)]
-
- elif config.pcu:
- for pcu in l_plcpcus:
- 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_pcus:
- print "ERROR: could not find pcu %s" % config.pcu
- sys.exit(1)
- else:
- l_pcus = [pcu['pcu_id'] for pcu in l_plcpcus]
-
- checkAndRecordState(l_pcus, l_plcpcus)
-
-hn2lb = plccache.plcdb_hn2lb
-
-def check_pcu_state(rec, pcu):
-
- pcu_state = rec.reboot_trial_status
-
- # DOWN
- if pcu_state not in [0, "0"] and pcu.status not in ['offline', 'down']:
- print "changed status from %s to offline" % pcu.status
- pcu.status = 'offline'
- pcu.last_changed = datetime.now()
-
- # ONLINE
- if pcu_state in [0, "0"] and pcu.status not in [ 'online', 'good' ]:
- print "changed status from %s to online" % pcu.status
- pcu.status = 'online'
- pcu.last_changed = datetime.now()
-
-
- # STATE TRANSITIONS
- if pcu.status == 'online' and changed_greaterthan(pcu.last_changed, 0.5):
- #send thank you notice, or on-line notice.
- print "changed status from %s to good" % pcu.status
- pcu.status = 'good'
- # NOTE: do not reset last_changed, or you lose how long it's been up.
-
- if pcu.status == 'offline' and changed_greaterthan(pcu.last_changed, 2):
- # send down pcu notice
- print "changed status from %s to down" % pcu.status
- pcu.status = 'down'
-
-# if pcu.status in [ 'offline', 'down' ] and changed_greaterthan(pcu.last_changed, 2*30):
-# print "changed status from %s to down" % pcu.status
-# pcu.status = 'down'
-# pcu.last_changed = datetime.now()
-
-def checkAndRecordState(l_pcus, l_plcpcus):
- count = 0
- for pcuname in l_pcus:
-
- d_pcu = None
- for pcu in l_plcpcus:
- if pcu['pcu_id'] == pcuname:
- d_pcu = pcu
- break
- if not d_pcu:
- continue
-
- pcuhist = HistoryPCURecord.findby_or_create(plc_pcuid=d_pcu['pcu_id'],
- if_new_set={'status' : 'offline',
- 'last_changed' : datetime.now()})
- pcuhist.last_checked = datetime.now()
-
- try:
- # Find the most recent record
- pcurec = FindbadPCURecord.query.filter(FindbadPCURecord.plc_pcuid==pcuname).first()
- except:
- print "COULD NOT FIND FB record for %s" % reboot.pcu_name(d_pcu)
- import traceback
- email_exception()
- print traceback.print_exc()
- # don't have the info to create a new entry right now, so continue.
- continue
-
- if not pcurec:
- print "none object for pcu %s"% reboot.pcu_name(d_pcu)
- continue
-
- check_pcu_state(pcurec, pcuhist)
-
- count += 1
- print "%d %35s %s since(%s)" % (count, reboot.pcu_name(d_pcu), pcuhist.status, diff_time(time.mktime(pcuhist.last_changed.timetuple())))
-
- # 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
- session.flush()
- print HistoryPCURecord.query.count()
-
- return True
-
-if __name__ == '__main__':
- parser = parsermodule.getParser()
- parser.set_defaults(filename=None, pcu=None, node=None, site=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("", "--site", dest="site", metavar="sitename",
- help="Provide a single sitename to operate on")
- parser.add_option("", "--node", dest="node", metavar="nodename",
- help="Provide a single node to operate on")
- parser.add_option("", "--pculist", dest="pculist", metavar="file.list",
- help="Provide a list of files to operate on")
-
- config = parsermodule.parse_args(parser)
-
- try:
- main2(config)
- except Exception, err:
- import traceback
- traceback.print_exc()
- print "Exception: %s" % err
- sys.exit(0)