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
|
#!/usr/bin/env python
from __future__ import print_function
import drmaa
import os
def main():
"""Submit an array job, wait for them to finish, and collect results.
This is slightly different from the example from sunsource because it only
waits for the jobs that it submitted.
"""
s = drmaa.Session()
s.initialize()
print('Creating job template')
jt = s.createJobTemplate()
jt.remoteCommand = os.getcwd() + '/sleeper.sh'
jt.args = ['42','Simon says:']
jt.joinFiles=True
joblist = s.runBulkJobs(jt,1,30,2)
print('Your job has been submitted with id ' + str(joblist))
s.synchronize(joblist, drmaa.Session.TIMEOUT_WAIT_FOREVER, False)
for curjob in joblist:
print('Collecting job ' + curjob)
retval = s.wait(curjob, drmaa.Session.TIMEOUT_WAIT_FOREVER)
print('Job: ' + str(retval.jobId) + ' finished with status ' + str(retval.hasExited))
print('Cleaning up')
s.deleteJobTemplate(jt)
s.exit()
if __name__=='__main__':
main()
|
