diff --git a/build.py b/build.py index 27f0102..1f5b39c 100755 --- a/build.py +++ b/build.py @@ -15,7 +15,7 @@ from pathlib import Path import requests import tomli import yaml -from jinja2 import Template +from jinja2 import Environment, BaseLoader logger = logging.getLogger(__name__) @@ -23,6 +23,12 @@ ORG_DIR = Path("org") PKG_VERSIONS: dict | None = None +# Jinja2 custom fail filter +def fail(message: str): + raise ValueError(message) + +JINJA_ENV = Environment(loader=BaseLoader()) +JINJA_ENV.filters['fail'] = fail @dataclass class Image: @@ -94,12 +100,10 @@ def api_get_source_package_version(branch: str, package_name: str) -> str: params = {"branch": branch, "name": package_name} response = requests.get(api_url, params) if response.status_code != 200: - print(response) api_url = "https://rdb.altlinux.org/api/site/package_versions" params = {"arch": "x86_64", "package_type": "source", "name": package_name} response = requests.get(api_url, params) if response.status_code != 200: - print(response) raise RuntimeError( f"failed to retrieve source package version: source package {package_name!r}, branch {branch!r} " ) @@ -121,7 +125,6 @@ def api_get_source_package_version_from_task(task_id: str, package_name: str): api_url = f"https://rdb.altlinux.org/api/task/packages/{task_id}" response = requests.get(api_url) if response.status_code != 200: - print(response) raise RuntimeError( f"failed to retrieve source package version from task: source package {package_name!r}, task_id {task_id}" ) @@ -155,8 +158,7 @@ class Tags: package_name = image.source_packages[0] if re.search("{%.*%}", package_name): - package_name = Template(package_name).render(branch=branch).strip() - print(f"{package_name=}") + package_name = JINJA_ENV.from_string(package_name).render(branch=branch).strip() if re.search("{{.*}}", package_name): if PKG_VERSIONS is None: @@ -174,7 +176,7 @@ class Tags: f"invalid version for image {image.canonical_name!r}: {PKG_VERSIONS[image.canonical_name]!r}" ) - package_name = Template(package_name).render( + package_name = JINJA_ENV.from_string(package_name).render( version=PKG_VERSIONS[image.canonical_name] ) @@ -204,13 +206,13 @@ class Tags: if image.version_template is not None: if isinstance(image.version_template, str): version_tag = ( - Template(image.version_template).render(version=version).strip() + JINJA_ENV.from_string(image.version_template).render(version=version).strip() ) tags.append(version_tag) elif isinstance(image.version_template, list): for v in image.version_template: version_tag = ( - Template(v).render(version=version).strip() + JINJA_ENV.from_string(v).render(version=version).strip() ) tags.append(version_tag) else: @@ -225,7 +227,7 @@ class Tags: if not isinstance(image.version_template, str): raise RuntimeError(f"expected type for version_template: 'str', got '{type(image.version_template)}' for image {image}") version = ( - Template(image.version_template).render(version=version).strip() + JINJA_ENV.from_string(image.version_template).render(version=version).strip() ) tags = [version] tags.append("latest") @@ -406,7 +408,7 @@ class DockerBuilder: else: registry = "" alt_image = "alt" - rendered = Template(template).render( + rendered = JINJA_ENV.from_string(template).render( alt_image=alt_image, branch=self.branch, install_packages=install_packages, diff --git a/org/k8s/flannel-cni-plugin/info.yaml b/org/k8s/flannel-cni-plugin/info.yaml index 4036b6c..25819e0 100644 --- a/org/k8s/flannel-cni-plugin/info.yaml +++ b/org/k8s/flannel-cni-plugin/info.yaml @@ -1,6 +1,14 @@ --- is_versioned: true -version_template: v{{ version.rsplit('.', 1) | first }}-flannel{{ version.rsplit('.', 1) | last }} +version_template: >- + {%- set version_parts = version.split('.') -%} + {%- if version_parts | length == 3 -%} + v{{ version }} + {%- elif version_parts | length == 4 -%} + v{{ version.rsplit('.', 1) | first }}-flannel{{ version.rsplit('.', 1) | last }} + {%- else -%} + {{ "Invalid version format: must have 3 or 4 parts" | fail }} + {%- endif %} source_packages: - cni-plugin-flannel ...