Changeset 1751 for palm/trunk/SCRIPTS/palm_wd

Timestamp:

Feb 15, 2016 7:44:16 AM (9 years ago)

Author:

maronga

Message:

watchdog changes and bugfixes

File:

• palm/trunk/SCRIPTS/palm_wd (modified) (15 diffs)

palm/trunk/SCRIPTS/palm_wd

@@ -20 +20 @@
 # Current revisions:
 # -----------------
-# 
+# Bugfixes: runs on multiple hosts caused crash of the watchdog. Progress bar
+# occasionally showed wrong progress.
+# New: Hosts can be switched on/off and update_frequency can be modified via 
+# File->Options. Security check for cancelation of jobs.
 # 
 # Former revisions:
@@ -69 +72 @@
 import time
 
+global update_frequency
+
 # Determine PALM directories
 try: 
@@ -99 +104 @@
  
     if ( description_tmp != 'Settings' ):
-       description.append(description_tmp)
-       hostname.append(config.get(description_tmp, 'hostname'))
-       username.append(config.get(description_tmp, 'username')) 
+      
+       if ( config.get(description_tmp, 'enabled') == 'true' ):
+          description.append(description_tmp)
+          hostname.append(config.get(description_tmp, 'hostname'))
+          username.append(config.get(description_tmp, 'username')) 
     else:
        update_frequency = int(config.get(description_tmp, 'update_frequency'))*60000
@@ -123 +130 @@
 job_data_list = []
 
+# Message box for showing RUN_CONTROL output
 class MessageBoxScroll(QtGui.QMessageBox):
     def __init__(self):
@@ -149 +157 @@
 
         return result
+
+
+# Message box for showing RUN_CONTROL output
+class OptionBox(QtGui.QDialog):
+    def __init__(self):
+      
+        super(OptionBox, self).__init__()
+      
+        uic.loadUi(palm_bin + '/palm_wd_files/wdoptions.ui', self)
+        
+        hostname = []
+        self.checkbox = []
+        j = -1
+        ypos = 0
+   
+        self.vbox = QtGui.QVBoxLayout(self)
+        self.vbox.setSpacing(0)
+        self.vbox.setMargin(0)       
+    
+        config.read(palm_dir + '/.wd.config')
+    
+        for i in range(0,len(config.sections())):
+
+           description_tmp = config.sections()[i]
+ 
+           if ( description_tmp != 'Settings' ):
+              description.append(description_tmp)
+              hostname.append(config.get(description_tmp, 'hostname'))
+              
+              j = j + 1
+              self.checkbox.append(j)
+              self.checkbox[j] = QtGui.QCheckBox(description_tmp, self)   
+              ypos = ypos + 20
+              self.checkbox[j].move(10,0)
+              if ( config.get(description_tmp, 'enabled') == 'true' ):
+                 self.checkbox[j].toggle()
+                       
+              self.vbox.addWidget(self.checkbox[j])
+           else:
+              self.update_spin.setValue(int(config.get(description_tmp, 'update_frequency')))
+     
+        self.hostBox.setLayout(self.vbox)
+        self.hostBox.setGeometry(0, 0, 272, 50+ypos)
+        self.settingBox.setGeometry(281, 0, 214, 50+ypos)
+        self.ok.setGeometry(385, 50+ypos, 111, 24)
+        self.cancel.setGeometry(270, 50+ypos, 111, 24)
+        
+        self.setGeometry(0, 0, 500, 80+ypos)
+        self.show()
+        
+        return
+
+#   save updtes to file
+    def SaveAndClose(self):
+      
+       for i in range(0,len(self.checkbox)):
+          
+          if ( self.checkbox[i].checkState() == 0 ):
+             config.set(str(self.checkbox[i].text()),'enabled','false')
+          else:
+             config.set(str(self.checkbox[i].text()),'enabled','true')       
+       
+       config.set('Settings','update_frequency',self.update_spin.text())  
+
+       with open(palm_dir + '/.wd.config', 'wb') as configfile:
+          config.write(configfile)
+
+
+       self.close()
+     
+       return
+
 
 
@@ -273 +353 @@
       
         # Use global variables
-        global jobs_save
         global pbars
+        global update_frequency
 
         # Deactivate timer processes until update is finished
@@ -317 +397 @@
             job_data = []
             job_data_n = []    
-            job_data_old = []
-            job_data_old_n = []
             job_progress = 0
 
@@ -353 +431 @@
             result = file2.readlines()
             file2.close()
-            
+
+            job_data_old = []
+            job_data_old_n = []
+            k = -1
             for j in range(0,len(result)):
                 if ( result[j].split()[1] == description[h] ):
-                    job_data_old.append(j)
-                    job_data_old[j] = result[j].split(" ")
-                    job_data_old[j] = filter(None, job_data_old[j])
-                    job_data_old[j] = [line.rstrip('\n') for line in job_data_old[j]]
-                
-                    job_data_old_n.append(j)
-                    job_data_old_n[j] = job_data_old[j][6]
+                    k = k + 1
+                    job_data_old.append(k)
+                    job_data_old[k] = result[j].split(" ")
+                    job_data_old[k] = filter(None, job_data_old[k])
+                    job_data_old[k] = [line.rstrip('\n') for line in job_data_old[k]]
+                    job_data_old_n.append(k)
+                    job_data_old_n[k] = job_data_old[k][6]
 
             # Merge the job_data and job_data_old to find completed, new and
@@ -410 +491 @@
                 # Apply a color depending on the status of the job
                 if ( job_data[to_display[j]][3] == "Running" ):
-                    self.table.setCellWidget(i,4,pbars[j]) 
+                    self.table.setCellWidget(i,4,pbars[i]) 
                     self.table.item(i,0).setBackground(QtGui.QColor(255, 251, 168))
                     self.table.item(i,1).setBackground(QtGui.QColor(255, 251, 168))
@@ -419 +500 @@
                 else:
                     pbars[j].setEnabled(False)
-                    self.table.setCellWidget(i,4,pbars[j])
+                    self.table.setCellWidget(i,4,pbars[i])
                     self.table.item(i,0).setBackground(QtGui.QColor(255, 112, 118))
                     self.table.item(i,1).setBackground(QtGui.QColor(255, 112, 118))
@@ -549 +630 @@
             del(jobs_new)
             del(result)
-
+            del(job_data_old)
+            del(job_data_old_n)
+            
         file.close()
 
@@ -564 +647 @@
         self.label2.setText("Last update: " + timestamp)
         self.label.setText("Update complete.")
-        self.timetimer.restart()
-        self.timer.start()
-        self.labeltimer.start()
+        self.timetimer.start()
+        self.timer.start(update_frequency)
+        self.labeltimer.start(update_frequency/10)
+        self.UpdateLabel()
         QtGui.QApplication.processEvents()
        
@@ -577 +661 @@
         # Check description of job in order to login on the correct host
         descr = self.table.item(self.table.currentRow(),1).text()
-        for h in range(0,len(description)):
-            if ( descr == description[h] ):
-                host = username[h] + "@" + hostname[h]
-
-        ssh = sub.Popen(["ssh", "%s" % host, "/sw/tools/python/2.7.6/generic/bin/python palm_wdd cancel " + jobname],
-                           shell=False,
-                           stdout=sub.PIPE,
-                           stderr=sub.PIPE)
-        result = ssh.stdout.readlines()
-        result = filter(None, result)
-        
-        # In case of error display a warning message
-        if ( len(result) == 0 ):
-            error = ssh.stderr.readlines()
-            notify = QtGui.QMessageBox.warning(self,'Cancel job',"Error. Could not cancel job " + jobname + ".\n\nError message:\n" + ''.join(error))
-
-        # Otherwise inform the user and mark the job in the table
-        else:
-            self.table.setItem(self.table.currentRow(),3,QtGui.QTableWidgetItem("Canceled"))
-            notify = QtGui.QMessageBox.information(self,'Cancel job',"Job" + jobname + " canceled!\n\nServer message:\n" + ''.join(result))
+        
+        querybox = QtGui.QMessageBox()
+        querybox.setText("Attention!\nYou are trying to cancel a job. It will not be able to terminate properly.")
+        querybox.setInformativeText("Do you really want to cancel " + jobname + " on host " + descr + "?")
+        querybox.setStandardButtons(QtGui.QMessageBox.Yes | QtGui.QMessageBox.No)
+        querybox.setDefaultButton(QtGui.QMessageBox.No)
+        returnvalue = querybox.exec_()
+       
+        if ( returnvalue == QtGui.QMessageBox.Yes ):
+        
+           for h in range(0,len(description)):
+               if ( descr == description[h] ):
+                  host = username[h] + "@" + hostname[h]
+
+           ssh = sub.Popen(["ssh", "%s" % host, "/sw/tools/python/2.7.6/generic/bin/python palm_wdd cancel " + jobname],
+                              shell=False,
+                              stdout=sub.PIPE,
+                              stderr=sub.PIPE)
+           result = ssh.stdout.readlines()
+           result = filter(None, result)
+        
+           # In case of error display a warning message
+           if ( len(result) == 0 ):
+               error = ssh.stderr.readlines()
+               notify = QtGui.QMessageBox.warning(self,'Cancel job',"Error. Could not cancel job " + jobname + ".\n\nError message:\n" + ''.join(error))
+
+           # Otherwise inform the user and mark the job in the table
+           else:
+               self.table.setItem(self.table.currentRow(),3,QtGui.QTableWidgetItem("Canceled"))
+               notify = QtGui.QMessageBox.information(self,'Cancel job',"Job" + jobname + " canceled!\n\nServer message:\n" + ''.join(result))
 
 
@@ -650 +744 @@
         else:
             self.table.setItem(self.table.currentRow(),3,QtGui.QTableWidgetItem("Terminated"))
-            notify = QtGui.QMessageBox.information(self,'Terminatie job',"Termination of job " + jobname + " was initiated!\n\nServer message:\n" + ''.join(result))
+            notify = QtGui.QMessageBox.information(self,'Terminate job',"Termination of job " + jobname + " was initiated!\n\nServer message:\n" + ''.join(result))
 
 
@@ -810 +904 @@
         
 
+    # Enter Options menu
+    def Options(self):
+  
+#       disable mainwindow  
+        self.setEnabled(False)
+      
+#       show Options Dialog      
+        opt = OptionBox()
+        opt.exec_()
+
+        self.UpdateHosts()
+        self.setEnabled(True)
+
+# Update settings
+    def UpdateHosts(self):
+
+       global update_frequency
+       description = []
+       hostname = []
+       username = []
+
+       for i in range(0,len(config.sections())):
+
+          description_tmp = config.sections()[i]
+ 
+          if ( description_tmp != 'Settings' ):
+      
+             if ( config.get(description_tmp, 'enabled') == 'true' ):
+                description.append(description_tmp)
+                hostname.append(config.get(description_tmp, 'hostname'))
+                username.append(config.get(description_tmp, 'username')) 
+          else:
+             update_frequency = int(config.get(description_tmp, 'update_frequency'))*60000
+
+
 # Main loop       
 def main():