Fix: qualify column in func with equality

tobymao · Jun 30, 2023 · 3005306 · 3005306
1 parent d4e0e0e
commit 3005306
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 8 deletions.
diff --git a/sqlglot/optimizer/scope.py b/sqlglot/optimizer/scope.py
@@ -268,14 +268,19 @@ def columns(self):
             self._columns = []
             for column in columns + external_columns:
                 ancestor = column.find_ancestor(
-                    exp.Select, exp.Qualify, exp.Order, exp.Having, exp.Hint
+                    exp.Select, exp.Qualify, exp.Order, exp.Having, exp.Hint, exp.Table
                 )
                 if (
                     not ancestor
                     or column.table
                     or isinstance(ancestor, exp.Select)
-                    or (isinstance(ancestor, exp.Order) and isinstance(ancestor.parent, exp.Window))
-                    or (column.name not in named_selects and not isinstance(ancestor, exp.Hint))
+                    or (
+                        isinstance(ancestor, exp.Order)
+                        and (
+                            isinstance(ancestor.parent, exp.Window)
+                            or column.name not in named_selects
+                        )
+                    )
                 ):
                     self._columns.append(column)
 

diff --git a/sqlglot/parser.py b/sqlglot/parser.py
@@ -3322,11 +3322,6 @@ def _parse_lambda(self, alias: bool = False) -> t.Optional[exp.Expression]:
         else:
             this = self._parse_select_or_expression(alias=alias)
 
-            if isinstance(this, exp.EQ):
-                left = this.this
-                if isinstance(left, exp.Column):
-                    left.replace(exp.var(left.text("this")))
-
         return self._parse_limit(self._parse_order(self._parse_respect_or_ignore_nulls(this)))
 
     def _parse_schema(self, this: t.Optional[exp.Expression] = None) -> t.Optional[exp.Expression]:

diff --git a/tests/fixtures/identity.sql b/tests/fixtures/identity.sql
@@ -155,6 +155,7 @@ REGEXP_LIKE('new york', '.')
 REGEXP_SPLIT('new york', '.')
 SPLIT('new york', '.')
 X((y AS z)).1
+X(a.b = 1)
 (x AS y, y AS z)
 REPLACE(1)
 DATE(x) = DATE(y)