source: palm/trunk/SCRIPTS/palm_wd @ 1729

Last change on this file since 1729 was 1619, checked in by maronga, 9 years ago

last commit documented

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