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-2016 Leibniz Universitaet Hannover |
---|
18 | ##--------------------------------------------------------------------------------# |
---|
19 | # |
---|
20 | # Current revisions: |
---|
21 | # ----------------- |
---|
22 | # |
---|
23 | # |
---|
24 | # Former revisions: |
---|
25 | # ----------------- |
---|
26 | # $Id: palm_jm 2504 2017-09-27 10:36:13Z basit $ |
---|
27 | # Bugfix: jobs were not loaded on start. |
---|
28 | # |
---|
29 | # 2484 2017-09-20 14:22:42Z maronga |
---|
30 | # Adapted for palmrun with USER_CODE always stored under /JOBS/name/USER_CODE/. |
---|
31 | # Added PALM logo |
---|
32 | # |
---|
33 | # 2068 2016-11-11 21:59:39Z maronga |
---|
34 | # |
---|
35 | # 2067 2016-11-11 21:55:49Z maronga |
---|
36 | # Minor bugfix: palm_jm can now be called from any location |
---|
37 | # |
---|
38 | # 2065 2016-11-11 12:25:10Z maronga |
---|
39 | # Initial revision |
---|
40 | # |
---|
41 | # |
---|
42 | # Description: |
---|
43 | # ------------ |
---|
44 | # PALM job manager works a file manager for PALM input files. It lists all jobs in |
---|
45 | # the working directory and displays associated input and user code files, which |
---|
46 | # can be opened directly from the interface. Also, it is possible to copy jobs and |
---|
47 | # to create entire new sets of jobs based on a reference job. |
---|
48 | # |
---|
49 | # Instructions: |
---|
50 | # ------------- |
---|
51 | # 1) Start palm_jm either from mrungui or from shell by "nohup palm_jm&" |
---|
52 | # |
---|
53 | # To do: |
---|
54 | # ------ |
---|
55 | # 1) Add display for the MONITORING and OUTPUT folder |
---|
56 | # 2) Add zip function to export single or multiple jobs |
---|
57 | # 3) Allow for copying INPUT files to a remote host |
---|
58 | #------------------------------------------------------------------------------! |
---|
59 | |
---|
60 | import os |
---|
61 | from PyQt4 import QtGui, QtCore, uic |
---|
62 | from PyQt4.QtCore import pyqtSlot,SIGNAL,SLOT |
---|
63 | import shutil |
---|
64 | import subprocess as sub |
---|
65 | import sys |
---|
66 | |
---|
67 | set_list = [] |
---|
68 | |
---|
69 | |
---|
70 | # Determine PALM directories |
---|
71 | try: |
---|
72 | devnull = open(os.devnull, 'w') |
---|
73 | out = sub.check_output("echo $PALM_BIN", shell=True, stderr=sub.STDOUT) |
---|
74 | palm_bin = out.rstrip() |
---|
75 | palm_dir = out.split("palm")[0] + "palm/" + out.split("palm")[1].split("/")[1] |
---|
76 | out = None |
---|
77 | |
---|
78 | job_dir = palm_dir + '/JOBS' |
---|
79 | user_dir = palm_dir + '/USER_CODE' |
---|
80 | |
---|
81 | except: |
---|
82 | print "Error. $PALM_BIN is not set." |
---|
83 | raise SystemExit |
---|
84 | |
---|
85 | |
---|
86 | |
---|
87 | # MainWindow class |
---|
88 | class JobManager(QtGui.QMainWindow): |
---|
89 | |
---|
90 | def __init__(self): |
---|
91 | super(JobManager, self).__init__() |
---|
92 | |
---|
93 | self.InitUi() |
---|
94 | |
---|
95 | |
---|
96 | # Initialized MainWindow UI |
---|
97 | def InitUi(self): |
---|
98 | |
---|
99 | # Load predefined mainwindow |
---|
100 | uic.loadUi(palm_bin + '/palm_jm_files/palm_jm.ui', self) |
---|
101 | |
---|
102 | self.palm_logo.setPixmap(QtGui.QPixmap(palm_dir + "/trunk/SCRIPTS/palmrungui_files/logo.png")) |
---|
103 | |
---|
104 | |
---|
105 | |
---|
106 | # Display MainWindow |
---|
107 | self.show() |
---|
108 | QtGui.QApplication.processEvents() |
---|
109 | |
---|
110 | self.load_jobs() |
---|
111 | |
---|
112 | |
---|
113 | def update_all(self): |
---|
114 | |
---|
115 | self.setEnabled(False) |
---|
116 | self.list_input.clear() |
---|
117 | self.list_user.clear() |
---|
118 | self.load_jobs() |
---|
119 | self.setEnabled(True) |
---|
120 | |
---|
121 | # Load jobs into list |
---|
122 | def load_jobs(self): |
---|
123 | |
---|
124 | self.list_jobs.clear() |
---|
125 | self.push_copy.setEnabled(False) |
---|
126 | self.push_create_set.setEnabled(False) |
---|
127 | |
---|
128 | self.line_path.setText(job_dir + "/") |
---|
129 | |
---|
130 | list_of_files = os.listdir(job_dir) |
---|
131 | |
---|
132 | for i in range(0,len(list_of_files)): |
---|
133 | tmp_file = job_dir + "/" + list_of_files[i] |
---|
134 | |
---|
135 | if ( os.path.isdir(tmp_file) ): |
---|
136 | self.list_jobs.addItem(str(list_of_files[i])) |
---|
137 | |
---|
138 | |
---|
139 | # Update input and user code lists |
---|
140 | def update_input(self): |
---|
141 | |
---|
142 | self.list_input.clear() |
---|
143 | self.list_user.clear() |
---|
144 | self.push_copy.setEnabled(True) |
---|
145 | self.push_create_set.setEnabled(True) |
---|
146 | |
---|
147 | job_to_show = job_dir + "/" + self.list_jobs.currentItem().text() + "/INPUT" |
---|
148 | |
---|
149 | if ( os.path.isdir(job_to_show) ): |
---|
150 | |
---|
151 | list_of_files = os.listdir(job_to_show) |
---|
152 | |
---|
153 | for i in range(0,len(list_of_files)): |
---|
154 | tmp_file = job_to_show + "/" + list_of_files[i] |
---|
155 | |
---|
156 | if ( os.path.isfile(tmp_file) ): |
---|
157 | self.list_input.addItem(str(list_of_files[i])) |
---|
158 | |
---|
159 | job_to_show = job_dir + "/" + self.list_jobs.currentItem().text() + "/USER_CODE" |
---|
160 | |
---|
161 | if ( os.path.isdir(job_to_show) ): |
---|
162 | |
---|
163 | list_of_files = os.listdir(job_to_show) |
---|
164 | |
---|
165 | for i in range(0,len(list_of_files)): |
---|
166 | tmp_file = job_to_show + "/" + list_of_files[i] |
---|
167 | |
---|
168 | if ( os.path.isfile(tmp_file) ): |
---|
169 | self.list_user.addItem(str(list_of_files[i])) |
---|
170 | |
---|
171 | |
---|
172 | # Make a copy of a job |
---|
173 | def copy_job(self): |
---|
174 | |
---|
175 | self.setEnabled(False) |
---|
176 | old_job_name = self.list_jobs.currentItem().text() |
---|
177 | |
---|
178 | text, ret = QtGui.QInputDialog.getText(self, "Copy job", "Enter new job name:", mode = QtGui.QLineEdit.Normal, text = old_job_name) |
---|
179 | |
---|
180 | if ( ret ): |
---|
181 | new_job_name = str(text) |
---|
182 | else: |
---|
183 | self.setEnabled(True) |
---|
184 | return |
---|
185 | |
---|
186 | new_input_dir = job_dir + "/" + new_job_name + "/INPUT" |
---|
187 | |
---|
188 | # check if a job exists with the new job name |
---|
189 | if ( os.path.isdir(new_input_dir) ): |
---|
190 | notify = QtGui.QMessageBox.warning(self,'Create new job directory',"Error. Could not create job directory. A job with the new name already exists.") |
---|
191 | self.setEnabled(True) |
---|
192 | return |
---|
193 | else: |
---|
194 | os.makedirs(new_input_dir) |
---|
195 | |
---|
196 | # copy and rename input files (if present) |
---|
197 | job_to_copy = job_dir + "/" + old_job_name + "/INPUT" |
---|
198 | |
---|
199 | if ( os.path.isdir(job_to_copy) ): |
---|
200 | |
---|
201 | list_of_files = os.listdir(job_to_copy) |
---|
202 | |
---|
203 | for i in range(0,len(list_of_files)): |
---|
204 | |
---|
205 | tmp_file = job_to_copy + "/" + list_of_files[i] |
---|
206 | new_file = new_input_dir + "/" + list_of_files[i].replace(old_job_name, new_job_name) |
---|
207 | shutil.copy(tmp_file, new_file) |
---|
208 | |
---|
209 | |
---|
210 | |
---|
211 | |
---|
212 | new_user_dir = job_dir + "/" + new_job_name + "/USER_CODE" |
---|
213 | |
---|
214 | # check if user code exists in the new job directory |
---|
215 | if ( os.path.isdir(new_user_dir) ): |
---|
216 | notify = QtGui.QMessageBox.warning(self,'Create new user code directory',"Error. Could not create user code directory. A user code directiory with the new name already exists.") |
---|
217 | self.setEnabled(True) |
---|
218 | return |
---|
219 | else: |
---|
220 | os.makedirs(new_user_dir) |
---|
221 | |
---|
222 | |
---|
223 | # copy user code files (if present) |
---|
224 | user_to_copy = job_dir + "/" + old_job_name + "/USER_CODE" |
---|
225 | |
---|
226 | if ( os.path.isdir(user_to_copy) ): |
---|
227 | |
---|
228 | list_of_files = os.listdir(user_to_copy) |
---|
229 | |
---|
230 | for i in range(0,len(list_of_files)): |
---|
231 | |
---|
232 | tmp_file = user_to_copy + "/" + list_of_files[i] |
---|
233 | new_file = new_user_dir + "/" + list_of_files[i] |
---|
234 | shutil.copy(tmp_file, new_file) |
---|
235 | |
---|
236 | self.load_jobs() |
---|
237 | self.list_input.clear() |
---|
238 | self.list_user.clear() |
---|
239 | self.setEnabled(True) |
---|
240 | |
---|
241 | |
---|
242 | # Create a whole set of jobs |
---|
243 | def create_set(self): |
---|
244 | |
---|
245 | global set_list |
---|
246 | # disable mainwindow |
---|
247 | self.setEnabled(False) |
---|
248 | |
---|
249 | # show Options Dialog |
---|
250 | opt = CreateSetBox() |
---|
251 | opt.exec_() |
---|
252 | |
---|
253 | old_job_name = self.list_jobs.currentItem().text() |
---|
254 | |
---|
255 | for j in range(0,len(set_list)): |
---|
256 | |
---|
257 | if ( set_list[j] != "" ): |
---|
258 | new_job_name = str(set_list[j]) |
---|
259 | new_input_dir = job_dir + "/" + str(set_list[j]) + "/INPUT" |
---|
260 | else: |
---|
261 | continue |
---|
262 | |
---|
263 | # check if a job exists with the new job name |
---|
264 | if ( os.path.isdir(new_input_dir) ): |
---|
265 | notify = QtGui.QMessageBox.warning(self,'Create new job directory',"Error. Could not create job directory. A job with the new name already exists.") |
---|
266 | self.setEnabled(True) |
---|
267 | return |
---|
268 | else: |
---|
269 | os.makedirs(new_input_dir) |
---|
270 | |
---|
271 | # copy and rename input files (if present) |
---|
272 | job_to_copy = job_dir + "/" + old_job_name + "/INPUT" |
---|
273 | |
---|
274 | if ( os.path.isdir(job_to_copy) ): |
---|
275 | |
---|
276 | list_of_files = os.listdir(job_to_copy) |
---|
277 | |
---|
278 | for i in range(0,len(list_of_files)): |
---|
279 | |
---|
280 | tmp_file = job_to_copy + "/" + list_of_files[i] |
---|
281 | new_file = new_input_dir + "/" + list_of_files[i].replace(old_job_name, new_job_name) |
---|
282 | shutil.copy(tmp_file, new_file) |
---|
283 | |
---|
284 | |
---|
285 | new_user_dir = job_dir + "/" + new_job_name + "/USER_CODE" |
---|
286 | |
---|
287 | # check if user code exists in the new job directory |
---|
288 | if ( os.path.isdir(new_user_dir) ): |
---|
289 | notify = QtGui.QMessageBox.warning(self,'Create new user code directory',"Error. Could not create user code directory. A user code directiory with the new name already exists.") |
---|
290 | self.setEnabled(True) |
---|
291 | return |
---|
292 | else: |
---|
293 | os.makedirs(new_user_dir) |
---|
294 | |
---|
295 | |
---|
296 | # copy user code files (if present) |
---|
297 | user_to_copy = job_dir + "/" + old_job_name + "/USER_CODE" |
---|
298 | |
---|
299 | if ( os.path.isdir(user_to_copy) ): |
---|
300 | |
---|
301 | list_of_files = os.listdir(user_to_copy) |
---|
302 | |
---|
303 | for i in range(0,len(list_of_files)): |
---|
304 | |
---|
305 | tmp_file = user_to_copy + "/" + list_of_files[i] |
---|
306 | new_file = new_user_dir + "/" + list_of_files[i] |
---|
307 | shutil.copy(tmp_file, new_file) |
---|
308 | |
---|
309 | self.load_jobs() |
---|
310 | self.list_input.clear() |
---|
311 | self.list_user.clear() |
---|
312 | |
---|
313 | self.setEnabled(True) |
---|
314 | |
---|
315 | |
---|
316 | # Add a custom context menu |
---|
317 | def openmenuinput(self, position): |
---|
318 | |
---|
319 | menu = QtGui.QMenu() |
---|
320 | |
---|
321 | selection = self.list_input.selectedItems() |
---|
322 | |
---|
323 | if ( len(selection) != 0 ): |
---|
324 | |
---|
325 | openAction = menu.addAction('Open selected files') |
---|
326 | openAction.setStatusTip('Open file(s) in your favorite editor') |
---|
327 | openAction.triggered.connect(self.OpenFilesInput) |
---|
328 | action = menu.exec_(self.list_input.mapToGlobal(position)) |
---|
329 | |
---|
330 | # Add a custom context menu |
---|
331 | def openmenuuser(self, position): |
---|
332 | |
---|
333 | menu = QtGui.QMenu() |
---|
334 | |
---|
335 | selection = self.list_user.selectedItems() |
---|
336 | |
---|
337 | if ( len(selection) != 0 ): |
---|
338 | |
---|
339 | openAction = menu.addAction('Open selected files') |
---|
340 | openAction.setStatusTip('Open file(s) in your favorite editor') |
---|
341 | openAction.triggered.connect(self.OpenFilesUser) |
---|
342 | action = menu.exec_(self.list_user.mapToGlobal(position)) |
---|
343 | |
---|
344 | |
---|
345 | def OpenFilesInput(self): |
---|
346 | |
---|
347 | sel_job = self.list_jobs.currentItem().text() |
---|
348 | sel_files = self.list_input.selectedItems() |
---|
349 | |
---|
350 | input_dir = job_dir + "/" + sel_job + "/INPUT/" |
---|
351 | |
---|
352 | open_files = "" |
---|
353 | for i in range(0,len(sel_files)): |
---|
354 | open_files = open_files + "xdg-open " + input_dir + sel_files[i].text() + "; " |
---|
355 | |
---|
356 | os.system(str(open_files)) |
---|
357 | |
---|
358 | def OpenFilesUser(self): |
---|
359 | |
---|
360 | sel_job = self.list_jobs.currentItem().text() |
---|
361 | sel_files = self.list_user.selectedItems() |
---|
362 | |
---|
363 | input_dir = job_dir + "/" + sel_job + "/USER_CODE/" |
---|
364 | |
---|
365 | open_files = "" |
---|
366 | for i in range(0,len(sel_files)): |
---|
367 | open_files = open_files + "xdg-open " + input_dir + sel_files[i].text() + "; " |
---|
368 | |
---|
369 | os.system(str(open_files)) |
---|
370 | selection = self.list_jobs.selectedItems() |
---|
371 | |
---|
372 | |
---|
373 | |
---|
374 | # Message box for showing RUN_CONTROL output |
---|
375 | class CreateSetBox(QtGui.QDialog): |
---|
376 | def __init__(self): |
---|
377 | |
---|
378 | super(CreateSetBox, self).__init__() |
---|
379 | |
---|
380 | uic.loadUi(palm_bin + '/palm_jm_files/palm_jm_create_set.ui', self) |
---|
381 | |
---|
382 | self.show() |
---|
383 | |
---|
384 | return |
---|
385 | |
---|
386 | # Cancel button |
---|
387 | def rejected(self): |
---|
388 | |
---|
389 | self.close() |
---|
390 | |
---|
391 | return |
---|
392 | |
---|
393 | |
---|
394 | # OK button |
---|
395 | def accept(self): |
---|
396 | |
---|
397 | global set_list |
---|
398 | |
---|
399 | text = self.list.toPlainText() |
---|
400 | set_list = text.split('\n') |
---|
401 | self.close() |
---|
402 | |
---|
403 | return |
---|
404 | |
---|
405 | # Main loop |
---|
406 | def main(): |
---|
407 | |
---|
408 | app = QtGui.QApplication(sys.argv) |
---|
409 | res = JobManager() |
---|
410 | sys.exit(app.exec_()) |
---|
411 | |
---|
412 | |
---|
413 | if __name__ == '__main__': |
---|
414 | main() |
---|