如何将变量传递给DBT中的post_hook？

Question

我想在雪花上放多个桌子。我没有将多个命令添加到我的DBT作业中，而只需传递表名称，而是在数据库的Information_Schema下使用了表视图，并找到了一种从那里构造我的Drop语句的方法。

SELECT
    LISTAGG(CONCAT('"DROP TABLE IF EXISTS ', CONCAT(CONCAT_WS('.', TABLE_CATALOG, TABLE_SCHEMA, TABLE_NAME), '"')), ', ')
FROM "{{ var('db_name') }}"."INFORMATION_SCHEMA"."TABLES"
WHERE table_owner = 'TRANSFORM_ROLE'
AND table_schema = '{{ var("schema_name") }}'
AND table_name LIKE '%TEST%TABLE%'

然后，我将上述语句插入我的DBT模型中的呼叫语句：

{%- call statement('generate_drop_table_statement', fetch_result=True) -%}
SELECT
    LISTAGG(CONCAT('"DROP TABLE IF EXISTS ', CONCAT(CONCAT_WS('.', TABLE_CATALOG, TABLE_SCHEMA, TABLE_NAME), '"')), ', ')
FROM "{{ var('db_name') }}"."INFORMATION_SCHEMA"."TABLES"
WHERE table_owner = 'TRANSFORM_ROLE'
AND table_schema = '{{ var("schema_name") }}'
AND table_name LIKE '%TEST%TABLE%'
{%- endcall -%}

然后加载结果：

{%- set drop_statements = load_result('generate_drop_table_statement')['data'][0][0] -%}

此后，我想将drop_statments传递给我的配置中的post_hook。

{{ config(
alias='DROP_STATEMENTS_TABLE',
materialized='table',
post_hook=['{{ drop_statements }}', "DROP TABLE IF EXISTS {{ var('db_name') }}.{{ var('schema_name') }}.FIRST_MODEL"]
) }}

问题是，生成的查询不起作用，而另一个drop语句与drop_statement变量并驾齐驱。我不确定我是否会错误地调用包含我的drop语句的变量，或者根本不可能。

整个代码：

{%- call statement('generate_drop_table_statement', fetch_result=True) -%}
SELECT
    LISTAGG(CONCAT('"DROP TABLE IF EXISTS ', CONCAT(CONCAT_WS('.', TABLE_CATALOG, TABLE_SCHEMA, TABLE_NAME), '"')), ', ')
FROM "{{ var('db_name') }}"."INFORMATION_SCHEMA"."TABLES"
WHERE table_owner = 'TRANSFORM_ROLE'
AND table_schema = '{{ var("schema_name") }}'
AND table_name LIKE '%TEST%TABLE%'
{%- endcall -%}

{%- set drop_statements = load_result('generate_drop_table_statement')['data'][0][0] -%}
{{ log(drop_statements, info=True) }}

{{ config(
alias='DROP_STATEMENTS_TABLE',
materialized='table',
post_hook=['{{ drop_statements }}', "DROP TABLE IF EXISTS {{ var('db_name') }}.{{ var('schema_name') }}.FIRST_MODEL"]
) }}

with statements as (
SELECT
    '{{ drop_statements }}'
)

SELECT *
FROM statements

感谢您的帮助。提前致谢。

更新：

我阅读a 尚未得到支持。显然，前后挂钩只接受字符串。

Answer 1

我认为问题是您已经在Jinja块中时使用了卷发（ {{}} ）。尝试删除这些，看看它是否有效！

Answer 2

您可以这样通过：

{% set SCHEMA = 'PUBLIC' %}
{% set TEST_ID = '123' %}

{{
    config(
        materialized='table',
        pre_hook=["DELETE FROM " ~ SCHEMA ~ ".TEST_TABLE WHERE SRC_ID = '" ~ TEST_ID ~ "';"]
    )
}}