Merge branch 'main' into fix/224/distkey-error-message

.changes/unreleased/Features-20230803-111146.yaml

@@ -0,0 +1,6 @@
+kind: Features
+body: Support DISTSTYLE for seeds
+time: 2023-08-03T11:11:46.571996-06:00
+custom:
+  Author: WillAyd
+  Issue: "255"

dbt/include/redshift/macros/materializations/seeds/helpers.sql

@@ -1,3 +1,27 @@
+{% macro redshift__create_csv_table(model, agate_table) %}
+  {%- set column_override = model['config'].get('column_types', {}) -%}
+  {%- set quote_seed_column = model['config'].get('quote_columns', None) -%}
+  {%- set _dist = model['config'].get('dist', None) -%}
+
+  {% set sql %}
+    create table {{ this.render() }} (
+        {%- for col_name in agate_table.column_names -%}
+            {%- set inferred_type = adapter.convert_type(agate_table, loop.index0) -%}
+            {%- set type = column_override.get(col_name, inferred_type) -%}
+            {%- set column_name = (col_name | string) -%}
+            {{ adapter.quote_seed_column(column_name, quote_seed_column) }} {{ type }} {%- if not loop.last -%}, {%- endif -%}
+        {%- endfor -%}
+    )
+    {{ dist(_dist) }}
+  {% endset %}
+
+  {% call statement('_') -%}
+    {{ sql }}
+  {%- endcall %}
+
+  {{ return(sql) }}
+{% endmacro %}
+
 {% macro redshift__get_batch_size() %}
   {{ return(500) }}
 {% endmacro %}

tests/functional/adapter/test_simple_seed.py

@@ -53,6 +53,24 @@
 """.lstrip()
 
 
+properties__schema_yml = """
+version: 2
+
+seeds:
+  - name: seed_dist_all
+    config:
+      dist: all
+"""
+
+
+seeds__dist_all_csv = """
+seed_id,weekday
+1,Saturday
+2,Sunday
+3,Monday
+""".lstrip()
+
+
 class TestSimpleSeedColumnOverride(BaseSimpleSeedColumnOverride):
     @pytest.fixture(scope="class")
     def schema(self):
@@ -82,3 +100,23 @@ def test_redshift_simple_seed_with_column_override_redshift(self, project):
         assert len(seed_results) == 2
         test_results = run_dbt(["test"])
         assert len(test_results) == 10
+
+
+class BaseSimpleSeedDiststyleAll:
+    @pytest.fixture(scope="class")
+    def models(self):
+        return {
+            "schema.yml": properties__schema_yml,
+        }
+
+    @pytest.fixture(scope="class")
+    def seeds(self):
+        return {"seed_dist_all.csv": seeds__dist_all_csv}
+
+    def test_simple_seed_with_diststyle_all(self, project):
+        seed_results = run_dbt(["seed", "--show"])
+        assert len(seed_results) == 1
+
+
+class TestSimpleSeedDiststyleAll(BaseSimpleSeedDiststyleAll):
+    pass