Skip to content

Commit ba4c900

Browse files
committed
nearly p
1 parent b1822be commit ba4c900

File tree

1 file changed

+30
-1
lines changed

1 file changed

+30
-1
lines changed

src/db_parsing.jl

Lines changed: 30 additions & 1 deletion
Original file line numberDiff line numberDiff 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
166192
end
167193

194+
195+
196+
168197
function parse_if_else(expr)
169198
transformed_expr = MacroTools.postwalk(expr) do x
170199
# Check if the expression is a call to if_else

0 commit comments

Comments
 (0)