aboutsummaryrefslogtreecommitdiffstats
path: root/utils/create-jobs.py
blob: 463d70600c2a62c3b00aab7163a1854a3361dd48 (plain)
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
#!/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="If using a custom URL specify it there. The files will be fetched from this URL.")
    parser.add_argument('--build-type', dest='build_type', action='store',
                        help="The type of build, can be one of: {'ci', 'daily', 'weekly', 'release'}.",
                        default="default")
    parser.add_argument('--branch', dest='vcs_branch', action='store',
                        help='The build branch.')
    parser.add_argument('--changeid', dest='changeid', action='store',
                        help='The build changeid.')
    parser.add_argument('--version', dest='version', action='store',
                        help='')
    parser.add_argument('--patchset', dest='patchset', action='store',
                        help='The build patchset.')
    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', nargs='+', 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 build.")
    parser.add_argument('--device-tags', dest='device_tags', action='store',
                        help="The device tags to be used to create the job.", nargs='*', default="")
    parser.add_argument('--commit', dest='vcs_commit', action='store',
                        help="The build hash.")
    parser.add_argument('--build-id', dest='build_id', action='store',
                        help="The uniq build information, the build id.")
    parser.add_argument('--applications-url', dest='applications_url', action='store',
                        help="An URL pointing to applications.", default=None)
    parser.add_argument('--app-changeid', dest='app_changeid', action='store',
                        help="An application changeid.", default=None)
    parser.add_argument('--app-patchset', dest='app_patchset', action='store',
                        help="An application patchset.", default=None)

    args = parser.parse_args()

    if args.callback_to and not args.callback_from:
        parser.error("When using '--callback-to', '--callback-from' is mandatory.")

    if (args.build_type == "ci") and not args.url:
        if (not args.changeid) or (not args.patchset):
            parser.error("when using '--build-type' '--changeid' and '--patchset' arguments needs to be set.")

    if (args.build_type == "release") and not args.url:
        if (not args.vcs_branch) or (not args.version):
            parser.error("when using '--build-type' '--branch' and '--versiom' arguments needs to be set.")

    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)

    job = ajt.render_job(url=args.url, changeid=args.changeid, patchset=args.patchset,
                         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, version=args.version,
                         kernel_image=args.kernel_img,
                         dtb_image=args.dtb_img, build_id=args.build_id,
                         modules_image=args.modules_img, vcs_branch=args.vcs_branch,
                         build_type=args.build_type, vcs_commit=args.vcs_commit,
                         build_version=args.build_version, device_tags=args.device_tags,
                         applications_url=args.applications_url,
                         app_changeid=args.app_changeid, app_patchset=args.app_patchset)

    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()