如何让Jinja2模板中所有if语句为真以检测所需变量

要让所有{% if %}分支强制生效，从而捕获其中的变量，最可靠的方法是修改Jinja2模板的抽象语法树（AST），将所有if/elif的条件表达式替换为常量True，这样渲染时会无条件进入所有if分支，无需硬编码场景。

具体实现步骤 #

1. 保留原有的自定义Undefined收集逻辑：你的create_collector()生成的Undefined类负责收集访问到的变量，这部分可继续使用。
2. 添加AST修改逻辑：加载模板后，遍历AST节点，将所有If和elif的条件替换为True。
3. 编译修改后的AST并渲染：用修改后的AST重新生成模板，再执行渲染收集变量。

修改后的完整代码 #

from jinja2 import Environment, FileSystemLoader, select_autoescape, TemplateError
from jinja2.nodes import If, Const

def create_collector():
    variables = set()
    class CollectingUndefined:
        def __getattr__(self, name):
            variables.add(name)
            return self
        def __str__(self):
            return ""
        def __bool__(self):
            # 确保Undefined对象在布尔判断中不会报错
            return False
    return variables, CollectingUndefined

def find_all_variables(template_path):
    variables, undefined_cls = create_collector()

    try:
        env = Environment(
            loader=FileSystemLoader(searchpath="./"),
            autoescape=select_autoescape(),
            undefined=undefined_cls
        )
        # 加载原始模板
        template = env.get_template(template_path)
        
        # 定义AST修改函数：强制所有if/elif条件为True
        def modify_ast(node):
            if isinstance(node, If):
                # 修改if的主条件
                node.test = Const(True)
                # 修改所有elif的条件
                for elif_clause in node.elif_:
                    elif_clause.test = Const(True)
            # 递归处理所有子节点
            for field in node.fields:
                value = getattr(node, field)
                if isinstance(value, list):
                    for item in value:
                        if hasattr(item, 'fields'):
                            modify_ast(item)
                elif hasattr(value, 'fields'):
                    modify_ast(value)
        
        # 解析模板为AST并修改
        ast = env.parse(template.source)
        modify_ast(ast)
        # 用修改后的AST生成新模板
        modified_template = env.from_string(env.compile(ast, template.name))
        
        # 渲染模板触发变量收集
        modified_template.render({})
        
    except TemplateError as error:
        print(f"Validation of script {template_path} failed:")
        print(error)
        exit(1)

    return variables

原理说明 #

• Jinja2的模板渲染流程是：解析模板生成AST → 编译AST为可执行代码 → 执行代码渲染。通过修改AST，我们直接控制了模板的执行逻辑，无需修改Jinja2核心源码。
• 将if/elif的条件替换为Const(True)后，渲染时会无条件进入对应的分支，自定义Undefined类就能捕获到这些分支内引用的所有变量。
• 这种方法不需要硬编码任何if-else场景，能覆盖所有模板中的条件分支。

内容的提问来源于stack exchange，提问作者Charma