Feat: add as_ builder

tobymao · May 16, 2023 · 51e8b1f · 51e8b1f
1 parent b9140b6
commit 51e8b1f
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/sqlglot/expressions.py b/sqlglot/expressions.py
@@ -712,6 +712,16 @@ def not_(self, copy=True):
         """
         return not_(self, copy=copy)
 
+    def as_(
+        self,
+        alias: str | Identifier,
+        quoted: t.Optional[bool] = None,
+        dialect: DialectType = None,
+        copy: bool = True,
+        **opts,
+    ) -> exp.Alias:
+        return alias_(self, alias, quoted=quoted, dialect=dialect, copy=copy, **opts)
+
     def _binop(self, klass: t.Type[E], other: t.Any, reverse: bool = False) -> E:
         this = self.copy()
         other = convert(other, copy=True)

diff --git a/tests/test_build.py b/tests/test_build.py
@@ -55,6 +55,7 @@ def test_build(self):
             (lambda: x.eq(1), "x = 1"),
             (lambda: x.neq(1), "x <> 1"),
             (lambda: x.is_(exp.Null()), "x IS NULL"),
+            (lambda: x.as_("y"), "x AS y"),
             (lambda: x.isin(1, "2"), "x IN (1, '2')"),
             (lambda: x.isin(query="select 1"), "x IN (SELECT 1)"),
             (lambda: x.between(1, 2), "x BETWEEN 1 AND 2"),