Jinja2 模板：优雅处理缺失的 YAML 嵌套键与默认值

import yaml
import sys
from jinja2 import Environment, ChainableUndefined

def render_jinja(template_str, context):
    # 设置 undefined=ChainableUndefined 允许访问未定义的中间键
    jinja_env = Environment(extensions=["jinja2.ext.do"], undefined=ChainableUndefined)
    template_obj = jinja_env.from_string(template_str)
    return template_obj.render(**context).strip()

if __name__ == "__main__":
    # 假设 template.yaml.jinja 是你的模板文件
    # 假设 sys.argv[1] 是你的输入 YAML 文件 (with_override.yaml 或 without_override.yaml)

    # 示例输入数据 (模拟 from_string)
    template_content = """
name: {{ name }}
source.property: {{ overrides.source.property | default("property of " + name) }}
source.property3: {{ overrides.source.property | default("property of " + name) }}
"""

    # 模拟两种输入情况
    config_with_override = {
        "name": "blah",
        "overrides": {
            "source": {
                "property": "something"
            }
        }
    }

    config_without_override = {
        "name": "blah"
    }

    print("--- 渲染 with_override.yaml ---")
    print(render_jinja(template_content, config_with_override))
    print("\n--- 渲染 without_override.yaml ---")
    print(render_jinja(template_content, config_without_override))

name: {{ name }}
source.property: {{ overrides.source.property | default("property of " + name) }}
source.property3: {{ overrides.source.property | default("property of " + name) }}

# 尝试从 overrides.source.property 获取，如果不存在，则尝试从 defaults.source.property 获取，
# 如果再不存在，则使用最终的字符串默认值。
some_other_property: {{ overrides.source.property | default(defaults.source.property) | default("fallback value for " + name) }}

import yaml
from jinja2 import Environment, ChainableUndefined # Jinja2 环境仍可保持 ChainableUndefined

def process_config(raw_config):
    processed_config = {
        "name": raw_config.get("name", "default_name")
    }

    # 设置默认值，并检查是否存在覆盖值
    # 使用 dict.get() 方法安全地访问嵌套键
    # get(key, default_value)
    # 对于嵌套字典，default_value 应为 {} 以便继续 .get()

    # 示例1: 为 source.property 设置默认值
    default_source_property = "default_property_value_from_python"

    # 尝试从 overrides.source.property 获取值
    # 如果 overrides 不存在，则 get("overrides", {}) 返回空字典
    # 如果 source 不存在，则 get("source", {}) 返回空字典
    # 如果 property 不存在，则 get("property", default_source_property) 返回默认值
    overridden_property = raw_config.get("overrides", {}).get("source", {}).get("property", default_source_property)

    processed_config["source_property"] = overridden_property

    # 示例2: 处理其他可选键
    # 假设有一个可选的 description 键
    processed_config["description"] = raw_config.get("description", "No description provided.")

    return processed_config

# 假设 template.yaml.jinja 现在只需要访问已处理的键
template_content_processed = """
name: {{ name }}
source.property: {{ source_property }}
description: {{ description }}
"""

if __name__ == "__main__":
    config_without_override = {
        "name": "blah"
    }
    config_with_override = {
        "name": "blah",
        "overrides": {
            "source": {
                "property": "something_overridden"
            }
        },
        "description": "This is a custom description."
    }

    # 处理数据
    processed_data_without_override = process_config(config_without_override)
    processed_data_with_override = process_config(config_with_override)

    # 渲染模板
    jinja_env = Environment(undefined=ChainableUndefined) # 即使预处理，ChainableUndefined 仍可作为良好实践
    template_obj = jinja_env.from_string(template_content_processed)

    print("--- 渲染 with_override.yaml (Python 预处理) ---")
    print(template_obj.render(**processed_data_with_override).strip())
    print("\n--- 渲染 without_override.yaml (Python 预处理) ---")
    print(template_obj.render(**processed_data_without_override).strip())