如何在DBT中用SELECT设置变量并用于FOR循环？解决报错问题

核心问题分析 #

你遇到的错误根源在于：

• dbt的编译阶段（如SQLFluff lint、dbt parse）不会执行数据库查询，load_result/run_query返回的是Undefined，导致Jinja渲染时无法将其作为整数传入range()。
• 直接使用run_query时，返回的是AgateTable对象而非直接数值，需要正确提取结果。
• 计算日期差无需关联表，直接用标量查询即可，不必加FROM {{ ref('my_table') }}。

修正后的完整代码（推荐用run_query） #

{% set default_months = 12 %}  -- 编译阶段占位默认值，可按需调整
{% set num_months = default_months %}

{% if execute %}
    {% set query_result = run_query("SELECT DATEDIFF('month', CAST('2019-01-01' AS DATE), CAST(NOW() AS DATE))") %}
    {% set num_months = query_result.columns[0][0] | int %}
{% endif %}

SELECT
    {% for month_offset in range(0, num_months) %}
        COUNT(DISTINCT id) AS month_{{ month_offset + 1 }}{% if not loop.last %},
        {% endif %}
    {% endfor %}
FROM {{ ref('my_table') }}
LIMIT 1

代码说明 #

1. 默认值占位：定义default_months避免编译阶段出现Undefined错误，SQLFluff或dbt parse时会用这个值渲染模板。
2. 执行阶段判断：{% if execute %}仅在dbt运行（如dbt run）时执行数据库查询，获取真实的月份差。
3. 结果提取：通过columns[0][0]从run_query返回的AgateTable中提取第一行第一列的数值，| int确保结果为整数类型。
4. 简化查询逻辑：去掉不必要的FROM和LIMIT，直接查询标量值，提升效率且不依赖表的存在。

替代方案（用statement+load_result） #

若坚持使用statement方法，同样需要添加execute判断：

{% set default_months = 12 %}
{% set num_months = default_months %}

{% if execute %}
    {% call statement('get_months', fetch_result=True) %}
        SELECT DATEDIFF('month', CAST('2019-01-01' AS DATE), CAST(NOW() AS DATE))
    {% endcall %}
    {% set num_months = load_result('get_months')['data'][0][0] | int %}
{% endif %}

SELECT
    {% for month_offset in range(0, num_months) %}
        COUNT(DISTINCT id) AS month_{{ month_offset + 1 }}{% if not loop.last %},
        {% endif %}
    {% endfor %}
FROM {{ ref('my_table') }}
LIMIT 1

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