instructs user how to create the 'auth.py' file.
[monitor.git] / policy.py
index 07ec83d..3673b8e 100644 (file)
--- a/policy.py
+++ b/policy.py
@@ -17,10 +17,11 @@ import pickle
 import Queue
 import plc
 import sys
+import os
 import reboot
 import soltesz
 import string
-from printbadbysite import cmpCategoryVal
+from www.printbadnodes import cmpCategoryVal
 from config import config
 print "policy"
 config = config()
@@ -86,6 +87,16 @@ def getdebug():
 def print_stats(key, stats):
        if key in stats: print "%20s : %d" % (key, stats[key])
 
+def get_ticket_id(record):
+       if 'ticket_id' in record and record['ticket_id'] is not "" and record['ticket_id'] is not None:
+               return record['ticket_id']
+       elif            'found_rt_ticket' in record and \
+                record['found_rt_ticket'] is not "" and \
+                record['found_rt_ticket'] is not None:
+               return record['found_rt_ticket']
+       else:
+               return None
+
 class Merge(Thread):
        def __init__(self, l_merge, toRT):
                self.toRT = toRT
@@ -130,7 +141,14 @@ class Merge(Thread):
 
                        fb_record = {}
                        fb_record['nodename'] = nodename
-                       fb_record['category'] = values['category']
+                       try:
+                               fb_record['category'] = values['category']
+                       except:
+                               print values
+                               print nodename
+                               print self.findbad['nodes'][nodename]
+                               count -= 1
+                               continue
                        fb_record['state'] = values['state']
                        fb_record['comonstats'] = values['comonstats']
                        fb_record['plcnode'] = values['plcnode']
@@ -191,68 +209,44 @@ class Merge(Thread):
                                if loginbase not in self.mergedb:
                                        self.mergedb[loginbase] = {}
 
+                               # take the info either from act_all or fb-record.
+                               # if node not in act_all
+                               #       then take it from fbrecord, obviously.
+                               # else node in act_all
+                               #   if act_all == 0 length (no previous records)
+                               #               then take it from fbrecord.
+                               #   else
+                               #           take it from act_all.
+                               #   
+
                                # We must compare findbad state with act_all state
                                if nodename not in self.act_all:
                                        # 1) ok, b/c it's a new problem. set ticket_id to null
                                        self.mergedb[loginbase][nodename] = {} 
                                        self.mergedb[loginbase][nodename].update(x)
                                        self.mergedb[loginbase][nodename]['ticket_id'] = ""
-                                       self.mergedb[loginbase][nodename]['prev_category'] = None
+                                       self.mergedb[loginbase][nodename]['prev_category'] = "NORECORD" 
                                else: 
                                        if len(self.act_all[nodename]) == 0:
-                                               continue
-
-                                       y = self.act_all[nodename][0]
-
-                                       # skip if end-stage
-                                       if 'stage' in y and "monitor-end-record" in y['stage']:
-                                               continue
-
-                                       # for legacy actions
-                                       if 'bucket' in y and y['bucket'][0] == 'dbg':
-                                               # Only bootcd debugs made it to the act_all db.
-                                               y['prev_category'] = "OLDBOOTCD"
-                                       elif 'bucket' in y and y['bucket'][0] == 'down':
-                                               y['prev_category'] = "ERROR"
-                                       elif 'bucket' not in y:
-                                               # for all other actions, just carry over the
-                                               # previous category
-                                               y['prev_category'] = y['category']
+                                               self.mergedb[loginbase][nodename] = {} 
+                                               self.mergedb[loginbase][nodename].update(x)
+                                               self.mergedb[loginbase][nodename]['ticket_id'] = ""
+                                               self.mergedb[loginbase][nodename]['prev_category'] = "NORECORD" 
                                        else:
-                                               print "UNKNOWN state for record: %s" % y
-                                               sys.exit(1)
-                                       # determine through translation, if the buckets match
-                                       #if 'category' in y and x['category'] == y['category']:
-                                       #       b_match = True
-                                       #elif x['category'] == "OLDBOOTCD" and y['bucket'][0] == 'dbg':
-                                       #       b_match = True
-                                       #elif x['category'] == "ERROR" and y['bucket'][0] == 'down':
-                                       #       b_match = True
-                                       #else:
-                                       #       b_match = False
-
-                                       #if b_match: 
-                                       #       # 2b) ok, b/c they agree that there's still a problem..
-                                       #       # 2b) Comon & Monitor still agree; RT ticket?
-                                       #       y['prev_category'] = y['category']
-                                       #else:
-                                       #       # 2a) mismatch, need a policy for how to resolve
-                                       #       #     resolution will be handled in __diagnoseNode()
-                                       #       #         for now just record the two categories.
-                                       #       #if x['category'] == "PROD" and x['state'] == "BOOT" and \
-                                       #       # ( y['bucket'][0] == 'down' or  y['bucket'][0] == 'dbg'):
-                                       #       print "FINDBAD and MONITOR have a mismatch: %s vs %s" % \
-                                       #                               (x['category'], y['bucket'])
-
-
-                                       self.mergedb[loginbase][nodename] = {}
-                                       self.mergedb[loginbase][nodename].update(y)
-                                       self.mergedb[loginbase][nodename]['comonstats'] = x['comonstats']
-                                       self.mergedb[loginbase][nodename]['category']   = x['category']
-                                       self.mergedb[loginbase][nodename]['state'] = x['state']
-                                       self.mergedb[loginbase][nodename]['kernel']=x['kernel']
-                                       self.mergedb[loginbase][nodename]['bootcd']=x['bootcd']
-                                       self.mergedb[loginbase][nodename]['plcnode']=x['plcnode']
+                                               y = self.act_all[nodename][0]
+                                               y['prev_category'] = y['category']
+
+                                               self.mergedb[loginbase][nodename] = {}
+                                               self.mergedb[loginbase][nodename].update(y)
+                                               self.mergedb[loginbase][nodename]['comonstats'] = x['comonstats']
+                                               self.mergedb[loginbase][nodename]['category']   = x['category']
+                                               self.mergedb[loginbase][nodename]['state'] = x['state']
+                                               self.mergedb[loginbase][nodename]['kernel']=x['kernel']
+                                               self.mergedb[loginbase][nodename]['bootcd']=x['bootcd']
+                                               self.mergedb[loginbase][nodename]['plcnode']=x['plcnode']
+                                               ticket = get_ticket_id(self.mergedb[loginbase][nodename])
+                                               self.mergedb[loginbase][nodename]['rt'] = mailer.getTicketStatus(ticket)
+
                                        # delete the entry from cache_all to keep it out of case 3)
                                        del self.cache_all[nodename]
 
@@ -477,6 +471,17 @@ class Diagnose(Thread):
                        diag_record['message'] = emailTxt.mailtxt.newdown
                        diag_record['args'] = {'nodename': nodename}
                        diag_record['info'] = (nodename, s_daysdown, "")
+
+                       if 'reboot_node_failed' in node_record:
+                               # there was a previous attempt to use the PCU.
+                               if node_record['reboot_node_failed'] == False:
+                                       # then the last attempt apparently, succeeded.
+                                       # But, the category is still 'ERROR'.  Therefore, the
+                                       # PCU-to-Node mapping is broken.
+                                       #print "Setting message for ERROR node to PCU2NodeMapping: %s" % nodename
+                                       diag_record['message'] = emailTxt.mailtxt.pcutonodemapping
+                                       diag_record['email_pcu'] = True
+
                        if diag_record['ticket_id'] == "":
                                diag_record['log'] = "DOWN: %20s : %-40s == %20s %s" % \
                                        (loginbase, nodename, diag_record['info'][1:], diag_record['found_rt_ticket'])
@@ -527,6 +532,14 @@ class Diagnose(Thread):
                                        diag_record['args'] = {'nodename': nodename}
                                        diag_record['info'] = (nodename, node_record['prev_category'], 
                                                                                                         node_record['category'])
+                                       if 'email_pcu' in diag_record:
+                                               if diag_record['email_pcu']:
+                                                       # previously, the pcu failed to reboot, so send
+                                                       # email. Now, reset these values to try the reboot
+                                                       # again.
+                                                       diag_record['email_pcu'] = False
+                                                       del diag_record['reboot_node_failed']
+
                                        if diag_record['ticket_id'] == "":
                                                diag_record['log'] = "IMPR: %20s : %-40s == %20s %20s %s %s" % \
                                                                        (loginbase, nodename, diag_record['stage'], 
@@ -536,27 +549,27 @@ class Diagnose(Thread):
                                                                        (loginbase, nodename, diag_record['stage'], 
                                                                         state, category, diag_record['ticket_id'])
                                        return diag_record
-                               elif time_diff >= 6*SPERHOUR:
-                                       # heartbeat is older than 30 min.
-                                       # then reset NM.
-                                       #print "Possible NM problem!! %s - %s = %s" % (now, last_contact, time_diff)
-                                       diag_record = {}
-                                       diag_record.update(node_record)
-                                       diag_record['message'] = emailTxt.mailtxt.NMReset
-                                       diag_record['args'] = {'nodename': nodename}
-                                       diag_record['stage'] = "nmreset"
-                                       diag_record['info'] = (nodename, 
-                                                                                       node_record['prev_category'], 
-                                                                                       node_record['category'])
-                                       if diag_record['ticket_id'] == "":
-                                               diag_record['log'] = "NM  : %20s : %-40s == %20s %20s %s %s" % \
-                                                                       (loginbase, nodename, diag_record['stage'], 
-                                                                        state, category, diag_record['found_rt_ticket'])
-                                       else:
-                                               diag_record['log'] = "NM  : %20s : %-40s == %20s" % \
-                                                                       (loginbase, nodename, diag_record['stage'])
-
-                                       return diag_record
+                               #elif time_diff >= 6*SPERHOUR:
+                               #       # heartbeat is older than 30 min.
+                               #       # then reset NM.
+                               #       #print "Possible NM problem!! %s - %s = %s" % (now, last_contact, time_diff)
+                               #       diag_record = {}
+                               #       diag_record.update(node_record)
+                               #       diag_record['message'] = emailTxt.mailtxt.NMReset
+                               #       diag_record['args'] = {'nodename': nodename}
+                               #       diag_record['stage'] = "nmreset"
+                               #       diag_record['info'] = (nodename, 
+                               #                                                       node_record['prev_category'], 
+                               #                                                       node_record['category'])
+                               #       if diag_record['ticket_id'] == "":
+                               #               diag_record['log'] = "NM  : %20s : %-40s == %20s %20s %s %s" % \
+                               #                                       (loginbase, nodename, diag_record['stage'], 
+                               #                                        state, category, diag_record['found_rt_ticket'])
+                               #       else:
+                               #               diag_record['log'] = "NM  : %20s : %-40s == %20s" % \
+                               #                                       (loginbase, nodename, diag_record['stage'])
+#
+#                                      return diag_record
                                else:
                                        return None
                        else:
@@ -579,36 +592,61 @@ class Diagnose(Thread):
        def __diagnoseNode(self, loginbase, node_record):
                # TODO: change the format of the hostname in this 
                #               record to something more natural.
-               nodename          = node_record['nodename']
-               category          = node_record['category']
-               prev_category = node_record['prev_category']
-               state             = node_record['state']
-
-               val = cmpCategoryVal(category, prev_category)
-               if val == -1:
-                       # current category is worse than previous, carry on
-                       pass
-               elif val == 1:
-                       # current category is better than previous
-                       # TODO: too generous for now, but will be handled correctly
-                       # TODO: if stage is currently ticket_waitforever, 
-                       if 'ticket_id' not in node_record:
-                               print "ignoring: ", node_record['nodename']
-                               return None
-                       else:
-                               if node_record['ticket_id'] == "" or \
-                                  node_record['ticket_id'] == None:
-                                       print "closing: ", node_record['nodename']
+               nodename                = node_record['nodename']
+               category                = node_record['category']
+               prev_category   = node_record['prev_category']
+               state                   = node_record['state']
+               #if 'prev_category' in node_record:
+               #       prev_category = node_record['prev_category']
+               #else:
+               #       prev_category = "ERROR"
+               if node_record['prev_category'] != "NORECORD":
+               
+                       val = cmpCategoryVal(category, prev_category)
+                       print "%s went from %s -> %s" % (nodename, prev_category, category)
+                       if val == 1:
+                               # improved
+                               if node_record['ticket_id'] == "" or node_record['ticket_id'] == None:
+                                       print "closing record with no ticket: ", node_record['nodename']
                                        node_record['action'] = ['close_rt']
                                        node_record['message'] = None
                                        node_record['stage'] = 'monitor-end-record'
                                        return node_record
-                                       #return None
                                else:
                                        node_record['stage'] = 'improvement'
-               else:
-                       #values are equal, carry on.
-                       pass
+
+                               #if 'monitor-end-record' in node_record['stage']:
+                               #       # just ignore it if it's already ended.
+                               #       # otherwise, the status should be worse, and we won't get
+                               #       # here.
+                               #       print "monitor-end-record: ignoring ", node_record['nodename']
+                               #       return None
+#
+#                                      #return None
+                       elif val == -1:
+                               # current category is worse than previous, carry on
+                               pass
+                       else:
+                               #values are equal, carry on.
+                               #print "why are we here?"
+                               pass
+
+               if 'rt' in node_record and 'Status' in node_record['rt']:
+                       if node_record['stage'] == 'ticket_waitforever':
+                               if 'resolved' in node_record['rt']['Status']:
+                                       print "ending waitforever record for: ", node_record['nodename']
+                                       node_record['action'] = ['noop']
+                                       node_record['message'] = None
+                                       node_record['stage'] = 'monitor-end-record'
+                                       print "oldlog: %s" % node_record['log'],
+                                       print "%15s" % node_record['action']
+                                       return node_record
+                               if 'new' in node_record['rt']['Status'] and \
+                                       'Queue' in node_record['rt'] and \
+                                       'Monitor' in node_record['rt']['Queue']:
+
+                                       print "RESETTING stage to findbad"
+                                       node_record['stage'] = 'findbad'
                        
                #### COMPARE category and prev_category
                # if not_equal
@@ -617,10 +655,13 @@ class Diagnose(Thread):
                #       then check category for stats.
                diag_record = self.diagRecordByCategory(node_record)
                if diag_record == None:
+                       #print "diag_record == None"
                        return None
 
                #### found_RT_ticket
                # TODO: need to record time found, and maybe add a stage for acting on it...
+               # NOTE: after found, if the support ticket is resolved, the block is
+               #               not removed. How to remove the block on this?
                if 'found_rt_ticket' in diag_record and \
                        diag_record['found_rt_ticket'] is not None:
                        if diag_record['stage'] is not 'improvement':
@@ -651,10 +692,18 @@ class Diagnose(Thread):
                        act_record['action'] = ['reset_nodemanager']
                        act_record['message'] = message[0]
                        act_record['stage']  = 'nmreset'
+                       return None
+
+               elif 'reboot_node' in diag_record['stage']:
+                       act_record['email'] = TECH
+                       act_record['action'] = ['noop']
+                       act_record['message'] = message[0]
+                       act_record['stage'] = 'stage_actinoneweek'
                        
                elif 'improvement' in diag_record['stage']:
                        # - backoff previous squeeze actions (slice suspend, nocreate)
                        # TODO: add a backoff_squeeze section... Needs to runthrough
+                       print "backing off of %s" % nodename
                        act_record['action'] = ['close_rt']
                        act_record['message'] = message[0]
                        act_record['stage'] = 'monitor-end-record'
@@ -674,6 +723,7 @@ class Diagnose(Thread):
                        else:
                                act_record['message'] = None
                                act_record['action'] = ['waitforoneweekaction' ]
+                               print "ignoring this record for: %s" % act_record['nodename']
                                return None                     # don't send if there's no action
 
                elif 'actintwoweeks' in diag_record['stage']:
@@ -731,11 +781,18 @@ class Diagnose(Thread):
                        #       1. stage is unknown, or 
                        #       2. delta is not big enough to bump it to the next stage.
                        # TODO: figure out which. for now assume 2.
-                       print "UNKNOWN!?!? %s" % nodename
+                       print "UNKNOWN stage for %s; nothing done" % nodename
                        act_record['action'] = ['unknown']
                        act_record['message'] = message[0]
-                       print "Exiting..."
-                       sys.exit(1)
+
+                       act_record['email'] = TECH
+                       act_record['action'] = ['noop']
+                       act_record['message'] = message[0]
+                       act_record['stage'] = 'stage_actinoneweek'
+                       act_record['time'] = current_time               # reset clock
+                       #print "Exiting..."
+                       #return None
+                       #sys.exit(1)
 
                print "%s" % act_record['log'],
                print "%15s" % act_record['action']
@@ -833,6 +890,10 @@ def close_rt_backoff(args):
                plc.enableSliceCreation(args['hostname'])
        return
 
+def reboot_node(args):
+       host = args['hostname']
+       return reboot.reboot_policy(host, True, config.debug)
+
 def reset_nodemanager(args):
        os.system("ssh root@%s /sbin/service nm restart" % nodename)
        return
@@ -856,6 +917,7 @@ class Action(Thread):
                self.actions['close_rt'] = lambda args: close_rt_backoff(args)
                self.actions['rins'] = lambda args: plc.nodeBootState(args['hostname'], "rins") 
                self.actions['noop'] = lambda args: args
+               self.actions['reboot_node'] = lambda args: reboot_node(args)
                self.actions['reset_nodemanager'] = lambda args: args # reset_nodemanager(args)
 
                self.actions['ticket_waitforever'] = lambda args: args
@@ -997,7 +1059,7 @@ class Action(Thread):
                return hlist
 
 
-       def get_email_args(self, act_recordlist):
+       def get_email_args(self, act_recordlist, loginbase=None):
 
                email_args = {}
                email_args['hostname_list'] = ""
@@ -1005,8 +1067,28 @@ class Action(Thread):
                for act_record in act_recordlist:
                        email_args['hostname_list'] += act_record['msg_format']
                        email_args['hostname'] = act_record['nodename']
+                       if  'plcnode' in act_record and \
+                               'pcu_ids' in act_record['plcnode'] and \
+                               len(act_record['plcnode']['pcu_ids']) > 0:
+                               print "setting 'pcu_id' for email_args %s"%email_args['hostname']
+                               email_args['pcu_id'] = act_record['plcnode']['pcu_ids'][0]
+                       else:
+                               email_args['pcu_id'] = "-1"
+                                       
                        if 'ticket_id' in act_record:
-                               email_args['ticket_id'] = act_record['ticket_id']
+                               if act_record['ticket_id'] == 0 or act_record['ticket_id'] == '0':
+                                       print "Enter the ticket_id for %s @ %s" % (loginbase, act_record['nodename'])
+                                       sys.stdout.flush()
+                                       line = sys.stdin.readline()
+                                       try:
+                                               ticket_id = int(line)
+                                       except:
+                                               print "could not get ticket_id from stdin..."
+                                               os._exit(1)
+                               else:
+                                       ticket_id = act_record['ticket_id']
+                                       
+                               email_args['ticket_id'] = ticket_id
 
                return email_args
 
@@ -1033,14 +1115,35 @@ class Action(Thread):
                        diag_record = site_record['nodes'][nodename]
                        act_record  = self.__actOnNode(diag_record)
                        #print "nodename: %s %s" % (nodename, act_record)
-                       act_recordlist += [act_record]
+                       if act_record is not None:
+                               act_recordlist += [act_record]
 
                unique_issues = self.get_unique_issues(act_recordlist)
 
                for issue in unique_issues.keys():
                        print "\tworking on issue: %s" % issue
                        issue_record_list = unique_issues[issue]
-                       email_args = self.get_email_args(issue_record_list)
+                       email_args = self.get_email_args(issue_record_list, loginbase)
+
+                       # for each record.
+                       for act_record in issue_record_list:
+                               # if there's a pcu record and email config is set
+                               if 'email_pcu' in act_record:
+                                       if act_record['message'] != None and act_record['email_pcu'] and site_record['config']['email']:
+                                               # and 'reboot_node' in act_record['stage']:
+
+                                               email_args['hostname'] = act_record['nodename']
+                                               ticket_id = self.__emailSite(loginbase, 
+                                                                                       act_record['email'], 
+                                                                                       emailTxt.mailtxt.pcudown[0],
+                                                                                       email_args)
+                                               if ticket_id == 0:
+                                                       # error.
+                                                       print "got a ticket_id == 0!!!! %s" % act_record['nodename']
+                                                       os._exit(1)
+                                                       pass
+                                               email_args['ticket_id'] = ticket_id
+
                        
                        act_record = issue_record_list[0]
                        # send message before squeezing
@@ -1050,12 +1153,25 @@ class Action(Thread):
                                ticket_id = self.__emailSite(loginbase, act_record['email'], 
                                                                                         act_record['message'], email_args)
 
+                               if ticket_id == 0:
+                                       # error.
+                                       print "ticket_id == 0 for %s %s" % (loginbase, act_record['nodename'])
+                                       os._exit(1)
+                                       pass
+
                                # Add ticket_id to ALL nodenames
                                for act_record in issue_record_list:
                                        nodename = act_record['nodename']
                                        # update node record with RT ticket_id
                                        if nodename in self.act_all:
                                                self.act_all[nodename][0]['ticket_id'] = "%s" % ticket_id
+                                               # if the ticket was previously resolved, reset it to new.
+                                               if 'rt' in act_record and \
+                                                       'Status' in act_record['rt'] and \
+                                                       act_record['rt']['Status'] == 'resolved':
+                                                       mailer.setTicketStatus(ticket_id, "new")
+                                               status = mailer.getTicketStatus(ticket_id)
+                                               self.act_all[nodename][0]['rt'] = status
                                        if config.mail: i_nodes_emailed += 1
 
                        print "\t\tconfig.squeeze: %s and %s" % (config.squeeze,
@@ -1088,9 +1204,57 @@ class Action(Thread):
                act_record.update(diag_record)
                act_record['nodename'] = nodename
                act_record['msg_format'] = self._format_diaginfo(diag_record)
+               print "act_record['stage'] == %s " % act_record['stage']
+
+               # avoid end records, and nmreset records                                        
+               # reboot_node_failed, is set below, so don't reboot repeatedly.
+
+               if 'monitor-end-record' not in act_record['stage'] and \
+                  'nmreset' not in act_record['stage'] and \
+                  'reboot_node_failed' not in act_record:
+
+                       if "DOWN" in act_record['log'] and \
+                                       'pcu_ids' in act_record['plcnode'] and \
+                                       len(act_record['plcnode']['pcu_ids']) > 0:
+
+                               print "%s" % act_record['log'],
+                               print "%15s" % (['reboot_node'],)
+                               # Set node to re-install
+                               plc.nodeBootState(act_record['nodename'], "rins")       
+                               try:
+                                       ret = reboot_node({'hostname': act_record['nodename']})
+                               except Exception, exc:
+                                       print "exception on reboot_node:"
+                                       import traceback
+                                       print traceback.print_exc()
+                                       ret = False
+
+                               if ret: # and ( 'reboot_node_failed' not in act_record or act_record['reboot_node_failed'] == False):
+                                       # Reboot Succeeded
+                                       print "reboot succeeded for %s" % act_record['nodename']
+                                       act_record2 = {}
+                                       act_record2.update(act_record)
+                                       act_record2['action'] = ['reboot_node']
+                                       act_record2['stage'] = "reboot_node"
+                                       act_record2['reboot_node_failed'] = False
+                                       act_record2['email_pcu'] = False
+
+                                       if nodename not in self.act_all: 
+                                               self.act_all[nodename] = []
+                                       print "inserting 'reboot_node' record into act_all"
+                                       self.act_all[nodename].insert(0,act_record2)
+
+                                       # return None to avoid further action
+                                       print "Taking no further action"
+                                       return None
+                               else:
+                                       print "reboot failed for %s" % act_record['nodename']
+                                       # set email_pcu to also send pcu notice for this record.
+                                       act_record['reboot_node_failed'] = True
+                                       act_record['email_pcu'] = True
 
-               print "%s" % act_record['log'],
-               print "%15s" % act_record['action']
+                       print "%s" % act_record['log'],
+                       print "%15s" % act_record['action']
 
                if act_record['stage'] is not 'monitor-end-record' and \
                   act_record['stage'] is not 'nmreset':