source: palm/trunk/SCRIPTS/palm_wd @ 2809

Last change on this file since 2809 was 2718, checked in by maronga, 7 years ago

deleting of deprecated files; headers updated where needed

  • Property svn:executable set to *
  • Property svn:keywords set to Id
File size: 39.8 KB
Line 
1#!/usr/bin/python
2# -*- coding: utf-8 -*-
3#--------------------------------------------------------------------------------#
4# This file is part of the PALM model system.
5#
6# PALM is free software: you can redistribute it and/or modify it under the terms
7# of the GNU General Public License as published by the Free Software Foundation,
8# either version 3 of the License, or (at your option) any later version.
9#
10# PALM is distributed in the hope that it will be useful, but WITHOUT ANY
11# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
12# A PARTICULAR PURPOSE.  See the GNU General Public License for more details.
13#
14# You should have received a copy of the GNU General Public License along with
15# PALM. If not, see <http://www.gnu.org/licenses/>.
16#
17# Copyright 1997-2018  Leibniz Universitaet Hannover
18#--------------------------------------------------------------------------------#
19#
20# Current revisions:
21# -----------------
22#
23#
24# Former revisions:
25# -----------------
26# $Id: palm_wd 2718 2018-01-02 08:49:38Z schwenkel $
27# Corrected "Former revisions" section
28#
29# 2696 2017-12-14 17:12:51Z kanani
30# Change in file header (GPL part)
31#
32# 2416 2017-09-06 14:28:14Z maronga
33# Adapted for palmrun
34#
35# 1754 2016-02-22 13:50:22Z maronga
36#
37# 1753 2016-02-22 13:49:49Z maronga
38# Bugfix: use of global variables is required after updating configuration file
39#
40# 1751 2016-02-15 07:44:16Z maronga
41# Bugfixes: runs on multiple hosts caused crash of the watchdog. Progress bar
42# occasionally showed wrong progress.
43# New: Hosts can be switched on/off and update_frequency can be modified via
44# File->Options. Security check for cancelation of jobs.
45#
46# 1618 2015-07-13 06:52:15Z maronga
47# Added steering via configuration file .wd.config, to be place in the local
48# palm directory. Watchdog save files are automatically created upon first start.
49#
50# 1613 2015-07-08 14:53:29Z maronga
51# Bugfix: tooltip for queuing name did not show up on first update.
52# New: added contect menu for showing the parameter file and the run control
53# output
54#
55# 1611 2015-07-07 12:23:22Z maronga
56# Initial revision
57#
58#
59# Description:
60# ------------
61# PALM watchdog client for monitoring batch jobs on a variety of hosts specified
62# by the user
63#
64# Instructions:
65# -------------
66# 1) Modify the header section of palm_wd
67# 2) Move .wd.olddata and .wd.newdata to your palm directory
68#    (e.g. /home/user/current_version/.wd.newdata etc.)
69# 3) Modify a copy of palm_wdd for each host to be monitored and move it to the
70#    respective hosts
71# 4) Start the client either from mrungui or from shell by "nohup palm_wd&"
72
73# To do:
74# ------
75# 1) Add "Options", "Help" and "Manual"
76# 2) Move user settings to a configuration file
77#------------------------------------------------------------------------------!
78
79import ConfigParser
80import datetime
81import os
82from PyQt4 import QtGui, QtCore, uic
83from PyQt4.QtCore import pyqtSlot,SIGNAL,SLOT
84import shutil
85import subprocess as sub
86import sys
87import time
88
89# Determine PALM directories
90try: 
91   devnull = open(os.devnull, 'w')
92   out = sub.check_output("echo $PALM_BIN", shell=True, stderr=sub.STDOUT)
93   palm_bin = out.rstrip()
94   palm_dir = out.split("palm")[0] + "palm/" + out.split("palm")[1].split("/")[1]
95   out = None
96except:   
97   print "Error. $PALM_BIN is not set."
98   raise SystemExit
99
100
101# Read configuration file
102# First check if the configuration file exists
103if ( os.path.exists(palm_dir + '/.wd.config') == False ):
104    print "Error. No configuration file .wd.config found in " + palm_dir
105    raise SystemExit     
106
107config = ConfigParser.RawConfigParser()
108config.read(palm_dir + '/.wd.config')
109
110description = []
111hostname = []
112username = []
113
114for i in range(0,len(config.sections())):
115
116    description_tmp = config.sections()[i]
117 
118    if ( description_tmp != 'Settings' ):
119     
120       if ( config.get(description_tmp, 'enabled') == 'true' ):
121          description.append(description_tmp)
122          hostname.append(config.get(description_tmp, 'hostname'))
123          username.append(config.get(description_tmp, 'username')) 
124    else:
125       update_frequency = int(config.get(description_tmp, 'update_frequency'))*60000
126
127# Check if .wd.olddata and .wd.newdata exist. Otherwise create the files
128if ( os.path.exists(palm_dir + '/.wd.olddata') == False ):
129   print "No .wd.olddata found. Creating..."
130   file1 = open(palm_dir + '/.wd.olddata', 'a')
131   file1.close()
132   
133
134if ( os.path.exists(palm_dir + '/.wd.newdata') == False ):
135   print "No .wd.newdata found. Creating..." 
136   file1 = open(palm_dir + '/.wd.newdata', 'a')
137   file1.close()
138
139# Dummy variables for the jobname and the progressbars
140jobname = ""
141timestamp = ""
142pbars = []
143job_data_list = []
144
145# Message box for showing RUN_CONTROL output
146class MessageBoxScroll(QtGui.QMessageBox):
147    def __init__(self):
148        QtGui.QMessageBox.__init__(self)
149        self.setSizeGripEnabled(True)
150
151    def event(self, e):
152        result = QtGui.QMessageBox.event(self, e)
153
154        self.setMinimumHeight(100)
155        self.setMaximumHeight(16777215)
156        self.setMinimumWidth(400)
157        self.setMaximumWidth(16777215)
158        self.setSizePolicy(QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Expanding)
159
160        textEdit = self.findChild(QtGui.QTextEdit)
161        if textEdit != None :
162            textEdit.setMinimumHeight(800)
163            textEdit.setMaximumHeight(16777215)
164            textEdit.setMinimumWidth(1000)
165            textEdit.setMaximumWidth(16777215)
166            textEdit.setSizePolicy(QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Expanding)
167            f = QtGui.QFont('not_a_font') #
168            f.setStyleHint(f.TypeWriter, f.PreferDefault)
169            textEdit.setFont(f)
170
171        return result
172
173
174# Message box for showing RUN_CONTROL output
175class OptionBox(QtGui.QDialog):
176    def __init__(self):
177     
178        super(OptionBox, self).__init__()
179     
180        uic.loadUi(palm_bin + '/palm_wd_files/wdoptions.ui', self)
181       
182        hostname = []
183        self.checkbox = []
184        j = -1
185        ypos = 0
186   
187        self.vbox = QtGui.QVBoxLayout(self)
188        self.vbox.setSpacing(0)
189        self.vbox.setMargin(0)       
190   
191        config.read(palm_dir + '/.wd.config')
192   
193        for i in range(0,len(config.sections())):
194
195           description_tmp = config.sections()[i]
196 
197           if ( description_tmp != 'Settings' ):
198              description.append(description_tmp)
199              hostname.append(config.get(description_tmp, 'hostname'))
200             
201              j = j + 1
202              self.checkbox.append(j)
203              self.checkbox[j] = QtGui.QCheckBox(description_tmp, self)   
204              ypos = ypos + 20
205              self.checkbox[j].move(10,0)
206              if ( config.get(description_tmp, 'enabled') == 'true' ):
207                 self.checkbox[j].toggle()
208                       
209              self.vbox.addWidget(self.checkbox[j])
210           else:
211              self.update_spin.setValue(int(config.get(description_tmp, 'update_frequency')))
212     
213        self.hostBox.setLayout(self.vbox)
214        self.hostBox.setGeometry(0, 0, 272, 50+ypos)
215        self.settingBox.setGeometry(281, 0, 214, 50+ypos)
216        self.ok.setGeometry(385, 50+ypos, 111, 24)
217        self.cancel.setGeometry(270, 50+ypos, 111, 24)
218       
219        self.setGeometry(0, 0, 500, 80+ypos)
220        self.show()
221       
222        return
223
224#   save updtes to file
225    def SaveAndClose(self):
226     
227       for i in range(0,len(self.checkbox)):
228         
229          if ( self.checkbox[i].checkState() == 0 ):
230             config.set(str(self.checkbox[i].text()),'enabled','false')
231          else:
232             config.set(str(self.checkbox[i].text()),'enabled','true')       
233       
234       config.set('Settings','update_frequency',self.update_spin.text()) 
235
236       with open(palm_dir + '/.wd.config', 'wb') as configfile:
237          config.write(configfile)
238
239
240       self.close()
241     
242       return
243
244
245
246# MainWindow class
247class Watchdog(QtGui.QMainWindow):
248   
249    def __init__(self):
250        super(Watchdog, self).__init__()
251       
252        self.InitUi()   
253       
254   
255    # Initialized MainWindow UI
256    def InitUi(self):
257
258        # Load predefined mainwindow
259        uic.loadUi(palm_bin + '/palm_wd_files/wd.ui', self)
260
261        # Resize columns and set number of rows to zero. Column 6 is hidden and
262        # contains the remote jobname (e.g. hannover.174610)
263        self.table.setColumnWidth(0,230)
264        self.table.setColumnWidth(1,80)
265        self.table.setColumnWidth(2,50)
266        self.table.setColumnWidth(3,80)     
267        self.table.setColumnWidth(4,180)
268        self.table.setColumnWidth(5,115)   
269        self.table.setColumnHidden(6, True)
270        self.table.setRowCount(0) 
271   
272        # Display MainWindow
273        self.show()
274        QtGui.QApplication.processEvents()
275
276        # Start refresh timer. On timeout perform update
277        self.timer = QtCore.QTimer(self)
278        self.timer.timeout.connect(self.Refresh)
279        self.timer.setSingleShot(False)
280        self.timer.start(update_frequency)
281
282        # The timetimer counts the time since last update
283        self.timetimer= QtCore.QElapsedTimer()
284        self.timetimer.start()
285
286        # The labeltimer induces the update of the remaining time in the UI
287        self.labeltimer = QtCore.QTimer(self)
288        self.labeltimer.timeout.connect(self.UpdateLabel)
289        self.labeltimer.setSingleShot(False)
290
291        # Update in the UI will be performed at each 1/10 of the update interval
292        self.labeltimer.start(update_frequency/10)   
293        self.label.setText("Next update in " + str(update_frequency/1000/60) + "min")
294
295        # Update the list now
296        self.Refresh()
297       
298
299    # Add a custom context menu
300    def OpenContextMenu(self, position):
301        menu = QtGui.QMenu()
302       
303        # "Show details" -> fetch details for the selected job
304        detailsjobAction = menu.addAction('Show job details')     
305        detailsjobAction.setStatusTip('Display detailed info for job')
306        detailsjobAction.triggered.connect(self.ShowDetails)
307
308        # "Show parameter file" -> show the contents of PARIN
309        parinAction = menu.addAction('Show parameter file')     
310        parinAction.setStatusTip('Display the parameter file of the job (e.g. PARIN / _p3d)')
311        parinAction.triggered.connect(self.ShowPARIN)
312       
313        rcAction = menu.addAction('Show run control file')     
314        rcAction.setStatusTip('Display the current run control file (e.g. RUN_CONTROL / _rc)')
315        rcAction.triggered.connect(self.ShowRC)
316       
317        # "Cancel job" -> remove job from queuing system
318        canceljobAction = menu.addAction('Cancel job')
319        canceljobAction.setStatusTip('Remove job from queueing system')
320        canceljobAction.triggered.connect(self.CancelJob)
321       
322        # "Force stop" -> force termination of job
323        stopjobAction = menu.addAction('Force stop')
324        stopjobAction.setStatusTip('Terminate job properly')
325        stopjobAction.triggered.connect(self.DoStop)
326   
327        # "Force restart" -> force rermination and restart of job
328        restartjobAction = menu.addAction('Force restart')
329        restartjobAction.setStatusTip('Terminate job properly and initiate restart')
330        restartjobAction.triggered.connect(self.DoRestart)
331 
332        # "Remove from list" -> remove a completed job from the list
333        removefromlistAction = menu.addAction('Remove from list')     
334        removefromlistAction.setStatusTip('Remove finished job')
335        removefromlistAction.triggered.connect(lambda: self.RemoveFromList(-1))     
336
337       
338        # Activate/deactive contect menu items based on the status of the runs
339        state = self.table.item(self.table.currentRow(),3).text()
340        if (state == "Canceled" or state == "Completed" or state == "Terminated"):
341            detailsjobAction.setEnabled(False)
342            canceljobAction.setEnabled(False) 
343            removefromlistAction.setEnabled(True) 
344        else:
345            detailsjobAction.setEnabled(True)
346            canceljobAction.setEnabled(True) 
347            removefromlistAction.setEnabled(False)   
348 
349        if ( state == "Running" ):
350            stopjobAction.setEnabled(True) 
351            restartjobAction.setEnabled(True)   
352            parinAction.setEnabled(True)
353            rcAction.setEnabled(True)         
354        else:
355            stopjobAction.setEnabled(False) 
356            restartjobAction.setEnabled(False)   
357            parinAction.setEnabled(False)
358            rcAction.setEnabled(False) 
359           
360        # Show the context menu
361        action = menu.exec_(self.table.mapToGlobal(position))
362
363
364    # Collecting watchdog data and display in the UI
365    def Refresh(self):
366     
367        # Use global variables
368        global pbars
369        global update_frequency
370
371        # Deactivate timer processes until update is finished
372        # QtGui.QApplication.processEvents() updates the UI whenever needed
373        self.labeltimer.stop() 
374        self.label.setText("Updating...") 
375        QtGui.QApplication.processEvents()
376        self.setEnabled(False)
377        QtGui.QApplication.processEvents()
378
379        # Get current timestamp
380        timestamp = datetime.datetime.now().strftime("%d/%m/%Y %H:%M")
381       
382        # Open an empty file for the watchdog data       
383        file = open(palm_dir + '/.wd.newdata', 'w')
384       
385 
386        # Set all required variables to their initial values. Sorting must be
387        # disabled.
388        self.table.setRowCount(0) 
389        self.table.setSortingEnabled(False)
390        i = -1
391        job_data_list = []
392        pbars = []
393       
394        # Scan all hosts in the variable hostname. For each host perform the
395        # update loop
396        for h in range(0,len(hostname)):
397         
398            # Perform ssh command
399            host    = username[h] + "@" + hostname[h]
400            ssh = sub.Popen(["ssh", "%s" % host, "/sw/tools/python/2.7.6/generic/bin/python palm_wdd queue " + username[h]],
401                           shell=False,
402                           stdout=sub.PIPE,
403                           stderr=sub.PIPE)
404            result = ssh.stdout.readlines()
405            result = filter(None, result)
406 
407 
408            # Delete all job data
409            job_data_tmp = []
410            job_data = []
411            job_data_n = []   
412            job_progress = 0
413
414            # In case of errors display error message
415            if ( len(result) < 1 ):
416                error = ssh.stderr.readlines()
417                if ( error != [] ):
418                    notify = QtGui.QMessageBox.warning(self,'Collect data',"Error. An error occured during read of job data for user " + username[h] +" for host " + description[h] + ".\n\nError message:\n" + ''.join(error))
419
420            # No error -> save job data
421            else:       
422
423                # Successively write to job_data
424                for j in range(0,len(result)):
425   
426                    job_data_tmp.append(j)
427                    job_data_tmp[j] = result[j].split(" ")
428                    job_data_tmp[j] = filter(None, job_data_tmp[j])
429 
430                    job_data.append(j)   
431                    job_data[j] = [job_data_tmp[j][0], description[h], job_data_tmp[j][2],                 
432                                   job_data_tmp[j][3], int(float(job_data_tmp[j][4])*100), job_data_tmp[j][5], 
433                                   job_data_tmp[j][1], job_data_tmp[j][6].rstrip()]
434                    job_data_n.append(j)
435                    job_data_n[j] = job_data_tmp[j][1]
436
437            del(result)
438
439
440
441            # Now read the data from the last update from file. These data are
442            # already in the prefered format
443            file2 = open(palm_dir + '/.wd.olddata', 'r')
444            result = file2.readlines()
445            file2.close()
446
447            job_data_old = []
448            job_data_old_n = []
449            k = -1
450            for j in range(0,len(result)):
451                if ( result[j].split()[1] == description[h] ):
452                    k = k + 1
453                    job_data_old.append(k)
454                    job_data_old[k] = result[j].split(" ")
455                    job_data_old[k] = filter(None, job_data_old[k])
456                    job_data_old[k] = [line.rstrip('\n') for line in job_data_old[k]]
457                    job_data_old_n.append(k)
458                    job_data_old_n[k] = job_data_old[k][6]
459
460            # Merge the job_data and job_data_old to find completed, new and
461            # still queued jobs     
462            if ( len(job_data_n) > 0 and len(job_data_old_n) > 0 ):
463               jobs_full  = list(set(job_data_old_n) | set(job_data_n))
464               jobs_known    = list(set(job_data_old_n) & set(job_data_n))
465               jobs_complete = set(job_data_old_n) - set(job_data_n)
466               jobs_new      = set(job_data_n) - set(job_data_old_n)
467            elif ( len(job_data_n) > 0 ):
468               jobs_full  = job_data_n
469               jobs_known    = []
470               jobs_new = job_data_n
471               jobs_complete = []
472            elif ( len(job_data_old_n) > 0 ):
473               jobs_full  = job_data_old_n
474               jobs_known    = []
475               jobs_new = []
476               jobs_complete = job_data_old_n   
477            else:
478               jobs_full  = []
479               jobs_known    = []         
480               jobs_new = []
481               jobs_complete = []
482
483            # Display all known jobs
484            to_display = [list(job_data_n).index(item) for item in list(jobs_known) if list(jobs_known) and list(job_data_n)]
485            for j in range(0,len(to_display)):
486                i = i + 1
487                self.table.insertRow(i)
488                item = QtGui.QTableWidgetItem(job_data[to_display[j]][0])
489                item.setToolTip("Queuing name: " + job_data[to_display[j]][6])
490                self.table.setItem(i, 0, item)           
491                self.table.setItem(i, 1, QtGui.QTableWidgetItem(job_data[to_display[j]][1])) 
492                self.table.setItem(i, 2, QtGui.QTableWidgetItem(job_data[to_display[j]][2]))   
493                self.table.setItem(i, 3, QtGui.QTableWidgetItem(job_data[to_display[j]][3]))
494                item = QtGui.QTableWidgetItem(job_data[to_display[j]][5])
495                item.setToolTip("Estimated job start: " + job_data[to_display[j]][7])               
496                self.table.setItem(i, 5, item) 
497                self.table.setItem(i, 6, QtGui.QTableWidgetItem(job_data[to_display[j]][6])) 
498                self.table.item(i,2).setTextAlignment(QtCore.Qt.AlignRight)
499                self.table.item(i,5).setTextAlignment(QtCore.Qt.AlignRight)
500                pbars.append(i)
501                pbars[i] = QtGui.QProgressBar(self)
502                pbars[i].setValue(int(job_data[to_display[j]][4])) 
503 
504                # Apply a color depending on the status of the job
505                if ( job_data[to_display[j]][3] == "Running" ):
506                    self.table.setCellWidget(i,4,pbars[i]) 
507                    self.table.item(i,0).setBackground(QtGui.QColor(255, 251, 168))
508                    self.table.item(i,1).setBackground(QtGui.QColor(255, 251, 168))
509                    self.table.item(i,2).setBackground(QtGui.QColor(255, 251, 168))
510                    self.table.item(i,3).setBackground(QtGui.QColor(255, 251, 168))
511                    self.table.item(i,5).setBackground(QtGui.QColor(255, 251, 168))
512                    self.table.item(i,6).setBackground(QtGui.QColor(255, 251, 168))
513                else:
514                    pbars[j].setEnabled(False)
515                    self.table.setCellWidget(i,4,pbars[i])
516                    self.table.item(i,0).setBackground(QtGui.QColor(255, 112, 118))
517                    self.table.item(i,1).setBackground(QtGui.QColor(255, 112, 118))
518                    self.table.item(i,2).setBackground(QtGui.QColor(255, 112, 118))
519                    self.table.item(i,3).setBackground(QtGui.QColor(255, 112, 118))
520                    self.table.item(i,5).setBackground(QtGui.QColor(255, 112, 118))
521                    self.table.item(i,6).setBackground(QtGui.QColor(255, 112, 118))
522               
523                # Save the job data to file
524                job_data_list.append(i)
525                job_data_list[i] = job_data[to_display[j]][0]
526               
527                printstring = str(job_data[to_display[j]][0]) + " " + \
528                              str(job_data[to_display[j]][1]) + " " + \
529                              str(job_data[to_display[j]][2]) + " " + \
530                              str(job_data[to_display[j]][3]) + " " + \
531                              str(job_data[to_display[j]][4]) + " " + \
532                              str(job_data[to_display[j]][5]) + " " + \
533                              str(job_data[to_display[j]][6]) + " " + \
534                              str(job_data[to_display[j]][7]) + "\n"
535                file.write(printstring)
536
537
538            # Display all new jobs
539            to_display = [list(job_data_n).index(item) for item in list(jobs_new) if list(jobs_new) and list(job_data_n)]
540            for j in range(0,len(to_display)):
541                i = i + 1
542                self.table.insertRow(i)
543                item = QtGui.QTableWidgetItem(job_data[to_display[j]][0])
544                item.setToolTip("Queuing name: " + job_data[to_display[j]][6])
545                self.table.setItem(i, 0, item)   
546                self.table.setItem(i, 1, QtGui.QTableWidgetItem(job_data[to_display[j]][1])) 
547                self.table.setItem(i, 2, QtGui.QTableWidgetItem(job_data[to_display[j]][2]))   
548                self.table.setItem(i, 3, QtGui.QTableWidgetItem(job_data[to_display[j]][3])) 
549                item = QtGui.QTableWidgetItem(job_data[to_display[j]][5])
550                item.setToolTip("Estimated job start: " + job_data[to_display[j]][7])               
551                self.table.setItem(i, 5, item) 
552                self.table.setItem(i, 6, QtGui.QTableWidgetItem(job_data[to_display[j]][6])) 
553                self.table.item(i,2).setTextAlignment(QtCore.Qt.AlignRight)
554                self.table.item(i,5).setTextAlignment(QtCore.Qt.AlignRight)
555               
556                pbars.append(i)
557                pbars[i] = QtGui.QProgressBar(self)
558                pbars[i].setValue(int(job_data[to_display[j]][4]))   
559
560                # Apply a color depending on the status of the job
561                if ( job_data[to_display[j]][3] == "Running" ):
562                    self.table.setCellWidget(i,4,pbars[i])           
563                    self.table.item(i,0).setBackground(QtGui.QColor(255, 251, 168))
564                    self.table.item(i,1).setBackground(QtGui.QColor(255, 251, 168))
565                    self.table.item(i,2).setBackground(QtGui.QColor(255, 251, 168))
566                    self.table.item(i,3).setBackground(QtGui.QColor(255, 251, 168))
567                    self.table.item(i,5).setBackground(QtGui.QColor(255, 251, 168))
568                    self.table.item(i,6).setBackground(QtGui.QColor(255, 251, 168))
569                else:
570                    pbars[j].setEnabled(False)
571                    self.table.setCellWidget(i,4,pbars[i])
572                    self.table.item(i,0).setBackground(QtGui.QColor(255, 112, 118))
573                    self.table.item(i,1).setBackground(QtGui.QColor(255, 112, 118))
574                    self.table.item(i,2).setBackground(QtGui.QColor(255, 112, 118))
575                    self.table.item(i,3).setBackground(QtGui.QColor(255, 112, 118))
576                    self.table.item(i,5).setBackground(QtGui.QColor(255, 112, 118))
577                    self.table.item(i,6).setBackground(QtGui.QColor(255, 112, 118))
578         
579                # Save job data to file
580                job_data_list.append(i)
581                job_data_list[i] = job_data[to_display[j]][0]
582               
583                printstring = str(job_data[to_display[j]][0]) + " " + \
584                              str(job_data[to_display[j]][1]) + " " + \
585                              str(job_data[to_display[j]][2]) + " " + \
586                              str(job_data[to_display[j]][3]) + " " + \
587                              str(job_data[to_display[j]][4]) + " " + \
588                              str(job_data[to_display[j]][5]) + " " + \
589                              str(job_data[to_display[j]][6]) + " " + \
590                              str(job_data[to_display[j]][7]) + "\n"
591                file.write(printstring)
592
593
594            # Display all completed/canceled/aborted jobs. The watchdog cannot
595            # distinguish why the job has been finished
596            to_display = [list(job_data_old_n).index(item) for item in list(jobs_complete) if list(jobs_complete) and list(job_data_old_n)]
597            for j in range(0,len(to_display)):
598                i = i + 1
599                self.table.insertRow(i)
600                item = QtGui.QTableWidgetItem(job_data_old[to_display[j]][0])
601                item.setToolTip("Queuing name: " + job_data_old[to_display[j]][6])
602                self.table.setItem(i, 0, item)         
603                self.table.setItem(i, 1, QtGui.QTableWidgetItem(job_data_old[to_display[j]][1])) 
604                self.table.setItem(i, 2, QtGui.QTableWidgetItem(job_data_old[to_display[j]][2]))   
605                self.table.setItem(i, 3, QtGui.QTableWidgetItem(job_data_old[to_display[j]][3])) 
606                pbars.append(i)
607                pbars[j] = QtGui.QProgressBar(self)
608                pbars[j].setValue(int(job_data_old[to_display[j]][4])) 
609                pbars[j].setEnabled(False)
610                self.table.setCellWidget(i,4,pbars[j]) 
611                item = QtGui.QTableWidgetItem(job_data_old[to_display[j]][5])
612                item.setToolTip("Estimated job start: " + job_data_old[to_display[j]][7])               
613                self.table.setItem(i, 5, item) 
614                self.table.setItem(i, 6, QtGui.QTableWidgetItem(job_data_old[to_display[j]][6])) 
615                self.table.item(i,2).setTextAlignment(QtCore.Qt.AlignRight)
616                self.table.item(i,5).setTextAlignment(QtCore.Qt.AlignRight)
617                self.table.setItem(i, 3, QtGui.QTableWidgetItem("Completed"))
618               
619                # Apply a color depending on the status of the job
620                self.table.item(i,0).setBackground(QtGui.QColor(172, 252, 175))
621                self.table.item(i,1).setBackground(QtGui.QColor(172, 252, 175))
622                self.table.item(i,2).setBackground(QtGui.QColor(172, 252, 175))
623                self.table.item(i,3).setBackground(QtGui.QColor(172, 252, 175))
624                self.table.item(i,5).setBackground(QtGui.QColor(172, 252, 175))
625                self.table.item(i,6).setBackground(QtGui.QColor(172, 252, 175))
626
627         
628                # Save job data to file
629                job_data_list.append(i)
630                job_data_list[i] = job_data_old[to_display[j]][0]
631
632                printstring = str(job_data_old[to_display[j]][0]) + " " + \
633                              str(job_data_old[to_display[j]][1]) + " " + \
634                              str(job_data_old[to_display[j]][2]) + " " + \
635                              str(job_data_old[to_display[j]][3]) + " " + \
636                              str(job_data_old[to_display[j]][4]) + " " + \
637                              str(job_data_old[to_display[j]][5]) + " " + \
638                              str(job_data_old[to_display[j]][6]) + " " + \
639                              str(job_data_old[to_display[j]][7]) + "\n"
640                file.write(printstring)
641
642            del(jobs_complete)
643            del(jobs_new)
644            del(result)
645            del(job_data_old)
646            del(job_data_old_n)
647           
648        file.close()
649
650       
651        # Update is complete, sorting can thus be re-enabled               
652        self.table.setSortingEnabled(True) 
653
654        # Update the logfile
655        if ( os.path.isfile(palm_dir + '/.wd.newdata') ):
656            shutil.copy(palm_dir + '/.wd.newdata',palm_dir + '/.wd.olddata')
657
658        # Re-enable timer processes             
659        self.setEnabled(True)
660        self.label2.setText("Last update: " + timestamp)
661        self.label.setText("Update complete.")
662        self.timetimer.start()
663        self.timer.start(update_frequency)
664        self.labeltimer.start(update_frequency/10)
665        self.UpdateLabel()
666        QtGui.QApplication.processEvents()
667       
668    # Canceling selected job from table
669    def CancelJob(self):
670
671        # Return internal jobname
672        jobname = self.table.item(self.table.currentRow(),6).text()
673        jobrealname = self.table.item(self.table.currentRow(),0).text()     
674 
675        # Check description of job in order to login on the correct host
676        descr = self.table.item(self.table.currentRow(),1).text()
677       
678        querybox = QtGui.QMessageBox()
679        querybox.setText("Attention!\nYou are trying to cancel a job. It will not be able to terminate properly.")
680        querybox.setInformativeText("Do you really want to cancel " + jobname + " on host " + descr + "?")
681        querybox.setStandardButtons(QtGui.QMessageBox.Yes | QtGui.QMessageBox.No)
682        querybox.setDefaultButton(QtGui.QMessageBox.No)
683        returnvalue = querybox.exec_()
684       
685        if ( returnvalue == QtGui.QMessageBox.Yes ):
686       
687           for h in range(0,len(description)):
688               if ( descr == description[h] ):
689                  host = username[h] + "@" + hostname[h]
690
691           ssh = sub.Popen(["ssh", "%s" % host, "/sw/tools/python/2.7.6/generic/bin/python palm_wdd cancel " + jobname],
692                              shell=False,
693                              stdout=sub.PIPE,
694                              stderr=sub.PIPE)
695           result = ssh.stdout.readlines()
696           result = filter(None, result)
697       
698           # In case of error display a warning message
699           if ( len(result) == 0 ):
700               error = ssh.stderr.readlines()
701               notify = QtGui.QMessageBox.warning(self,'Cancel job',"Error. Could not cancel job " + jobname + ".\n\nError message:\n" + ''.join(error))
702
703           # Otherwise inform the user and mark the job in the table
704           else:
705               self.table.setItem(self.table.currentRow(),3,QtGui.QTableWidgetItem("Canceled"))
706               notify = QtGui.QMessageBox.information(self,'Cancel job',"Job" + jobname + " canceled!\n\nServer message:\n" + ''.join(result))
707
708
709    # Show detailed information on job. For documentation see above
710    def ShowDetails(self):
711
712        jobname = self.table.item(self.table.currentRow(),6).text() 
713        descr   = self.table.item(self.table.currentRow(),1).text()
714        for h in range(0,len(description)):
715            if ( descr == description[h] ):
716                host = username[h] + "@" + hostname[h]
717
718        ssh = sub.Popen(["ssh", "%s" % host, "/sw/tools/python/2.7.6/generic/bin/python palm_wdd check " + jobname],
719                           shell=False,
720                           stdout=sub.PIPE,
721                           stderr=sub.PIPE)
722        result = ssh.stdout.readlines()
723        result = filter(None, result)
724
725        if ( len(result) == 0 ):
726            error = ssh.stderr.readlines()
727            notify = QtGui.QMessageBox.warning(self,'Show job details',"Error. Could not fetch job details for " + jobname + ".\n\nError message:\n" + ''.join(error))
728        else:
729            notify = QtGui.QMessageBox.information(self,'Job details',"Details for job " + jobname + ":\n\n" + ''.join(result))
730
731
732    # Perform a forced stop on the job
733    def DoStop(self):
734
735        # Return internal jobname
736        jobname = self.table.item(self.table.currentRow(),6).text()
737        jobrealname = self.table.item(self.table.currentRow(),0).text()   
738        # Check description of job in order to login on the correct host
739        descr = self.table.item(self.table.currentRow(),1).text()
740        for h in range(0,len(description)):
741            if ( descr == description[h] ):
742                host = username[h] + "@" + hostname[h]
743                user = username[h]
744
745        ssh = sub.Popen(["ssh", "%s" % host, "/sw/tools/python/2.7.6/generic/bin/python palm_wdd stop " + jobrealname],
746                           shell=False,
747                           stdout=sub.PIPE,
748                           stderr=sub.PIPE)
749        result = ssh.stdout.readlines()
750        result = filter(None, result) 
751       
752        # In case of error display a warning message
753        if ( len(result) == 0 ):
754            error = ssh.stderr.readlines()
755            notify = QtGui.QMessageBox.warning(self,'Proper termination of job',"Error. Could not stop job " + jobname + ".\n\nError message:\n" + ''.join(error))
756
757        # Otherwise inform the user and mark the job in the table
758        else:
759            self.table.setItem(self.table.currentRow(),3,QtGui.QTableWidgetItem("Terminated"))
760            notify = QtGui.QMessageBox.information(self,'Terminate job',"Termination of job " + jobname + " was initiated!\n\nServer message:\n" + ''.join(result))
761
762
763    # Perform a forced stop on the job
764    def DoRestart(self):
765
766        # Return internal jobname
767        jobname = self.table.item(self.table.currentRow(),6).text()
768        jobrealname = self.table.item(self.table.currentRow(),0).text()   
769         
770        # Check description of job in order to login on the correct host
771        descr = self.table.item(self.table.currentRow(),1).text()
772        for h in range(0,len(description)):
773            if ( descr == description[h] ):
774                host = username[h] + "@" + hostname[h]
775                user = username[h]
776
777        ssh = sub.Popen(["ssh", "%s" % host, "/sw/tools/python/2.7.6/generic/bin/python palm_wdd restart " + jobrealname],
778                           shell=False,
779                           stdout=sub.PIPE,
780                           stderr=sub.PIPE)
781        result = ssh.stdout.readlines()
782        result = filter(None, result)
783       
784        # In case of error display a warning message
785        if ( len(result) == 0 ):
786            error = ssh.stderr.readlines()
787            notify = QtGui.QMessageBox.warning(self,'Proper termination of job',"Error. Could not stop job " + jobname + ".\n\nError message:\n" + ''.join(error))
788
789        # Otherwise inform the user and mark the job in the table
790        else:
791            self.table.setItem(self.table.currentRow(),3,QtGui.QTableWidgetItem("Terminated"))
792            notify = QtGui.QMessageBox.information(self,'Terminate job for restart',"Restart for job" + jobname + " was initiated!\n\nServer message:\n" + ''.join(result))
793           
794
795    # Read the PARIN file of the job
796    def ShowPARIN(self):
797
798        # Return internal jobname
799        jobname = self.table.item(self.table.currentRow(),6).text()
800        jobrealname = self.table.item(self.table.currentRow(),0).text()     
801 
802        # Check description of job in order to login on the correct host
803        descr = self.table.item(self.table.currentRow(),1).text()
804        for h in range(0,len(description)):
805            if ( descr == description[h] ):
806                host = username[h] + "@" + hostname[h]
807                user = username[h]
808
809        ssh = sub.Popen(["ssh", "%s" % host, "/sw/tools/python/2.7.6/generic/bin/python palm_wdd parin " + jobrealname],
810                           shell=False,
811                           stdout=sub.PIPE,
812                           stderr=sub.PIPE)
813        result = ssh.stdout.readlines()
814        result = filter(None, result)
815       
816        # In case of error display a warning message
817        if ( len(result) == 0 ):
818            error = ssh.stderr.readlines()
819            notify = QtGui.QMessageBox.warning(self,'Showing parameter file',"Error. Could not fetch parameter file for job " + jobrealname + " (" + jobname + ").\n\nError message:\n" + ''.join(error))
820
821        # Otherwise inform the user and mark the job in the table
822        else:
823           mb = MessageBoxScroll()
824           mb.setText("Parameter file for job: " + jobrealname + "")
825           mb.setDetailedText(''.join(result))
826           mb.exec_()
827
828    # Read the PARIN file of the job
829    def ShowRC(self):
830
831        # Return internal jobname and real job name
832        jobname = self.table.item(self.table.currentRow(),6).text()
833        jobrealname = self.table.item(self.table.currentRow(),0).text()     
834 
835        # Check description of job in order to login on the correct host
836        descr = self.table.item(self.table.currentRow(),1).text()
837        for h in range(0,len(description)):
838            if ( descr == description[h] ):
839                host = username[h] + "@" + hostname[h]
840                user = username[h]
841
842        ssh = sub.Popen(["ssh", "%s" % host, "/sw/tools/python/2.7.6/generic/bin/python palm_wdd rc " + jobrealname],
843                           shell=False,
844                           stdout=sub.PIPE,
845                           stderr=sub.PIPE)
846        result = ssh.stdout.readlines()
847        result = filter(None, result)
848       
849        # In case of error display a warning message
850        if ( len(result) == 0 ):
851            error = ssh.stderr.readlines()
852            notify = QtGui.QMessageBox.warning(self,'Showing run control',"Error. Could not fetch run control file for job " + jobrealname + "(" + jobname + ").\n\nError message:\n" + ''.join(error))
853
854        # Otherwise inform the user and mark the job in the table
855        else:
856           mb = MessageBoxScroll()
857           lastline = result[len(result)-2].split()[2]
858           mb.setText("Simulated time for job " + jobrealname + " is currently: " + lastline)
859           mb.setDetailedText(''.join(result))
860           mb.exec_()
861
862    # Remove a job from list - removes the current row from the table and from
863    # save file
864    def RemoveFromList(self, row):
865        if ( row == -1 ):
866            row = self.table.currentRow()
867
868        # Read data from save file
869        job_to_delete = self.table.item(row,6).text()
870        self.table.removeRow(row)
871        file = open(palm_dir + '/.wd.olddata', 'r')
872        result = file.readlines()
873        result = filter(None, result)
874        file.close()
875        file = open(palm_dir + '/.wd.olddata', 'w')
876       
877        job_data_old = []
878       
879        if ( len(result) == 0 ):
880            notify = QtGui.QMessageBox.warning(self,'Read from .wd.olddata',"Error message:\n\nCould not read from file. I/O error.")
881        else: 
882            # Save data in array job_data_old
883            for j in range(0,len(result)):
884                job_data_old.append(j)
885                job_data_old[j] = result[j].split(" ")
886                job_data_old[j] = filter(None, job_data_old[j])
887                job_data_old[j] = [line.rstrip('\n') for line in job_data_old[j]]
888               
889                # Check if line j is the selected job, if not -> save to file
890                if ( job_data_old[j][6] != job_to_delete ):
891                    printstring = str(job_data_old[j][0]) + " " + \
892                                  str(job_data_old[j][1]) + " " + \
893                                  str(job_data_old[j][2]) + " " + \
894                                  str(job_data_old[j][3]) + " " + \
895                                  str(job_data_old[j][4]) + " " + \
896                                  str(job_data_old[j][5]) + " " + \
897                                  str(job_data_old[j][6]) + " " + \
898                                  str(job_data_old[j][7]) + "\n"
899                    file.write(printstring)
900           
901        file.close() 
902
903    # Remove all completed jobs from list
904    def ClearList(self):
905     
906       num_of_lines = self.table.rowCount()
907       
908       # Delete all lines with completed/canceled jobs from list. The counter
909       # must decrease as the line numbers would be messed up otherwise
910       for j in range(num_of_lines-1,-1,-1): 
911           state = self.table.item(j,3).text()
912           if ( state == "Completed" or state == "Canceled" or state == "Terminated"):
913              self.RemoveFromList(j)
914         
915    # Update the label
916    def UpdateLabel(self):
917        remaining_time = (update_frequency - self.timetimer.elapsed()) / 1000 / 60
918        self.label.setText("Next update in " + str(remaining_time) + " min")
919       
920
921    # Enter Options menu
922    def Options(self):
923 
924#       disable mainwindow 
925        self.setEnabled(False)
926     
927#       show Options Dialog     
928        opt = OptionBox()
929        opt.exec_()
930
931        self.UpdateHosts()
932        self.setEnabled(True)
933
934# Update settings
935    def UpdateHosts(self):
936
937       global update_frequency
938       global description
939       global hostname
940       global username
941       
942       description = []
943       hostname = []
944       username = []
945
946       for i in range(0,len(config.sections())):
947
948          description_tmp = config.sections()[i]
949 
950          if ( description_tmp != 'Settings' ):
951     
952             if ( config.get(description_tmp, 'enabled') == 'true' ):
953                description.append(description_tmp)
954                hostname.append(config.get(description_tmp, 'hostname'))
955                username.append(config.get(description_tmp, 'username')) 
956          else:
957             update_frequency = int(config.get(description_tmp, 'update_frequency'))*60000
958
959
960# Main loop       
961def main():
962   
963    app = QtGui.QApplication(sys.argv)
964    res = Watchdog() 
965    sys.exit(app.exec_())
966
967
968if __name__ == '__main__':
969    main()   
Note: See TracBrowser for help on using the repository browser.