如何在Snowflake中通过宏批量删除符合条件的Schema

以下是整合查询逻辑后的批量删除宏，解决循环执行报错的问题：

{% macro drop_dev_el_schemas() %}
  -- 定义查询符合条件Schema的SQL
  {% set get_schemas_query %}
    SELECT SCHEMA_NAME
    FROM {{ target.database }}.INFORMATION_SCHEMA.SCHEMATA
    WHERE SCHEMA_NAME != 'INFORMATION_SCHEMA'
      AND SCHEMA_NAME ILIKE '%DEV_EL_%'
  {% endset %}

  -- 执行查询并获取结果集
  {% set schemas = run_query(get_schemas_query) %}

  -- 判断是否有需要删除的Schema
  {% if schemas.rows %}
    -- 循环处理每个Schema
    {% for row in schemas.rows %}
      {% set schema_name = row[0] %}
      {% set relation = api.Relation.create(database=target.database, schema=schema_name) %}
      
      -- 先检查Schema是否存在，避免无意义报错
      {% if relation.exists() %}
        {% do drop_schema(relation) %}
        {{ log("成功删除Schema: " ~ schema_name, info=True) }}
      {% else %}
        {{ log("Schema不存在，跳过: " ~ schema_name, info=True) }}
      {% endif %}
    {% endfor %}
  {% else %}
    {{ log("未匹配到任何符合条件的Schema", info=True) }}
  {% endif %}
{% endmacro %}

关键说明 #

1. 查询结果获取：使用dbt内置的run_query()执行自定义SQL，这是获取查询结果的标准方式，避免直接拼接字符串导致的语法错误
2. 结果遍历：通过schemas.rows访问查询结果的行数据，每行的第一个元素即为Schema名称
3. 存在性检查：加入relation.exists()判断，防止删除不存在的Schema触发报错
4. 日志输出：用log()函数打印操作日志，便于追踪执行过程

执行方式 #

在终端运行以下命令触发宏执行：

dbt run-operation drop_dev_el_schemas

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