Python Cookie Cutter - 条件代码块？

Question

是否可以根据设置的变量执行某种条件代码块？我正在尝试模板化具有一些 yaml 配置文件的项目。我希望 yaml 配置的一部分是可选的，但我不知道是否有办法使用cookiecutter 来做到这一点。我知道cookiecutter 支持可选文件和目录。

这是一个例子：

{{ cookiecutter.pipeline_name }}:
  models:
    {{ cookiecutter.model_name }}:
      inference:
        instance_type: {{ cookiecutter.instance_type }}
        containers:
          - image:
              name: {{ cookiecutter.image_name }}
              repo: {{ cookiecutter.image_repo }}
              tag: {{ cookiecutter.image_tag }}
            provider: ecr
            data: {{ cookiecutter.model_artifact }}
        async_inference_config:
          s3_output_path: {{ cookiecutter.async_output_path }}
          max_concurrent_invocations_per_instance: {{ cookiecutter.max_invocations }}

async_inference_config 块应该是可选的。如果用户没有填写async_output_path 和max_invocations 变量，那么应该删除整个块。如果这不可能，我可以创建 2 个不同的 cookiecutter 模板。但考虑到这两个模板之间的唯一区别是async_inference_config 块，这似乎是一种浪费。

Answer 1

这不是 cookiecutter 的真正功能，而是 Jinja2 的功能。 看看this part of it's documentation.

所以你的代码应该和这个类似：

{{ cookiecutter.pipeline_name }}:
  models:
    {{ cookiecutter.model_name }}:
      inference:
        instance_type: {{ cookiecutter.instance_type }}
        containers:
          - image:
              name: {{ cookiecutter.image_name }}
              repo: {{ cookiecutter.image_repo }}
              tag: {{ cookiecutter.image_tag }}
            provider: ecr
            data: {{ cookiecutter.model_artifact }}
{% if cookiecutter.async_output_path and cookiecutter.max_invocations %}
        async_inference_config:
          s3_output_path: {{ cookiecutter.async_output_path }}
          max_concurrent_invocations_per_instance: {{ cookiecutter.max_invocations }}
{% endif %}