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
|
# Copyright (C) 2010, 2011 Linaro Limited
#
# Author: Michael Hudson-Doyle <michael.hudson@linaro.org>
#
# This file is part of lava-scheduler-tool.
#
# lava-scheduler-tool is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License version 3
# as published by the Free Software Foundation
#
# lava-scheduler-tool is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with lava-scheduler-tool. If not, see <http://www.gnu.org/licenses/>.
import os
import sys
import argparse
import xmlrpclib
from lava_tool.authtoken import AuthenticatingServerProxy, KeyringAuthBackend
from lava.tool.command import Command, CommandGroup
from lava.tool.errors import CommandError
from lava.tool.commands import ExperimentalCommandMixIn
class scheduler(CommandGroup):
"""
Interact with LAVA Scheduler
"""
namespace = "lava.scheduler.commands"
class submit_job(ExperimentalCommandMixIn, Command):
"""
Submit a job to lava-scheduler
"""
@classmethod
def register_arguments(cls, parser):
super(submit_job, cls).register_arguments(parser)
parser.add_argument("SERVER")
parser.add_argument("JSON_FILE")
def invoke(self):
self.print_experimental_notice()
server = AuthenticatingServerProxy(
self.args.SERVER, auth_backend=KeyringAuthBackend())
with open(self.args.JSON_FILE, 'rb') as stream:
command_text = stream.read()
try:
job_id = server.scheduler.submit_job(command_text)
except xmlrpclib.Fault, e:
raise CommandError(str(e))
else:
print "submitted as job id:", job_id
class resubmit_job(ExperimentalCommandMixIn, Command):
@classmethod
def register_arguments(self, parser):
parser.add_argument("SERVER")
parser.add_argument("JOB_ID", type=int)
def invoke(self):
self.print_experimental_notice()
server = AuthenticatingServerProxy(
self.args.SERVER, auth_backend=KeyringAuthBackend())
try:
job_id = server.scheduler.resubmit_job(self.args.JOB_ID)
except xmlrpclib.Fault, e:
raise CommandError(str(e))
else:
print "resubmitted as job id:", job_id
class cancel_job(ExperimentalCommandMixIn, Command):
@classmethod
def register_arguments(self, parser):
parser.add_argument("SERVER")
parser.add_argument("JOB_ID", type=int)
def invoke(self):
self.print_experimental_notice()
server = AuthenticatingServerProxy(
self.args.SERVER, auth_backend=KeyringAuthBackend())
server.scheduler.cancel_job(self.args.JOB_ID)
class job_output(Command):
"""
Get job output from the scheduler.
"""
@classmethod
def register_arguments(cls, parser):
super(job_output, cls).register_arguments(parser)
parser.add_argument("SERVER")
parser.add_argument("JOB_ID",
type=int,
help="Job ID to download output file")
parser.add_argument("--overwrite",
action="store_true",
help="Overwrite files on the local disk")
parser.add_argument("--output", "-o",
type=argparse.FileType("wb"),
default=None,
help="Alternate name of the output file")
def invoke(self):
if self.args.output is None:
filename = str(self.args.JOB_ID) + '_output.txt'
if os.path.exists(filename) and not self.args.overwrite:
print >> sys.stderr, "File {filename!r} already exists".format(
filename=filename)
print >> sys.stderr, "You may pass --overwrite to write over it"
return -1
stream = open(filename, "wb")
else:
stream = self.args.output
filename = self.args.output.name
server = AuthenticatingServerProxy(
self.args.SERVER, auth_backend=KeyringAuthBackend())
stream.write(server.scheduler.job_output(self.args.JOB_ID).data)
print "Downloaded job output of {0} to file {1!r}".format(
self.args.JOB_ID, filename)
|
