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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
#!/usr/bin/python
#
# Copyright 2010 Google Inc. All Rights Reserved.
#
import getpass
import os
from automation.common.state_machine import BasicStateMachine
STATUS_NOT_EXECUTED = "NOT_EXECUTED"
STATUS_EXECUTING = "EXECUTING"
STATUS_SUCCEEDED = "SUCCEEDED"
STATUS_FAILED = "FAILED"
class JobGroupStateMachine(BasicStateMachine):
state_machine = {
STATUS_NOT_EXECUTED: [STATUS_EXECUTING],
STATUS_EXECUTING: [STATUS_SUCCEEDED, STATUS_FAILED]}
final_states = [STATUS_SUCCEEDED, STATUS_FAILED]
class JobGroup(object):
HOMEDIR_PREFIX = os.path.join("/home", getpass.getuser(), "www", "automation")
def __init__(self, label, jobs=None, cleanup_on_completion=True,
cleanup_on_failure=False, description=""):
self._state = JobGroupStateMachine(STATUS_NOT_EXECUTED)
self.id = 0
self.label = label
self.jobs = []
self.cleanup_on_completion = cleanup_on_completion
self.cleanup_on_failure = cleanup_on_failure
self.description = description
if jobs:
for job in jobs:
self.AddJob(job)
def _StateGet(self):
return self._state
def _StateSet(self, new_state):
self._state.Change(new_state)
status = property(_StateGet, _StateSet)
@property
def home_dir(self):
return os.path.join(self.HOMEDIR_PREFIX, "job-group-%d" % self.id)
@property
def time_submitted(self):
try:
return self.status.timeline[1].time_started
except IndexError:
return None
def __repr__(self):
return '{%s: %s}' % (self.__class__.__name__, self.id)
def __str__(self):
return "\n".join(["Job-Group:",
"ID: %s" % self.id] +
[str(job) for job in self.jobs])
def AddJob(self, job):
self.jobs.append(job)
job.group = self
|