1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
import os
import subprocess
class Wait(object):
'''
-debug Show extra debugging info
-status Show job start and terminate info
-echo Echo log events relevant to [job-number]
-num <number> Wait for this many jobs to end
(default is all jobs)
-wait <seconds> Wait no more than this time
(default is unlimited)
'''
def __init__(self, submit, **kwargs):
self.submit = submit
self.cli_args = []
def toggle_debug(self):
self.cli_args.append(['-debug'])
def toggle_status(self):
self.cli_args.append(['-status'])
def toggle_echo(self):
self.cli_args.append(['-echo'])
def num(self, num_jobs):
self.cli_args.append(['-num', num_jobs])
def timeout(self, seconds):
self.cli_args.append(['-wait', seconds])
def execute(self):
logfile = self.submit.job.logs['log'].replace('$(Cluster)', self.submit.cluster)
print "logfile: ", logfile
condor_wait_args = ' '.join([str(arg) for record in self.cli_args for arg in record])
condor_wait = ' '.join(['condor_wait', condor_wait_args, logfile])
proc = subprocess.Popen(condor_wait.split(), stdout=subprocess.PIPE, stderr=subprocess.PIPE, env=self.submit.environ)
stdout, stderr = proc.communicate()
print(stdout)
print(stderr)
proc.wait()
'''
if not stderr:
if 'cluster' in stdout:
for line in stdout.split(os.linesep):
if 'cluster' in line:
self.cluster = line.split()[-1].strip('.')
'''
|
