.gitlab-ci/generate_lava.py

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