BigQuery Labels

plugins/bigquery/dbt/adapters/bigquery/impl.py

@@ -49,7 +49,7 @@ class BigQueryAdapter(BaseAdapter):
  ConnectionManager = BigQueryConnectionManager
 
  AdapterSpecificConfigs = frozenset({"cluster_by", "partition_by",
- "kms_key_name"})
+ "kms_key_name", "labels"})
 
  ###
  # Implementations of abstract methods

plugins/bigquery/dbt/include/bigquery/macros/adapters.sql

@@ -26,49 +26,59 @@
 
 {%- endmacro -%}
 
-{% macro bigquery_table_options(persist_docs, temporary, kms_key_name) %}
- {% set opts = {} %}
+{% macro bigquery_table_options(persist_docs, temporary, kms_key_name, labels) %}
+ {% set opts = {} -%}
 
- {% set description = get_relation_comment(persist_docs, model) %}
+ {%- set description = get_relation_comment(persist_docs, model) -%}
  {%- if description is not none -%}
- {% do opts.update({'description': "'" ~ description ~ "'"}) %}
- {% endif %}
- {% if temporary %}
+ {%- do opts.update({'description': "'" ~ description ~ "'"}) -%}
+ {%- endif -%}
+ {%- if temporary -%}
  {% do opts.update({'expiration_timestamp': 'TIMESTAMP_ADD(CURRENT_TIMESTAMP(), INTERVAL 12 hour)'}) %}
- {% endif %}
- {% if kms_key_name %}
- {% do opts.update({'kms_key_name': "'" ~ kms_key_name ~ "'"}) %}
- {% endif %}
+ {%- endif -%}
+ {%- if kms_key_name -%}
+ {%- do opts.update({'kms_key_name': "'" ~ kms_key_name ~ "'"}) -%}
+ {%- endif -%}
+ {%- if labels -%}
+ {%- set label_list = [] -%}
+ {%- for label, value in labels.items() -%}
+ {%- do label_list.append((label, value)) -%}
+ {%- endfor -%}
+ {%- do opts.update({'labels': label_list}) -%}
+ {%- endif -%}
 
  {% set options -%}
  OPTIONS({% for opt_key, opt_val in opts.items() %}
  {{ opt_key }}={{ opt_val }}{{ "," if not loop.last }}
  {% endfor %})
  {%- endset %}
- {% do return(options) %}
+ {%- do return(options) -%}
 {%- endmacro -%}
 
 {% macro bigquery__create_table_as(temporary, relation, sql) -%}
  {%- set raw_partition_by = config.get('partition_by', none) -%}
  {%- set raw_cluster_by = config.get('cluster_by', none) -%}
  {%- set raw_persist_docs = config.get('persist_docs', {}) -%}
  {%- set raw_kms_key_name = config.get('kms_key_name', none) -%}
-
+ {%- set raw_labels = config.get('labels', []) -%}
  create or replace table {{ relation }}
  {{ partition_by(raw_partition_by) }}
  {{ cluster_by(raw_cluster_by) }}
- {{ bigquery_table_options(persist_docs=raw_persist_docs, temporary=temporary, kms_key_name=raw_kms_key_name) }}
+ {{ bigquery_table_options(
+ persist_docs=raw_persist_docs, temporary=temporary, kms_key_name=raw_kms_key_name,
+ labels=raw_labels) }}
  as (
  {{ sql }}
  );
-{% endmacro %}
+{%- endmacro -%}
 
 
 {% macro bigquery__create_view_as(relation, sql) -%}
  {%- set raw_persist_docs = config.get('persist_docs', {}) -%}
+ {%- set raw_labels = config.get('labels', []) -%}
 
  create or replace view {{ relation }}
- {{ bigquery_table_options(persist_docs=raw_persist_docs, temporary=false) }}
+ {{ bigquery_table_options(persist_docs=raw_persist_docs, temporary=false, labels=raw_labels) }}
  as (
  {{ sql }}
  );

test/integration/022_bigquery_test/models/labeled_model.sql

@@ -0,0 +1,8 @@
+{{
+ config(
+ materialized = "table",
+ labels = {'town': 'fish', 'analytics': 'yes'}
+ )
+}}
+
+select * from {{ ref('view_model') }}

test/integration/022_bigquery_test/test_simple_bigquery_view.py

@@ -52,7 +52,8 @@ def test__bigquery_simple_run(self):
  self.run_dbt(['seed'])
  self.run_dbt(['seed', '--full-refresh'])
  results = self.run_dbt()
- self.assertEqual(len(results), 6)
+ # Bump expected number of results when adding new model
+ self.assertEqual(len(results), 7)
  self.assert_nondupes_pass()
 
 
@@ -63,7 +64,7 @@ class TestUnderscoreBigQueryRun(TestBaseBigQueryRun):
  def test_bigquery_run_twice(self):
  self.run_dbt(['seed'])
  results = self.run_dbt()
- self.assertEqual(len(results), 6)
+ self.assertEqual(len(results), 7)
  results = self.run_dbt()
- self.assertEqual(len(results), 6)
+ self.assertEqual(len(results), 7)
  self.assert_nondupes_pass()