From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: Received: from mails.dpdk.org (mails.dpdk.org [217.70.189.124]) by inbox.dpdk.org (Postfix) with ESMTP id 9AC40A0584 for ; Wed, 19 Oct 2022 14:53:13 +0200 (CEST) Received: from [217.70.189.124] (localhost [127.0.0.1]) by mails.dpdk.org (Postfix) with ESMTP id 30B15427F3; Wed, 19 Oct 2022 14:53:13 +0200 (CEST) Received: from mail-pl1-f227.google.com (mail-pl1-f227.google.com [209.85.214.227]) by mails.dpdk.org (Postfix) with ESMTP id 8EC62410D1 for ; Wed, 19 Oct 2022 14:53:11 +0200 (CEST) Received: by mail-pl1-f227.google.com with SMTP id f23so17157378plr.6 for ; Wed, 19 Oct 2022 05:53:11 -0700 (PDT) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=iol.unh.edu; s=unh-iol; h=content-transfer-encoding:mime-version:references:in-reply-to :message-id:date:subject:cc:to:from:from:to:cc:subject:date :message-id:reply-to; bh=HFoG07LT7wR59hTKeB2GYG2a9YQP9v6BZafQupEYAts=; b=GTcKjyj0lwcxa2AxfBFtVKIFUjGNyt7JxbI53JYgn7DRJa8z5Td3u8tAAlhChoIl/W LKyaMmJQpAU9H/NwAOZWsnR9D3yBO2Fr2RbOqQ0PYtOwg7pdIgspE7rrbq5W2vhoTCg4 n0URSI071mYfoRykTO5sbQVjg7mJwHg+iJqSs= X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20210112; h=content-transfer-encoding:mime-version:references:in-reply-to :message-id:date:subject:cc:to:from:x-gm-message-state:from:to:cc :subject:date:message-id:reply-to; bh=HFoG07LT7wR59hTKeB2GYG2a9YQP9v6BZafQupEYAts=; b=3iD1yf03yhA38N1OjhwrEr36GdXbDrZqWoYaiXqcC7WAJqLAhH0rlHogiLjjgdQQgR hwVO9Pp/Ygktz7PSYdPLOOtsMnZKn4DW0bNnldDwwT7bTnWYMIqq/frqidfBHD6zMi6L iZkyR3QOaq/KLddKxzplKihQljNO5qliDOOZLBF8soX0ASiqpJVTOpxMMfEp481TBYqr wd+kH1AfRfBfkMw8hhr25Hg2Q35Azj88QQfnopdjch8AReb/fuML5n5/BIeTBPSA2mT4 YkRZf2/GIkqT96chKEO/sxu+TNgD6F7vFRqBp7X+fT5dN22wm/fLUc2MRwz87ClyuZha YuQw== X-Gm-Message-State: ACrzQf16C5U3ja3Wixg7K5sd7zBIlyhaSn9QB/H6cG3OokYJVh90MvTO /MIZBglAwoTPNbUSmbk/0JTfoNGAv1BMafW0Hs8/SmH4WmbEu6rU99zgN2Ay6nlI6XiY4tj1D/m stbkBcp1vTOMON7iy7ZPjU5qAlxie3Y3Tyz7EvK3MDYYoZTAaxO/5wIAV4Uvdqe2lLJotNaoyrA VXMdTqxuS4yg/b X-Google-Smtp-Source: AMsMyM6wv/pZxXgOwc1ZL8VuRIGQMlDw+UuIoc7IvsNt4kBTG0J3RL1GlyfLprmnRIbudhpXq5jSHcV3hvX4 X-Received: by 2002:a17:90b:1806:b0:20d:3256:38 with SMTP id lw6-20020a17090b180600b0020d32560038mr9355369pjb.94.1666183990554; Wed, 19 Oct 2022 05:53:10 -0700 (PDT) Received: from postal.iol.unh.edu (postal.iol.unh.edu. [2606:4100:3880:1234::84]) by smtp-relay.gmail.com with ESMTPS id j21-20020a170902c3d500b0017f799d0ad6sm632684plj.35.2022.10.19.05.53.10 (version=TLS1_2 cipher=ECDHE-ECDSA-AES128-GCM-SHA256 bits=128/128); Wed, 19 Oct 2022 05:53:10 -0700 (PDT) X-Relaying-Domain: iol.unh.edu Received: from iol.unh.edu (unknown [IPv6:2606:4100:3880:1257::105d]) by postal.iol.unh.edu (Postfix) with ESMTP id E606660524E7; Wed, 19 Oct 2022 08:53:09 -0400 (EDT) From: ohilyard@iol.unh.edu To: ci@dpdk.org Cc: aconole@redhat.com, alialnu@nvidia.com, Owen Hilyard Subject: [PATCH v2 3/6] containers/builder: Dockerfile creation script Date: Wed, 19 Oct 2022 08:53:04 -0400 Message-Id: <20221019125307.258582-4-ohilyard@iol.unh.edu> X-Mailer: git-send-email 2.34.1 In-Reply-To: <20221019125307.258582-1-ohilyard@iol.unh.edu> References: <20221019125307.258582-1-ohilyard@iol.unh.edu> MIME-Version: 1.0 Content-Transfer-Encoding: 8bit X-BeenThere: ci@dpdk.org X-Mailman-Version: 2.1.29 Precedence: list List-Id: DPDK CI discussions List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , Errors-To: ci-bounces@dpdk.org From: Owen Hilyard This script will template out all of the Dockerfiles based on the definitions provided in the inventory using the jinja2 templating library. Signed-off-by: Owen Hilyard --- containers/template_engine/make_dockerfile.py | 221 ++++++++++++++++++ 1 file changed, 221 insertions(+) create mode 100755 containers/template_engine/make_dockerfile.py diff --git a/containers/template_engine/make_dockerfile.py b/containers/template_engine/make_dockerfile.py new file mode 100755 index 0000000..ddef84c --- /dev/null +++ b/containers/template_engine/make_dockerfile.py @@ -0,0 +1,221 @@ +#!/usr/bin/env python3 +# SPDX-License-Identifier: BSD-3-Clause +# Copyright (c) 2022 University of New Hampshire +from datetime import datetime +from jinja2 import Environment, FileSystemLoader, select_autoescape +import logging +import os +import argparse +from dataclasses import dataclass +import json +from typing import Any, Dict, List, Optional +import yaml +import jsonschema + + +@dataclass(frozen=True) +class Options: + on_rhel: bool + fail_on_unbuildable: bool + build_abi: bool + output_dir: str + registry_hostname: str + + +def _get_arg_parser() -> argparse.ArgumentParser: + parser = argparse.ArgumentParser(description="Makes the dockerfile") + parser.add_argument("--output-dir", required=True) + parser.add_argument( + "--rhel", + action="store_true", + help="Overwrite the check for running on RHEL", + default=False, + ) + parser.add_argument( + "--fail-on-unbuildable", + action="store_true", + help="If any container would not be possible to build, fail and exit with a non-zero exit code.", + default=False, + ) + parser.add_argument( + "--build-abi", + action="store_true", + help="Whether to build the ABI references into the image. Disabled by default due to producing 10+ GB images. ", + ) + return parser + + +def parse_args() -> Options: + parser = _get_arg_parser() + args = parser.parse_args() + + registry_hostname = ( + os.environ.get("DPDK_CI_CONTAINERS_REGISTRY_HOSTNAME") or "localhost" + ) + + return Options( + on_rhel=args.rhel, + fail_on_unbuildable=args.fail_on_unbuildable, + build_abi=args.build_abi, + output_dir=args.output_dir, + registry_hostname=registry_hostname, + ) + + +def running_on_RHEL(options: Options) -> bool: + """ + RHEL containers can only be built on RHEL, so disable them and emit a + warning if not on RHEL. + """ + redhat_release_path = "/etc/redhat-release" + + if os.path.exists(redhat_release_path): + with open(redhat_release_path) as f: + first_line = f.readline() + on_rhel = "Red Hat Enterprise Linux" in first_line + if on_rhel: + logging.info("Running on RHEL, allowing RHEL containers") + return True + + logging.warning("Not on RHEL, disabling RHEL containers") + assert options is not None, "Internal state error, OPTIONS should not be None" + + if options.on_rhel: + logging.info("Override enabled, enabling RHEL containers") + + return options.on_rhel + + +def get_path_to_parent_directory() -> str: + return os.path.dirname(__file__) + + +def get_raw_inventory(): + parent_dir = get_path_to_parent_directory() + + schema_path = os.path.join(parent_dir, "inventory_schema.json") + inventory_path = os.path.join(parent_dir, "inventory.yaml") + + inventory: Dict[str, Any] + with open(inventory_path, "r") as f: + inventory = yaml.safe_load(f) + + schema: Dict[str, Any] + with open(schema_path, "r") as f: + schema = json.load(f) + + jsonschema.validate(instance=inventory, schema=schema) + return inventory + + +def apply_group_config_to_target( + target: Dict[str, Any], + raw_inventory: Dict[str, Any], + on_rhel: bool, + fail_on_unbuildable: bool, +) -> Optional[Dict[str, Any]]: + groups_for_target: List[Dict[str, Any]] = [] + groups: List[Dict[str, Any]] = raw_inventory["dockerfiles"]["groups"] + group = groups[target["group"]] + + target_primary_group = target["group"] + + assert isinstance(target_primary_group, str), "Target group name was not a string" + + requires_rhel = "rhel" in target_primary_group.lower() + + if requires_rhel and not on_rhel: + logging.warning( + f"Disabling target {target['name']}, because it must be built on RHEL." + ) + if fail_on_unbuildable: + raise AssertionError( + f"Not on RHEL and target {target['name']} must be built on RHEL" + ) + + return None + + while group["parent"] != "NONE": + groups_for_target.append(group) + group = groups[group["parent"]] + + groups_for_target.append(group) # add the "all" group + groups_for_target.reverse() # reverse it so overrides work + + target_packages: List[str] = target.get("packages") or [] + + for group in groups_for_target: + target_packages = [*target_packages, *(group.get("packages") or [])] + target = dict(target, **group) + + target["packages"] = target_packages + + return target + + +def get_processed_inventory(options: Options) -> Dict[str, Any]: + raw_inventory: Dict[str, Any] = get_raw_inventory() + on_rhel = running_on_RHEL(options) + targets = raw_inventory["dockerfiles"]["targets"] + targets = [ + apply_group_config_to_target( + target, raw_inventory, on_rhel, options.fail_on_unbuildable + ) + for target in targets + ] + # remove disabled options + targets = [target for target in targets if target is not None] + raw_inventory["dockerfiles"]["targets"] = targets + + return raw_inventory + + +def main(): + options: Options = parse_args() + + env = Environment( + loader=FileSystemLoader("templates"), + ) + + inventory = get_processed_inventory(options) + + timestamp = datetime.now().strftime("%Y-%m-%d") + + for target in inventory["dockerfiles"]["targets"]: + template = env.get_template(f"containers/{target['group']}.dockerfile.j2") + dockerfile_location = os.path.join( + options.output_dir, target["name"] + ".dockerfile" + ) + + tags: list[str] = target.get("extra_tags") or [] + tags.insert(0, "$R/$N:latest") + tags.insert(1, "$R/$N:$T") + + target["tags"] = tags + + rendered_dockerfile = template.render( + timestamp=timestamp, + target=target, + build_abi=options.build_abi, + registry_hostname=options.registry_hostname, + **inventory, + ) + with open(dockerfile_location, "w") as output_file: + output_file.write(rendered_dockerfile) + + makefile_template = env.get_template(f"containers.makefile.j2") + rendered_makefile = makefile_template.render( + timestamp=timestamp, + build_abi=options.build_abi, + registry_hostname=options.registry_hostname, + **inventory, + ) + makefile_output_path = os.path.join(options.output_dir, "Makefile") + with open(makefile_output_path, "w") as f: + f.write(rendered_makefile) + + +if __name__ == "__main__": + logging.basicConfig() + logging.root.setLevel(0) # log everything + main() -- 2.34.1