#!/usr/bin/python # -*- coding: utf-8 -*- #--------------------------------------------------------------------------------# # This file is part of PALM. # # PALM is free software: you can redistribute it and/or modify it under the terms # of the GNU General Public License as published by the Free Software Foundation, # either version 3 of the License, or (at your option) any later version. # # PALM is distributed in the hope that it will be useful, but WITHOUT ANY # WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR # A PARTICULAR PURPOSE. See the GNU General Public License for more details. # # You should have received a copy of the GNU General Public License along with # PALM. If not, see . # # Copyright 1997-2016 Leibniz Universitaet Hannover #--------------------------------------------------------------------------------# # # Current revisions: # ----------------- # # # Former revisions: # ----------------- # $Id: palm_jm 2068 2016-11-11 21:59:39Z sward $ # # 2067 2016-11-11 21:55:49Z maronga # Minor bugfix: palm_jm can now be called from any location # # 2065 2016-11-11 12:25:10Z maronga # Initial revision # # # Description: # ------------ # PALM job manager works a file manager for PALM input files. It lists all jobs in # the working directory and displays associated input and user code files, which # can be opened directly from the interface. Also, it is possible to copy jobs and # to create entire new sets of jobs based on a reference job. # # Instructions: # ------------- # 1) Start palm_jm either from mrungui or from shell by "nohup palm_jm&" # # To do: # ------ # 1) Add display for the MONITORING and OUTPUT folder # 2) Add zip function to export single or multiple jobs # 3) Allow for copying INPUT files to a remote host #------------------------------------------------------------------------------! import os from PyQt4 import QtGui, QtCore, uic from PyQt4.QtCore import pyqtSlot,SIGNAL,SLOT import shutil import subprocess as sub import sys set_list = [] # Determine PALM directories try: devnull = open(os.devnull, 'w') out = sub.check_output("echo $PALM_BIN", shell=True, stderr=sub.STDOUT) palm_bin = out.rstrip() palm_dir = out.split("palm")[0] + "palm/" + out.split("palm")[1].split("/")[1] out = None # remove this line on release # palm_dir = '/home/maronga/PUB/thinktank/maronga/job_manager' job_dir = palm_dir + '/JOBS' user_dir = palm_dir + '/USER_CODE' except: print "Error. $PALM_BIN is not set." raise SystemExit # MainWindow class class JobManager(QtGui.QMainWindow): def __init__(self): super(JobManager, self).__init__() self.InitUi() # Initialized MainWindow UI def InitUi(self): # Load predefined mainwindow uic.loadUi(palm_bin + '/palm_jm_files/palm_jm.ui', self) # Display MainWindow self.show() QtGui.QApplication.processEvents() self.load_jobs() def update_all(self): self.setEnabled(False) self.list_input.clear() self.list_user.clear() self.load_jobs() self.setEnabled(True) # Load jobs into list def load_jobs(self): self.list_jobs.clear() self.push_copy.setEnabled(False) self.push_create_set.setEnabled(False) self.line_path.setText(job_dir + "/") list_of_files = os.listdir(job_dir) for i in range(0,len(list_of_files)): tmp_file = job_dir + "/" + list_of_files[i] if ( os.path.isdir(tmp_file) ): self.list_jobs.addItem(str(list_of_files[i])) # Update input and user code lists def update_input(self): self.list_input.clear() self.list_user.clear() self.push_copy.setEnabled(True) self.push_create_set.setEnabled(True) job_to_show = job_dir + "/" + self.list_jobs.currentItem().text() + "/INPUT" if ( os.path.isdir(job_to_show) ): list_of_files = os.listdir(job_to_show) for i in range(0,len(list_of_files)): tmp_file = job_to_show + "/" + list_of_files[i] if ( os.path.isfile(tmp_file) ): self.list_input.addItem(str(list_of_files[i])) job_to_show = user_dir + "/" + self.list_jobs.currentItem().text() if ( os.path.isdir(job_to_show) ): list_of_files = os.listdir(job_to_show) for i in range(0,len(list_of_files)): tmp_file = job_to_show + "/" + list_of_files[i] if ( os.path.isfile(tmp_file) ): self.list_user.addItem(str(list_of_files[i])) # Make a copy of a job def copy_job(self): self.setEnabled(False) old_job_name = self.list_jobs.currentItem().text() text, ret = QtGui.QInputDialog.getText(self, "Copy job", "Enter new job name:", mode = QtGui.QLineEdit.Normal, text = old_job_name) if ( ret ): new_job_name = str(text) else: self.setEnabled(True) return new_input_dir = job_dir + "/" + new_job_name + "/INPUT" # check if a job exists with the new job name if ( os.path.isdir(new_input_dir) ): notify = QtGui.QMessageBox.warning(self,'Create new job directory',"Error. Could not create job directory. A job with the new name already exists.") self.setEnabled(True) return else: os.makedirs(new_input_dir) # copy and rename input files (if present) job_to_copy = job_dir + "/" + old_job_name + "/INPUT" if ( os.path.isdir(job_to_copy) ): list_of_files = os.listdir(job_to_copy) for i in range(0,len(list_of_files)): tmp_file = job_to_copy + "/" + list_of_files[i] new_file = new_input_dir + "/" + list_of_files[i].replace(old_job_name, new_job_name) shutil.copy(tmp_file, new_file) # copy and rename user code (if present) user_code_to_copy = user_dir + "/" + old_job_name new_user_code_dir = user_dir + "/" + new_job_name # check if a job exists with the new job name if ( os.path.isdir(new_user_code_dir) ): notify = QtGui.QMessageBox.warning(self,'Create new user code directory',"Error. Could not create user code directory. A user code directory with the new name already exists.") self.setEnabled(True) return else: os.makedirs(new_user_code_dir) if ( os.path.isdir(user_code_to_copy) ): list_of_files = os.listdir(user_code_to_copy) for i in range(0,len(list_of_files)): tmp_file = user_code_to_copy + "/" + list_of_files[i] new_file = new_user_code_dir + "/" + list_of_files[i] shutil.copy(tmp_file, new_file) self.load_jobs() self.list_input.clear() self.list_user.clear() self.setEnabled(True) # Create a whole set of jobs def create_set(self): global set_list # disable mainwindow self.setEnabled(False) # show Options Dialog opt = CreateSetBox() opt.exec_() old_job_name = self.list_jobs.currentItem().text() for j in range(0,len(set_list)): if ( set_list[j] != "" ): new_job_name = str(set_list[j]) new_input_dir = job_dir + "/" + str(set_list[j]) + "/INPUT" else: continue # check if a job exists with the new job name if ( os.path.isdir(new_input_dir) ): notify = QtGui.QMessageBox.warning(self,'Create new job directory',"Error. Could not create job directory. A job with the new name already exists.") self.setEnabled(True) return else: os.makedirs(new_input_dir) # copy and rename input files (if present) job_to_copy = job_dir + "/" + old_job_name + "/INPUT" if ( os.path.isdir(job_to_copy) ): list_of_files = os.listdir(job_to_copy) for i in range(0,len(list_of_files)): tmp_file = job_to_copy + "/" + list_of_files[i] new_file = new_input_dir + "/" + list_of_files[i].replace(old_job_name, new_job_name) shutil.copy(tmp_file, new_file) # copy and rename user code (if present) user_code_to_copy = user_dir + "/" + old_job_name new_user_code_dir = user_dir + "/" + new_job_name # check if a job exists with the new job name if ( os.path.isdir(new_user_code_dir) ): notify = QtGui.QMessageBox.warning(self,'Create new user code directory',"Error. Could not create user code directory. A user code directory with the new name already exists.") self.setEnabled(True) return else: os.makedirs(new_user_code_dir) if ( os.path.isdir(user_code_to_copy) ): list_of_files = os.listdir(user_code_to_copy) for i in range(0,len(list_of_files)): tmp_file = user_code_to_copy + "/" + list_of_files[i] new_file = new_user_code_dir + "/" + list_of_files[i] shutil.copy(tmp_file, new_file) self.load_jobs() self.list_input.clear() self.list_user.clear() self.setEnabled(True) # Add a custom context menu def openmenuinput(self, position): menu = QtGui.QMenu() selection = self.list_input.selectedItems() if ( len(selection) != 0 ): openAction = menu.addAction('Open selected files') openAction.setStatusTip('Open file(s) in your favorite editor') openAction.triggered.connect(self.OpenFilesInput) action = menu.exec_(self.list_input.mapToGlobal(position)) # Add a custom context menu def openmenuuser(self, position): menu = QtGui.QMenu() selection = self.list_user.selectedItems() if ( len(selection) != 0 ): openAction = menu.addAction('Open selected files') openAction.setStatusTip('Open file(s) in your favorite editor') openAction.triggered.connect(self.OpenFilesUser) action = menu.exec_(self.list_user.mapToGlobal(position)) def OpenFilesInput(self): sel_job = self.list_jobs.currentItem().text() sel_files = self.list_input.selectedItems() input_dir = job_dir + "/" + sel_job + "/INPUT/" open_files = "" for i in range(0,len(sel_files)): open_files = open_files + "xdg-open " + input_dir + sel_files[i].text() + "; " os.system(str(open_files)) def OpenFilesUser(self): sel_job = self.list_jobs.currentItem().text() sel_files = self.list_user.selectedItems() input_dir = user_dir + "/" + sel_job + "/" open_files = "" for i in range(0,len(sel_files)): open_files = open_files + "xdg-open " + input_dir + sel_files[i].text() + "; " os.system(str(open_files)) selection = self.list_jobs.selectedItems() # Message box for showing RUN_CONTROL output class CreateSetBox(QtGui.QDialog): def __init__(self): super(CreateSetBox, self).__init__() uic.loadUi(palm_bin + '/palm_jm_files/palm_jm_create_set.ui', self) self.show() return # Cancel button def rejected(self): self.close() return # OK button def accept(self): global set_list text = self.list.toPlainText() set_list = text.split('\n') self.close() return # Main loop def main(): app = QtGui.QApplication(sys.argv) res = JobManager() sys.exit(app.exec_()) if __name__ == '__main__': main()