fix(sql): handle parsing aliases

ibis-project · Jul 31, 2023 · 3645cf4 · 3645cf4
1 parent b6b3c2b
commit 3645cf4
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/ibis/expr/sql.py b/ibis/expr/sql.py
@@ -60,6 +60,13 @@ def to_sqlglot(self):
 
     def overlay(self, step):
         updates = {dep.name: convert(dep, catalog=self) for dep in step.dependencies}
+
+        # handle scan aliases: FROM foo AS bar
+        source = getattr(step, "source", None)
+        alias = getattr(source, "args", {}).get("alias")
+        if alias is not None and (source_name := self.get(source.name)) is not None:
+            self[alias.name] = source_name
+
         return Catalog({**self, **updates})