File tree Expand file tree Collapse file tree 1 file changed +30
-1
lines changed
Expand file tree Collapse file tree 1 file changed +30
-1
lines changed Original file line number Diff line number Diff line change @@ -130,7 +130,33 @@ function parse_tidy_db(exprs, metadata::DataFrame)
130130 end
131131
132132 end
133-
133+ elseif isa (actual_expr, AbstractVector)
134+ for item in actual_expr
135+ col_name = string (item)
136+ if current_sql_mode[] == snowflake ()
137+ col_name = uppercase (col_name)
138+ end
139+ if is_excluded
140+ push! (excluded_columns, col_name)
141+ else
142+ push! (included_columns, col_name)
143+ end
144+ end
145+ elseif isa (actual_expr, Tuple) && all (isa .(actual_expr, Vector{Symbol}))
146+ for vec in actual_expr
147+ for item in vec
148+ col_name = string (item)[2 : end ]
149+ if current_sql_mode[] == snowflake ()
150+ col_name = uppercase (col_name)
151+ end
152+ if is_excluded
153+ push! (excluded_columns, col_name)
154+ else
155+ push! (included_columns, col_name)
156+ end
157+ end
158+ end
159+
134160 else
135161 error (" Unsupported expression type: $expr " )
136162 end
@@ -165,6 +191,9 @@ function parse_tidy_db(exprs, metadata::DataFrame)
165191 return included_columns
166192end
167193
194+
195+
196+
168197function parse_if_else (expr)
169198 transformed_expr = MacroTools. postwalk (expr) do x
170199 # Check if the expression is a call to if_else
You can’t perform that action at this time.
0 commit comments