Merge pull request #79 from davesgonechina/feature/materialize-base-models-as-view

Suggestion to materialize base models as view or table

Author: joellabes

README.md

@@ -92,6 +92,7 @@ model.
 * `table_name` (required): The source table you wish to generate base model SQL for.
 * `leading_commas` (optional, default=False): Whether you want your commas to be leading (vs trailing).
 * `case_sensitive_cols ` (optional, default=False): Whether your source table has case sensitive column names. If true, keeps the case of the column names from the source.
+* `materialized` (optional, default=None): Set materialization style (e.g. table, view, incremental) inside of the model's `config` block. If not set, materialization style will be controlled by `dbt_project.yml`
 
 
 ### Usage:
@@ -101,7 +102,8 @@ model.
 ```
 {{ codegen.generate_base_model(
     source_name='raw_jaffle_shop',
-    table_name='customers'
+    table_name='customers',
+    materialized='table'
 ) }}
 ```
 

integration_tests/tests/test_generate_base_models.sql

@@ -6,6 +6,7 @@
 %}
 
 {% set expected_base_model %}
+
 with source as (
 
     select * from {%raw%}{{ source('codegen_integration_tests__data_source_schema', 'codegen_integration_tests__data_source_table') }}{%endraw%}

integration_tests/tests/test_generate_base_models_all_args.sql

@@ -3,11 +3,14 @@
     source_name='codegen_integration_tests__data_source_schema',
     table_name='codegen_integration_tests__data_source_table_case_sensitive',
     leading_commas=True,
-    case_sensitive_cols=True
+    case_sensitive_cols=True,
+    materialized='table'
   )
 %}
 
 {% set expected_base_model %}
+{{ "{{ config(materialized='table') }}" }}
+
 with source as (
 
     select * from {%raw%}{{ source('codegen_integration_tests__data_source_schema', 'codegen_integration_tests__data_source_table_case_sensitive') }}{%endraw%}

integration_tests/tests/test_generate_base_models_case_sensitive.sql

@@ -6,6 +6,7 @@
 %}
 
 {% set expected_base_model %}
+
 with source as (
 
     select * from {%raw%}{{ source('codegen_integration_tests__data_source_schema', 'codegen_integration_tests__data_source_table_case_sensitive') }}{%endraw%}

integration_tests/tests/test_generate_base_models_leading.sql

@@ -7,6 +7,7 @@
 %}
 
 {% set expected_base_model %}
+
 with source as (
 
     select * from {%raw%}{{ source('codegen_integration_tests__data_source_schema', 'codegen_integration_tests__data_source_table') }}{%endraw%}

macros/generate_base_model.sql

@@ -1,10 +1,15 @@
-{% macro generate_base_model(source_name, table_name, leading_commas=False, case_sensitive_cols=False) %}
+{% macro generate_base_model(source_name, table_name, leading_commas=False, case_sensitive_cols=False, materialized=None) %}
 
 {%- set source_relation = source(source_name, table_name) -%}
 
 {%- set columns = adapter.get_columns_in_relation(source_relation) -%}
 {% set column_names=columns | map(attribute='name') %}
 {% set base_model_sql %}
+
+{%- if materialized is not none -%}
+    {{ "{{ config(materialized='" ~ materialized ~ "') }}" }}
+{%- endif %}
+
 with source as (
 
     select * from {% raw %}{{ source({% endraw %}'{{ source_name }}', '{{ table_name }}'{% raw %}) }}{% endraw %}