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
|
#!/usr/bin/env python2
# -*- coding: utf-8 -*-
import agljobtemplate
import argparse
import os
def parse_cmdline(machines, tests, rfs_types):
parser = argparse.ArgumentParser(description="AGL create job",
formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument('-v', action='version', version='%(prog)s 1.0')
parser.add_argument('--machine', action='store', choices=machines,
help="machine to generate the job for",
required=True)
parser.add_argument('--priority', '-p', action='store', dest='priority',
help="job priority",
default='medium')
parser.add_argument('--url', '-u', action='store', dest='url',
help="url fetch base",
default='release')
parser.add_argument('--branch', '--changeid', dest='url_branch', action='store',
help='The branch (or changeid) to generate the job for.')
parser.add_argument('--version', '--patchset', dest='url_version', action='store',
help='The version (or patchset) to generate the job for.')
parser.add_argument('--boot', action='store', dest='rfs_type',
choices=rfs_types, help='select boot type')
parser.add_argument('--callback-from', action='store', dest='callback_from',
help='The LAVA lab (name) that will be responsible of doing the callback. '
'Please read: ./templates/callback/callback_readme.txt')
parser.add_argument('--callback-to', action='store', dest='callback_to',
help='The KernelCI instance (name) that will receive the callback from LAVA. '
'Please read: ./templates/callback/callback_readme.txt')
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('--rootfs-img', dest='rootfs_img', action='store',
help="The name of the root file system image (such as agl-demo-platform-raspberrypi3.ext4.xz)")
parser.add_argument('--kernel-img', dest='kernel_img', action='store',
help="The name of the kernel to boot (such as uImage)")
parser.add_argument('--dtb-img', dest='dtb_img', action='store',
help="The name of the dtb to use (such as uImage-bcm2710-rpi-3-b.dtb)")
parser.add_argument('--modules-img', dest='modules_img', action='store',
help="The name of the modules to use (such as modules.tar.xz)")
parser.add_argument('--build-version', dest='build_version', action='store',
help="the version number of the AGL build.")
args = parser.parse_args()
if (args.url == 'release'):
if (args.url_branch is None) and (args.url_version is None):
args.url = 'default'
elif (args.url_branch is not None) != (args.url_version is not None):
parser.error("Both arguments: '--branch' and '--version' needs to be set")
elif (args.url == 'daily'):
if (not args.url_branch) or (not args.url_version):
parser.error("The argument '--url daily' requires '--branch' and '--version' to be set")
elif (args.url == 'ci'):
if (not args.url_branch) or (not args.url_version):
parser.error("The argument '--url ci' requires '--changeid' and '--patchset' to be set. "
+ "For more information on how to use patchset and changeid use --help.")
return args
def main():
img = None
templates_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), '../templates')
ajt = agljobtemplate.Agljobtemplate(templates_dir)
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
job = ajt.render_job(url=args.url, url_branch=args.url_branch, url_version=args.url_version,
machine=args.machine, tests=args.tests, priority=args.priority,
rfs_type=args.rfs_type, job_name=args.job_name,
lava_callback=args.callback_from, kci_callback=args.callback_to,
rfs_image=args.rootfs_img,
kernel_image=args.kernel_img,
dtb_image=args.dtb_img,
modules_image=args.modules_img,
build_version=args.build_version)
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()
|
