feat(duckdb): Parse ** and ^ operators as POW (#4193)

tobymao · Oct 1, 2024 · 0adbbf7 · 0adbbf7
1 parent 0388a51
commit 0adbbf7
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/sqlglot/dialects/duckdb.py b/sqlglot/dialects/duckdb.py
@@ -289,6 +289,7 @@ class Tokenizer(tokens.Tokenizer):
  KEYWORDS = {
  **tokens.Tokenizer.KEYWORDS,
  "//": TokenType.DIV,
+ "**": TokenType.DSTAR,
  "ATTACH": TokenType.COMMAND,
  "BINARY": TokenType.VARBINARY,
  "BITSTRING": TokenType.BIT,
@@ -325,6 +326,13 @@ class Parser(parser.Parser):
  **parser.Parser.BITWISE,
  TokenType.TILDA: exp.RegexpLike,
  }
+ BITWISE.pop(TokenType.CARET)
+
+ EXPONENT = {
+ **parser.Parser.EXPONENT,
+ TokenType.CARET: exp.Pow,
+ TokenType.DSTAR: exp.Pow,
+ }
 
  FUNCTIONS_WITH_ALIASED_ARGS = {*parser.Parser.FUNCTIONS_WITH_ALIASED_ARGS, "STRUCT_PACK"}
 

diff --git a/tests/dialects/test_duckdb.py b/tests/dialects/test_duckdb.py
@@ -864,6 +864,8 @@ def test_duckdb(self):
  "redshift": UnsupportedError,
  },
  )
+ self.validate_identity("a ^ b", "POWER(a, b)")
+ self.validate_identity("a ** b", "POWER(a, b)")
 
  def test_array_index(self):
  with self.assertLogs(helper_logger) as cm: