1 |
nsmirnov |
1.1 |
from Actor import *
|
2 |
nsmirnov |
1.2 |
import common
|
3 |
nsmirnov |
1.1 |
|
4 |
|
|
class Submitter(Actor):
|
5 |
|
|
def __init__(self, cfg_params, nsjobs):
|
6 |
|
|
self.cfg_params = cfg_params
|
7 |
nsmirnov |
1.2 |
self.nsjobs = nsjobs
|
8 |
nsmirnov |
1.1 |
return
|
9 |
|
|
|
10 |
|
|
def run(self):
|
11 |
nsmirnov |
1.2 |
"""
|
12 |
|
|
The main method of the class.
|
13 |
|
|
"""
|
14 |
|
|
|
15 |
|
|
common.logger.debug(5, "Submitter::run() called")
|
16 |
|
|
|
17 |
nsmirnov |
1.3 |
total_njobs = common.jobDB.nJobs()
|
18 |
nsmirnov |
1.2 |
if total_njobs == 0 :
|
19 |
|
|
msg = '\nTotal of 0 jobs submitted -- no created jobs found.\n'
|
20 |
|
|
msg += "Maybe you forgot '-create' or '-continue' ?\n"
|
21 |
|
|
common.logger.message(msg)
|
22 |
|
|
return
|
23 |
|
|
|
24 |
nsmirnov |
1.3 |
if self.nsjobs == 'all': self.nsjobs = total_njobs
|
25 |
|
|
if self.nsjobs > total_njobs : self.nsjobs = total_njobs
|
26 |
|
|
|
27 |
nsmirnov |
1.2 |
# Loop over jobs
|
28 |
|
|
|
29 |
|
|
njs = 0
|
30 |
|
|
for nj in range(total_njobs):
|
31 |
|
|
if njs == self.nsjobs : break
|
32 |
|
|
st = common.jobDB.status(nj)
|
33 |
|
|
if st != 'C': continue
|
34 |
|
|
|
35 |
|
|
common.logger.debug(6, "Submitter::run(): job # "+`nj`)
|
36 |
|
|
|
37 |
|
|
jid = common.scheduler.submit(nj)
|
38 |
|
|
|
39 |
|
|
common.jobDB.setStatus(nj, 'S')
|
40 |
|
|
common.jobDB.setJobId(nj, jid)
|
41 |
|
|
njs = njs + 1
|
42 |
|
|
pass
|
43 |
|
|
|
44 |
|
|
####
|
45 |
|
|
|
46 |
|
|
common.jobDB.save()
|
47 |
|
|
|
48 |
|
|
msg = '\nTotal of %d jobs submitted'%njs
|
49 |
|
|
if njs != self.nsjobs: msg = msg + ' from %d requested'%self.nsjobs
|
50 |
|
|
msg = msg + '.\n'
|
51 |
|
|
common.logger.message(msg)
|
52 |
nsmirnov |
1.1 |
return
|
53 |
|
|
|