Parse StructType columns in case expressions
@@ -11,6 +11,8 @@ def get_columns_used_from_sql(sql):
subtree = tup[0]
if type(subtree) is Column:
column_names.add(subtree.sql())
+ elif type(subtree) is Bracket:
+ column_names.add(subtree.text("this"))
return list(column_names)
