src/get_builds: service to fetch kernel build nodes

docker-compose.yaml

@@ -209,3 +209,17 @@ services:
  - './data/src:/home/kernelci/data/src'
  - './data/output:/home/kernelci/data/output'
 
+ get-builds:
+ container_name: 'kernelci-pipeline-get-builds'
+ image: 'kernelci/staging-kernelci'
+ env_file: ['.env']
+ stop_signal: 'SIGINT'
+ command:
+ - './pipeline/get_builds.py'
+ - '--settings=${KCI_SETTINGS:-/home/kernelci/config/kernelci.toml}'
+ - 'run'
+ volumes:
+ - './src:/home/kernelci/pipeline'
+ - './config:/home/kernelci/config'
+ extra_hosts:
+ - "host.docker.internal:host-gateway"

src/get_builds.py

@@ -0,0 +1,58 @@
+#!/usr/bin/env python3
+#
+# SPDX-License-Identifier: LGPL-2.1-or-later
+#
+# Copyright (C) 2024 Collabora Limited
+# Author: Jeny Sadadia <jeny.sadadia@collabora.com>
+
+import sys
+
+import kernelci
+import kernelci.config
+from kernelci.legacy.cli import Args, Command, parse_opts
+
+from base import Service
+
+
+class GetBuilds(Service):
+
+ def __init__(self, configs, args):
+ super().__init__(configs, args, 'get_builds')
+
+ def _setup(self, args):
+ return self._api_helper.subscribe_filters({
+ 'kind': 'kbuild',
+ 'state': 'done',
+ 'result': 'pass'
+ }, 'build')
+
+ def _stop(self, sub_id):
+ if sub_id:
+ self._api_helper.unsubscribe_filters(sub_id)
+ sys.stdout.flush()
+
+ def _run(self, sub_id):
+ self.log.info("Listening for events... ")
+ self.log.info("Press Ctrl-C to stop.")
+
+ while True:
+ node, _ = self._api_helper.receive_event_node(sub_id)
+ self.log.info(f"Build node received: {node}")
+ self.log.info(f"artifacts: {node.get('artifacts')}")
+ return True
+
+
+class cmd_run(Command):
+ help = "Listen for successful kernel build events"
+ args = [Args.api_config]
+
+ def __call__(self, configs, args):
+ return GetBuilds(configs, args).run(args)
+
+
+if __name__ == '__main__':
+ opts = parse_opts('get_builds', globals())
+ yaml_configs = opts.get_yaml_configs() or 'config'
+ configs = kernelci.config.load(yaml_configs)
+ status = opts.command(configs, opts)
+ sys.exit(0 if status is True else 1)