Feat: add dot to executor closes tobymao#1676

adrianisk · Jun 21, 2023 · 7d00300 · 7d00300
1 parent 1fc715e
commit 7d00300
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/sqlglot/executor/env.py b/sqlglot/executor/env.py
@@ -153,6 +153,7 @@ def interval(this, unit):
     "CONCAT": null_if_any(lambda *args: "".join(args)),
     "CONCATWS": null_if_any(lambda this, *args: this.join(args)),
     "DIV": null_if_any(lambda e, this: e / this),
+    "DOT": null_if_any(lambda e, this: e[this]),
     "EQ": null_if_any(lambda this, e: this == e),
     "EXTRACT": null_if_any(lambda this, e: getattr(e, this)),
     "GT": null_if_any(lambda this, e: this > e),
@@ -167,6 +168,7 @@ def interval(this, unit):
     "LOWER": null_if_any(lambda arg: arg.lower()),
     "LT": null_if_any(lambda this, e: this < e),
     "LTE": null_if_any(lambda this, e: this <= e),
+    "MAP": null_if_any(lambda k, v: dict(zip(k, v))),
     "MOD": null_if_any(lambda e, this: e % this),
     "MUL": null_if_any(lambda e, this: e * this),
     "NEQ": null_if_any(lambda this, e: this != e),

diff --git a/tests/test_executor.py b/tests/test_executor.py
@@ -595,6 +595,7 @@ def test_scalar_functions(self):
             ("YEAR(CURRENT_DATE()) <> (YEAR(CURRENT_DATE()))", False),
             ("1::bool", True),
             ("0::bool", False),
+            ("MAP(['a'], [1]).a", 1),
         ]:
             with self.subTest(sql):
                 result = execute(f"SELECT {sql}")