www/printbadnodes.py

[monitor.git] / clean_policy.py

diff --git a/clean_policy.py b/clean_policy.py
index dba9b9b..d2bde41 100644 (file)
--- a/clean_policy.py
+++ b/clean_policy.py
@@ -1,15 +1,12 @@
-from config import config
-#print "policy"
-config = config()
-import soltesz
+import config
+import database 
 import time
 import mailer
-from www.printbadnodes import cmpCategoryVal
+from unified_model import cmpCategoryVal
 import sys
 import emailTxt
 import string
 
-from policy import get_ticket_id, print_stats, close_rt_backoff, reboot_node
 from rt import is_host_in_rt_tickets
 import plc
 
@@ -23,18 +20,28 @@ from const import *
 
 from unified_model import *
 
+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 MonitorMergeDiagnoseSendEscellate:
        def __init__(self, hostname, act):
                self.hostname = hostname
                self.act = act
                self.plcdb_hn2lb = None
                if self.plcdb_hn2lb is None:
-                       self.plcdb_hn2lb = soltesz.dbLoad("plcdb_hn2lb")
+                       self.plcdb_hn2lb = database.dbLoad("plcdb_hn2lb")
                self.loginbase = self.plcdb_hn2lb[self.hostname]
                return
 
        def getFBRecord(self):
-               fb = soltesz.dbLoad("findbad")
+               fb = database.dbLoad("findbad")
                if self.hostname in fb['nodes']:
                        fbnode = fb['nodes'][self.hostname]['values']
                else:
@@ -43,7 +50,7 @@ class MonitorMergeDiagnoseSendEscellate:
 
        def getActionRecord(self):
                # update ticket status
-               act_all = soltesz.dbLoad("act_all")
+               act_all = database.dbLoad("act_all")
                if self.hostname in act_all and len(act_all[self.hostname]) > 0:
                        actnode = act_all[self.hostname][0]
                else:
@@ -121,7 +128,7 @@ class MonitorMergeDiagnoseSendEscellate:
                        record.data['log'] = self.getDownLog(record)
 
                elif category == "prod":
-                       state = diag.getState()
+                       state = record.getState()
                        if state == "boot":
                                diag.setFlag('SendThankyou')
                                record.data['message'] = emailTxt.mailtxt.newthankyou
@@ -199,9 +206,9 @@ class MonitorMergeDiagnoseSendEscellate:
                return True
 
        def add_and_save_act_all(self, record):
-               self.act_all = soltesz.dbLoad("act_all")
+               self.act_all = database.dbLoad("act_all")
                self.act_all[self.hostname].insert(0,record.data)
-               soltesz.dbDump("act_all", self.act_all)
+               database.dbDump("act_all", self.act_all)
                
        def getDownLog(self, record):