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
|
#!/usr/bin/env python2
# -*- coding: utf-8 -*-
import agljobtemplate
import argparse
import urlparse
def parse_cmdline(machines, tests, rfs_types):
parser = argparse.ArgumentParser(description="AGL create job",
formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument('--version', '-v', action='version', version='%(prog)s 1.0')
parser.add_argument('machine', action='store', choices=machines,
help="machine to generate the job for")
parser.add_argument('--priority', '-p', action='store', dest='priority',
help="job priority",
default='medium')
parser.add_argument('--urlbase', '-u', action='store', dest='urlbase',
help="url fetch base",
default='https://download.automotivelinux.org/AGL/upload/ci/')
parser.add_argument('--boot', action='store', dest='rfs_type', nargs=1, required=True,
choices=rfs_types, help='select boot type')
parser.add_argument('--test', dest='tests', action='store', choices=tests + ['all'],
help="add these test to the job", nargs='*', default=[])
parser.add_argument('-o', '--output', dest='job_file', action='store',
help="destination file")
parser.add_argument('-n', '--name', dest='job_name', action='store',
help="job name", default='AGL-short-smoke-wip')
parser.add_argument('-j', '--jobid', dest='job_id', action='store',
help='job id for link creation: URLBASE/JOB_ID', required=True)
parser.add_argument('-i', '--jobidx', dest='job_index', action='store',
help='job index for link creation: URLBASE/JOB_ID/JOB_INDEX', default='1')
return parser.parse_args()
def main():
ajt = agljobtemplate.Agljobtemplate('templates')
args = parse_cmdline(ajt.machines, ajt.tests, ajt.rfs_types)
if args.tests is not None and 'all' in args.tests:
args.tests = ajt.tests
if args.job_id is not None:
args.urlbase = urlparse.urljoin(args.urlbase, args.job_id + '/')
args.job_name += ' - {}'.format(args.job_id)
if args.job_index is not None:
args.urlbase = urlparse.urljoin(args.urlbase, args.job_index)
args.job_name += ' - {}'.format(args.job_index)
job = ajt.render_job(args.urlbase, args.machine, tests=args.tests, priority=args.priority,
rfs_type=args.rfs_type[0], job_name=args.job_name)
if args.job_file is None:
print job
else:
try:
with open(args.job_file, 'w') as j:
j.write(job)
except IOError as e:
print "{}: {}".format(e.strerror, args.job_file)
exit(e.errno)
else:
print "Job written to: {}".format(args.job_file)
if __name__ == '__main__':
main()
|