Tomeu Vizoso | cc6bbf6 | 2019-04-29 16:33:22 +0000 | [diff] [blame] | 1 | #!/usr/bin/env python3 |
| 2 | |
| 3 | from jinja2 import Environment, FileSystemLoader |
| 4 | import argparse |
Tomeu Vizoso | 22d9764 | 2019-12-17 11:50:14 +0100 | [diff] [blame] | 5 | import os |
Tomeu Vizoso | dcd171f | 2020-04-01 13:07:46 +0200 | [diff] [blame] | 6 | import datetime |
Tomeu Vizoso | 6397dff | 2019-10-06 08:49:56 -0700 | [diff] [blame] | 7 | |
Tomeu Vizoso | cc6bbf6 | 2019-04-29 16:33:22 +0000 | [diff] [blame] | 8 | parser = argparse.ArgumentParser() |
| 9 | parser.add_argument("--template") |
Neil Armstrong | dc594c9 | 2019-10-15 15:22:07 +0200 | [diff] [blame] | 10 | parser.add_argument("--pipeline-info") |
Tomeu Vizoso | cc6bbf6 | 2019-04-29 16:33:22 +0000 | [diff] [blame] | 11 | parser.add_argument("--base-artifacts-url") |
Tomeu Vizoso | dcd171f | 2020-04-01 13:07:46 +0200 | [diff] [blame] | 12 | parser.add_argument("--mesa-url") |
Tomeu Vizoso | 22d9764 | 2019-12-17 11:50:14 +0100 | [diff] [blame] | 13 | parser.add_argument("--device-type") |
Tomeu Vizoso | 6c8b921 | 2020-07-03 10:04:20 +0200 | [diff] [blame] | 14 | parser.add_argument("--dtb", nargs='?', default="") |
Tomeu Vizoso | dab0134 | 2019-05-03 17:48:48 +0200 | [diff] [blame] | 15 | parser.add_argument("--kernel-image-name") |
Tomeu Vizoso | 22d9764 | 2019-12-17 11:50:14 +0100 | [diff] [blame] | 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") |
Vasily Khoruzhick | 5e5b534 | 2020-01-16 19:37:32 -0800 | [diff] [blame] | 22 | parser.add_argument("--ci-node-index") |
| 23 | parser.add_argument("--ci-node-total") |
Rohan Garg | 087be7e | 2020-01-28 15:19:53 +0100 | [diff] [blame] | 24 | parser.add_argument("--job-type") |
Tomeu Vizoso | cc6bbf6 | 2019-04-29 16:33:22 +0000 | [diff] [blame] | 25 | args = parser.parse_args() |
| 26 | |
Tomeu Vizoso | 22d9764 | 2019-12-17 11:50:14 +0100 | [diff] [blame] | 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)) |
Tomeu Vizoso | cc6bbf6 | 2019-04-29 16:33:22 +0000 | [diff] [blame] | 29 | |
Tomeu Vizoso | fcd8308 | 2019-12-18 16:11:33 +0100 | [diff] [blame] | 30 | env_vars = "%s CI_NODE_INDEX=%s CI_NODE_TOTAL=%s" % (args.env_vars, args.ci_node_index, args.ci_node_total) |
| 31 | |
Tomeu Vizoso | 22d9764 | 2019-12-17 11:50:14 +0100 | [diff] [blame] | 32 | values = {} |
Neil Armstrong | dc594c9 | 2019-10-15 15:22:07 +0200 | [diff] [blame] | 33 | values['pipeline_info'] = args.pipeline_info |
Tomeu Vizoso | 22d9764 | 2019-12-17 11:50:14 +0100 | [diff] [blame] | 34 | values['base_artifacts_url'] = args.base_artifacts_url |
Tomeu Vizoso | dcd171f | 2020-04-01 13:07:46 +0200 | [diff] [blame] | 35 | values['mesa_url'] = args.mesa_url |
Tomeu Vizoso | 22d9764 | 2019-12-17 11:50:14 +0100 | [diff] [blame] | 36 | values['device_type'] = args.device_type |
Tomeu Vizoso | 6c8b921 | 2020-07-03 10:04:20 +0200 | [diff] [blame] | 37 | values['dtb'] = args.dtb |
Tomeu Vizoso | 22d9764 | 2019-12-17 11:50:14 +0100 | [diff] [blame] | 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 |
Tomeu Vizoso | fcd8308 | 2019-12-18 16:11:33 +0100 | [diff] [blame] | 43 | values['env_vars'] = env_vars |
Tomeu Vizoso | 22d9764 | 2019-12-17 11:50:14 +0100 | [diff] [blame] | 44 | values['deqp_version'] = args.deqp_version |
Tomeu Vizoso | dcd171f | 2020-04-01 13:07:46 +0200 | [diff] [blame] | 45 | |
Rohan Garg | 087be7e | 2020-01-28 15:19:53 +0100 | [diff] [blame] | 46 | f = open(os.path.splitext(os.path.basename(args.template))[0], "w") |
Tomeu Vizoso | 22d9764 | 2019-12-17 11:50:14 +0100 | [diff] [blame] | 47 | f.write(template.render(values)) |
| 48 | f.close() |
Tomeu Vizoso | 6397dff | 2019-10-06 08:49:56 -0700 | [diff] [blame] | 49 | |