Hacksplore

dbt_project.yml

@@ -1,3 +1,5 @@
 name: 'dbt_snowflake_query_tags'
 version: '2.5.4'
 config-version: 2
+flags:
+  require_generic_test_arguments_property: true # add dbt-autoifix suggestion

macros/query_tags.sql → macros/default__set_query_tag.sql

@@ -1,7 +1,3 @@
-{% macro set_query_tag(extra = {}) -%}
-    {{ return(adapter.dispatch('set_query_tag', 'dbt_snowflake_query_tags')(extra=extra)) }}
-{%- endmacro %}
-
 {% macro default__set_query_tag(extra = {}) -%}
     {# Get session level query tag #}
     {% set original_query_tag = get_current_query_tag() %}
@@ -64,16 +60,3 @@
     {{ return(original_query_tag)}}
 {% endmacro %}
 
-{% macro unset_query_tag(original_query_tag) -%}
-    {{ return(adapter.dispatch('unset_query_tag', 'dbt_snowflake_query_tags')(original_query_tag)) }}
-{%- endmacro %}
-
-{% macro default__unset_query_tag(original_query_tag) -%}
-    {% if original_query_tag %}
-    {{ log("Resetting query_tag to '" ~ original_query_tag ~ "'.") }}
-    {% do run_query("alter session set query_tag = '{}'".format(original_query_tag)) %}
-    {% else %}
-    {{ log("No original query_tag, unsetting parameter.") }}
-    {% do run_query("alter session unset query_tag") %}
-    {% endif %}
-{% endmacro %}

macros/default__unset_query_tag.sql

@@ -0,0 +1,10 @@
+{% macro default__unset_query_tag(original_query_tag) -%}
+    {% if original_query_tag %}
+    {{ log("Resetting query_tag to '" ~ original_query_tag ~ "'.") }}
+    {% do run_query("alter session set query_tag = '{}'".format(original_query_tag)) %}
+    {% else %}
+    {{ log("No original query_tag, unsetting parameter.") }}
+    {% do run_query("alter session unset query_tag") %}
+    {% endif %}
+{% endmacro %}
+

macros/set_query_tag.sql

@@ -0,0 +1,4 @@
+{% macro set_query_tag(extra = {}) -%}
+    {{ return(adapter.dispatch('set_query_tag', 'dbt_snowflake_query_tags')(extra=extra)) }}
+{%- endmacro %}
+

macros/unset_query_tag.sql

@@ -0,0 +1,4 @@
+{% macro unset_query_tag(original_query_tag) -%}
+    {{ return(adapter.dispatch('unset_query_tag', 'dbt_snowflake_query_tags')(original_query_tag)) }}
+{%- endmacro %}
+