initial snapshot: LAVA job creation based on templates

Change-Id: I5d545d3531f4c4190453724738076bd4eddfc4d6
Signed-off-by: Jerome Brunet <jbrunet@baylibre.com>
Signed-off-by: Loys Ollivier <lollivier@baylibre.com>
Signed-off-by: Kevin Hilman <khilman@baylibre.com>

1 files changed, 68 insertions, 0 deletions

diff --git a/utils/create-jobs.py b/utils/create-jobs.py
new file mode 100755
index 0000000..1c15bb6
--- /dev/null
+++ b/utils/create-jobs.py
@@ -0,0 +1,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()