source: palm/trunk/SCRIPTS/palm_wd @ 3248

Last change on this file since 3248 was 2825, checked in by maronga, 4 years ago

adjustments in gui tools

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