Adding initial Driver Update Disk building support

2022-01-14 17:33:22 +01:00 · 2022-01-14 17:33:22 +01:00 · aa26a0a8c2
commit aa26a0a8c2
parent 0f1df4035a
4 changed files with 482 additions and 0 deletions
--- a/plugins/cli/dud.py
+++ b/plugins/cli/dud.py
@ -0,0 +1,74 @@
+from optparse import OptionParser
+
+from koji import canonArch
+
+from koji.plugin import export_cli
+from koji_cli.lib import (
+    _running_in_bg,
+    activate_session,
+    watch_tasks,
+)
+
+# All client related stuff, to be located in ~/.koji/plugins/dud.py
+@export_cli
+def handle_dud_build(goptions, session, args):
+    "[build] Run a command in a buildroot"
+    usage = "usage: %prog dud-build [options] <koji_target> --scmurl=<git_repo> <dud_name> <version> <pkg(nvr optional)> [<pkg> ...]"
+    usage += "\n(Specify the --help global option for a list of other help options)"
+    parser = OptionParser(usage=usage)
+    parser.add_option("--scratch", action="store_true", default=False,
+                    help="Perform a scratch build")
+    parser.add_option("--scmurl", metavar="SCMURL", default=None,
+                    help="SCM repository URL for non-rpm related content to be included in the ISO")
+    parser.add_option("--alldeps", action="store_true", default=False,
+                    help="Download all involved rpm dependencies and put them inside the DUD ISO as well")
+    parser.add_option("--arch", action="append", dest="arches", default=[],
+                      help="Limit arches to this subset")
+    parser.add_option("--can-fail", action="store", dest="optional_arches",
+                      metavar="ARCH1,ARCH2,...", default="",
+                      help="List of archs which are not blocking for build "
+                           "(separated by commas.")
+    parser.add_option("--nowait", action="store_false", dest="wait", default=True)
+    parser.add_option("--wait", action="store_true",
+                      help="Wait on the image creation, even if running in the background")
+    (options, args) = parser.parse_args(args)
+
+    if len(args) < 4:
+        parser.error("Incorrect number of arguments")
+        assert False  # pragma: no cover
+    pkg_list = []
+
+    # Can't use * operator with unpacking in Python 2.7, but this works for both Python 2 and 3
+    target, dud_name, dud_version, pkg_list = args[0], args[1], args[2], args[3:]
+
+    activate_session(session, goptions)
+
+    kwargs = {
+        'scratch': options.scratch,
+        'alldeps': options.alldeps,
+        'scmurl': options.scmurl,
+        'optional_arches': [canonArch(arch)
+                            for arch in options.optional_arches.split(',')
+                            if arch],
+    }
+
+    arches = []
+    if options.arches:
+        arches = [canonArch(arch) for arch in options.arches]
+
+    task_id = session.dudBuild(
+        target=target,
+        arches=arches,
+        dud_version=dud_version,
+        dud_name=dud_name,
+        pkg_list=pkg_list,
+        **kwargs)
+
+    if not goptions.quiet:
+        print("Created task: %d" % task_id)
+        print("Task info: %s/taskinfo?taskID=%s" % (goptions.weburl, task_id))
+    if options.wait or (options.wait is None and not _running_in_bg()):
+        session.logout()
+        return watch_tasks(session, [task_id], quiet=goptions.quiet,
+                           poll_interval=goptions.poll_interval, topurl=goptions.topurl)
+