如何在dbt_project.yml中参数化访问dbt嵌套变量？

问题场景 #

我在dbt_project.yml中配置了以下变量：

vars:
    env: env_var('DBT_ENVIRONMENT')
    bucket:
        dev: "dev-bucket"
        uat: "uat-bucket"
        ppd: "ppd-bucket"
        prd: "prd-bucket"

我希望根据env变量的值动态获取对应的bucket名称。目前可以通过{{ var('bucket')['dev'] }}直接引用dev环境的存储桶，但尝试用{{ var('bucket')[ {{ var('env') }} ] }}实现参数化调用时，出现编译错误：

Compilation Error Could not render {{ var('bucket')[ {{ var('env') }} ] }}

不确定这种语法是否符合dbt规范，请问该怎么实现这个需求？

解决方案 #

错误根源 #

问题出在嵌套使用Jinja模板标签——你在一个{{ }}内部又嵌套了另一个{{ }}，这违反了Jinja的语法规则，dbt编译时无法解析这种结构。

直接修正写法 #

只需要一层Jinja标签，直接将var('env')作为字典键传入即可：

{{ var('bucket')[var('env')] }}

如果环境名称是合法的变量名（无特殊字符），也可以用点号+get()方法的写法，这种方式更安全，还能设置默认值：

# 基础用法，不存在时返回None
{{ var('bucket').get(var('env')) }}

# 指定默认值，避免不存在时报错
{{ var('bucket').get(var('env'), 'default-bucket') }}

提前在配置中预计算（优化方案） #

你也可以直接在dbt_project.yml的vars里提前计算好当前环境对应的bucket，后续在模型中直接引用即可：

vars:
    env: env_var('DBT_ENVIRONMENT')
    bucket:
        dev: "dev-bucket"
        uat: "uat-bucket"
        ppd: "ppd-bucket"
        prd: "prd-bucket"
    current_bucket: "{{ bucket[env] }}"

之后在模型里直接使用：

{{ var('current_bucket') }}

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