.gitlab-ci/generate_lava.py

   1 #!/usr/bin/env python3
   2
   3 from jinja2 import Environment, FileSystemLoader
   4 import argparse
   5 import os
   6
   7 parser = argparse.ArgumentParser()
   8 parser.add_argument("--template")
   9 parser.add_argument("--pipeline-info")
  10 parser.add_argument("--base-artifacts-url")
  11 parser.add_argument("--device-type")
  12 parser.add_argument("--kernel-image-name")
  13 parser.add_argument("--kernel-image-type", nargs='?', default="")
  14 parser.add_argument("--gpu-version")
  15 parser.add_argument("--boot-method")
  16 parser.add_argument("--lava-tags", nargs='?', default="")
  17 parser.add_argument("--env-vars", nargs='?', default="")
  18 parser.add_argument("--deqp-version")
  19 parser.add_argument("--arch")
  20 parser.add_argument("--ci-node-index")
  21 parser.add_argument("--ci-node-total")
  22 args = parser.parse_args()
  23
  24 env = Environment(loader = FileSystemLoader(os.path.dirname(args.template)), trim_blocks=True, lstrip_blocks=True)
  25 template = env.get_template(os.path.basename(args.template))
  26
  27 values = {}
  28 values['pipeline_info'] = args.pipeline_info
  29 values['base_artifacts_url'] = args.base_artifacts_url
  30 values['device_type'] = args.device_type
  31 values['kernel_image_name'] = args.kernel_image_name
  32 values['kernel_image_type'] = args.kernel_image_type
  33 values['gpu_version'] = args.gpu_version
  34 values['boot_method'] = args.boot_method
  35 values['tags'] = args.lava_tags
  36 values['env_vars'] = args.env_vars
  37 values['deqp_version'] = args.deqp_version
  38 values['arch'] = args.arch
  39 values['ci_node_index'] = args.ci_node_index
  40 values['ci_node_total'] = args.ci_node_total
  41
  42 f = open('lava-deqp.yml', "w")
  43 f.write(template.render(values))
  44 f.close()
  45