jinja2 实现return macro

设计上来自dbt 的return macro， 实际上就是一个macro 会触发一个exception 我们对于这个exception 单独捕获处理下

returnmacro 函数开发

• return_macro 函数定义

def return_macro(values):

    raise MacroReturn(values)

• MacroReturn 异常定义

class MacroReturn(Exception):

    """This is how we return a value from a macro, not an exception.

 

    Hack of all hacks

    """

 

    def __init__(self, value) -> None:

        self.value = value

使用

• demo.py
  实际就是处理的时候包含一个exception 的处理，代码包含了一些其他功能，可以不关注

from jinja import MacroFuzzEnvironment,get_dbt_macro_name,MacroReturn

 

env = MacroFuzzEnvironment()

 

macro_template_v2 = """

{% macro mydemo(name,version) %}

   {%  if version =='v1' %}

       {{return("appdemo") }}

    {% else %}

        {{return("dalongdemo")}}

    {% endif %}

    {{name }} --  {{version}}

{% endmacro  %}

"""

 

 

def app_echo(name, age):

    return f"========{name}-----{age}======"

 

def return_macro(values):

    raise MacroReturn(values)

# 此处通过globals 将return 函数放到context 中

template = env.from_string(macro_template_v2, globals={"echo":app_echo,"return":return_macro})

 

context = {"name":"dalong","version":"v2"}

 

module  = template.make_module(vars=context, shared=False)

 

print(module.__dict__)

# 通过异常进行return macro 的处理

try:

    macro_func  = module.__dict__[get_dbt_macro_name("mydemo")]

    # 调用方法

    print(macro_func.__dict__)

    info = macro_func(**context)

    print(info)

except MacroReturn as e:

    print(e.value)

• 效果

参考资料

dbt_common/clients/jinja.py
dbt_common/exceptions/macros.py
core/dbt/context/base.py