summaryrefslogtreecommitdiffstats
path: root/utils/job-prereq.py
blob: a7b414b22fbd21ae418ce29249b3c6a57aacc1e1 (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
#!/usr/bin/env python2
# -*- coding: utf-8 -*-

import agljobtemplate
import argparse
import os
import yaml

FILE_MAP = {
    "kernel",
    "dtb",
    "initrd",
    "nbdroot",
}

FILE_MAP_X86 = {
    "kernel",
    "initrd",
    "nbdroot",
}

# Mapping for qemu between command line QEMU args and LAVA yaml template file names
FILE_MAP_QEMU = {
    "kernel": "kernel",
    "initrd": "ramdisk",
}

def parse_cmdline(machines):
    description = "Print to stdout the file names needed to create a LAVA job"
    parser = argparse.ArgumentParser(description=description,
                                     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 output the job prerequisites.",
                        required=True)
    parser.add_argument('--dtb', action='store_true')
    parser.add_argument('--kernel', action='store_true')
    parser.add_argument('--initrd', action='store_true')
    parser.add_argument('--nbdroot', action='store_true')
    parser.add_argument('--build-type', action='store', dest='build_type',
                        nargs=3,
                        help="The type of build. It defines the URL to upload to.",
                        required=True)
    parser.add_argument('--branch', dest='vcs_branch', action='store', default='master',
                         help='The build branch.')

    args = parser.parse_args()
    return args


def main():
    templates_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), '../templates')
    ajt = agljobtemplate.Agljobtemplate(templates_dir)
    args = parse_cmdline(ajt.machines)
    if args.build_type[0] == "ci":
        job = ajt.render_job(build_type=args.build_type[0],
                         changeid=args.build_type[1],
                         patchset=args.build_type[2],
                         machine=args.machine, vcs_branch=args.vcs_branch)
    else:
        job = ajt.render_job(build_type=args.build_type[0],
                        vcs_branch=args.vcs_branch,
                        version=args.build_type[2],
                        machine=args.machine)

    job_yaml = yaml.safe_load(job)
    if args.machine == "qemux86-64" or args.machine == "qemuarm" or args.machine == "qemuarm64":
        for key in FILE_MAP_QEMU:
            if getattr(args, key):
                    print job_yaml["actions"][0]["deploy"]["images"][FILE_MAP_QEMU[key]].get("url").split('/')[-1]
    elif args.machine == "intel-corei7-64" or args.machine == "upsquare":
        for key in FILE_MAP_X86:
            if getattr(args, key):
                    print job_yaml["actions"][0]["deploy"][key].get("url").split('/')[-1]
    else:
        for key in FILE_MAP:
            if getattr(args, key):
                    print job_yaml["actions"][0]["deploy"][key].get("url").split('/')[-1]

if __name__ == '__main__':
    main()