make star macro work with ref or schema.table

jocelynstinson · Oct 30, 2017 · 2884e94 · 2884e94
1 parent 2f151b8
commit 2884e94
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/macros/sql/star.sql b/macros/sql/star.sql
@@ -1,8 +1,13 @@
 {% macro star(from, except=[]) -%}
 
-    {%- set table_parts = from.split('.') %}
+    {%- if from.name -%}
+        {%- set schema_name, table_name = from.schema, from.name -%}
+    {%- else -%}
+        {%- set schema_name, table_name = (from | string).split(".") -%}
+    {%- endif -%}
+
     {%- set include_cols = [] %}
-    {%- set cols = adapter.get_columns_in_table(*table_parts) %}
+    {%- set cols = adapter.get_columns_in_table(schema_name, table_name) -%}
     {%- for col in cols -%}
 
         {%- if col.column not in except -%}