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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
#!/usr/bin/env python
#
# Copyright (C) 2017 The Android Open Source Project
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
import datetime
from webapp.src import vtslab_status
from webapp.src.handlers.base import BaseHandler
from webapp.src.proto import model
class JobPage(BaseHandler):
"""Main class for /job web page."""
def get(self):
"""Generates an HTML page based on the job queue info kept in DB."""
self.template = "job.html"
job_query = model.JobModel.query()
jobs = job_query.fetch()
template_values = {
"jobs": sorted(jobs, key=lambda x: x.timestamp,
reverse=True),
}
self.render(template_values)
class CreateJobTemplatePage(BaseHandler):
"""Main class for /create_job_template web page."""
def get(self):
"""Generates an HTML page to get custom job info."""
self.template = "create_job_template.html"
template_values = {}
self.render(template_values)
class CreateJobPage(BaseHandler):
"""Main class for /create_job web page."""
def get(self):
"""Generates an HTML page that stores the provided custom job info to DB."""
self.template = "job.html"
serials = self.request.get("serial", default_value="").split(",")
# TODO: check serial >= shards and select only required ones
device_query = model.DeviceModel.query(model.DeviceModel.serial.IN(
serials))
devices = device_query.fetch()
error_devices = []
for device in devices:
if device.scheduling_status in [
vtslab_status.DEVICE_SCHEDULING_STATUS_DICT["reserved"],
vtslab_status.DEVICE_SCHEDULING_STATUS_DICT["use"]
]:
error_devices.append(device.serial)
if error_devices:
message = "Can't create a job because at some devices are not available (%s)." % error_devices
else:
for device in devices:
device.scheduling_status = vtslab_status.DEVICE_SCHEDULING_STATUS_DICT[
"reserved"]
device.put()
message = "A new job is created! Please click 'Job Queue' menu above to see the new job."
new_job = model.JobModel()
new_job.hostname = self.request.get("hostname", default_value="")
new_job.priority = str(vtslab_status.PrioritySortHelper(
self.request.get("priority", default_value="high")))
new_job.test_name = self.request.get("test_name", default_value="")
new_job.device = self.request.get("device", default_value="")
new_job.period = int(self.request.get("period",
default_value="high"))
new_job.serial.extend(serials)
new_job.manifest_branch = self.request.get("manifest_branch",
default_value="")
new_job.build_target = self.request.get("build_target",
default_value="")
new_job.shards = int(self.request.get(
"shards", default_value=str(len(new_job.serial))))
new_job.param = self.request.get("param",
default_value=[]).split(",")
new_job.gsi_branch = self.request.get("gsi_branch",
default_value="")
new_job.gsi_build_target = self.request.get("gsi_build_target",
default_value="")
new_job.gsi_build_id = self.request.get("gsi_build_id",
default_value="latest")
new_job.gsi_pab_account_id = self.request.get("gsi_pab_account_id",
default_value="")
new_job.test_branch = self.request.get("test_branch",
default_value="")
new_job.test_build_target = self.request.get("test_build_target",
default_value="")
new_job.test_build_id = self.request.get("test_build_id",
default_value="latest")
new_job.test_pab_account_id = self.request.get(
"test_pab_account_id", default_value="")
new_job.build_id = self.request.get("build_id",
default_value="latest")
new_job.pab_account_id = self.request.get("pab_account_id",
default_value="")
new_job.status = vtslab_status.JOB_STATUS_DICT["ready"]
new_job.timestamp = datetime.datetime.now()
new_job.put()
job_query = model.JobModel.query()
jobs = job_query.fetch()
template_values = {"message": message, }
self.render(template_values)
|
