如何在dbt_project.yml中全局配置模型node_color（按文件名前缀）

问题背景 #

我希望根据文件名前缀为所有dbt模型设置node_color，方便浏览dbt文档：

• fact_ 前缀 → 红色
• base__ 前缀 → 黑色

已编写可用宏，但目前只能在每个模型的.sql文件中手动调用，尝试在dbt_project.yml全局配置时报错：'get_model_color' is undefined，需要实现无需逐个修改模型的全局配置方案。

解决方案步骤 #

1. 调整宏的返回格式

原宏返回包含node_color的字典，而全局配置需要直接返回颜色字符串，修改宏如下：

{% macro get_model_color(model) %}
    {% set default_color = 'blue' %}
    {% set ns = namespace(model_color=default_color) %}

    {% set dict_patterns = {
        "base__[a-z0-9_]+": "black",
        "ref_[a-z0-9_]+": "yellow",
        "fact_[a-z0-9_]+": "red"
    } %}

    {% set re = modules.re %}

    {% for pattern, color in dict_patterns.items() %}
        {% set is_match = re.match(pattern, model.identifier, re.IGNORECASE) %}
        {% if is_match %}
            {% set ns.model_color = color %}
        {% endif %}
    {% endfor %}

    {{ return(ns.model_color) }}
{% endmacro %}

将返回值从字典改为直接返回颜色字符串，适配全局配置的赋值要求。

2. 在dbt_project.yml中配置全局规则

在models块下添加全局配置，注意替换为你的项目名称，确保宏文件放在macros/目录下（dbt会自动加载该目录的宏）：

models:
  your_project_name:  # 替换成实际的dbt项目名称
    +docs:
      node_color: "{{ get_model_color(this) }}"

3. 验证效果

执行以下命令生成并查看文档：

dbt docs generate
dbt docs serve

所有模型会自动根据文件名前缀匹配对应颜色，无需逐个修改模型文件。

失败原因说明 #

• 原宏返回字典格式，而node_color需要字符串值，格式不匹配导致配置失效。
• 若宏未放在macros/目录下，dbt无法识别宏，会出现undefined错误。

内容的提问来源于stack exchange，提问作者L.GAYET