sqlglot 27.4.0__py3-none-any.whl → 27.5.0__py3-none-any.whl

sqlglot/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '27.4.0'
-__version_tuple__ = version_tuple = (27, 4, 0)
+__version__ = version = '27.5.0'
+__version_tuple__ = version_tuple = (27, 5, 0)

sqlglot/dialects/bigquery.py

@@ -378,11 +378,13 @@ def _annotate_array(self: TypeAnnotator, expression: exp.Array) -> exp.Array:
         and (query_type := select.meta.get("query_type")) is not None
         and query_type.is_type(exp.DataType.Type.STRUCT)
         and len(query_type.expressions) == 1
+        and isinstance(col_def := query_type.expressions[0], exp.ColumnDef)
+        and (projection_type := col_def.kind) is not None
+        and not projection_type.is_type(exp.DataType.Type.UNKNOWN)
     ):
-        projection_type = query_type.expressions[0].kind.copy()
         array_type = exp.DataType(
             this=exp.DataType.Type.ARRAY,
-            expressions=[projection_type],
+            expressions=[projection_type.copy()],
             nested=True,
         )
         return self._annotate_with_type(expression, array_type)

sqlglot/dialects/duckdb.py

@@ -665,6 +665,7 @@ class DuckDB(Dialect):
         ARRAY_CONCAT_IS_VAR_LEN = False
         ARRAY_SIZE_DIM_REQUIRED = False
         NORMALIZE_EXTRACT_DATE_PARTS = True
+        SUPPORTS_LIKE_QUANTIFIERS = False
 
         TRANSFORMS = {
             **generator.Generator.TRANSFORMS,

sqlglot/dialects/exasol.py

@@ -1,17 +1,25 @@
 from __future__ import annotations
+
+import typing as t
+
 from sqlglot import exp, generator, parser, tokens
+from sqlglot.dialects.clickhouse import timestamptrunc_sql
 from sqlglot.dialects.dialect import (
     Dialect,
-    rename_func,
     binary_from_function,
     build_formatted_time,
-    timestrtotime_sql,
+    rename_func,
     strposition_sql,
+    timestrtotime_sql,
+    unit_to_str,
 )
-from sqlglot.helper import seq_get
 from sqlglot.generator import unsupported_args
+from sqlglot.helper import seq_get
 from sqlglot.tokens import TokenType
 
+if t.TYPE_CHECKING:
+    from sqlglot.dialects.dialect import DialectType
+
 
 def _sha2_sql(self: Exasol.Generator, expression: exp.SHA2) -> str:
     length = expression.text("length")
@@ -19,6 +27,25 @@ def _sha2_sql(self: Exasol.Generator, expression: exp.SHA2) -> str:
     return self.func(func_name, expression.this)
 
 
+# https://docs.exasol.com/db/latest/sql_references/functions/alphabeticallistfunctions/trunc%5Bate%5D%20(datetime).htm
+# https://docs.exasol.com/db/latest/sql_references/functions/alphabeticallistfunctions/trunc%5Bate%5D%20(number).htm
+def _build_trunc(args: t.List[exp.Expression], dialect: DialectType) -> exp.Expression:
+    first, second = seq_get(args, 0), seq_get(args, 1)
+
+    if not first or not second:
+        return exp.Anonymous(this="TRUNC", expressions=args)
+
+    if not first.type:
+        from sqlglot.optimizer.annotate_types import annotate_types
+
+        first = annotate_types(first, dialect=dialect)
+
+    if first.is_type(exp.DataType.Type.DATE, exp.DataType.Type.TIMESTAMP) and second.is_string:
+        return exp.DateTrunc(this=first, unit=second)
+
+    return exp.Anonymous(this="TRUNC", expressions=args)
+
+
 class Exasol(Dialect):
     TIME_MAPPING = {
         "yyyy": "%Y",
@@ -63,12 +90,17 @@ class Exasol(Dialect):
             "BIT_NOT": lambda args: exp.BitwiseNot(this=seq_get(args, 0)),
             "BIT_LSHIFT": binary_from_function(exp.BitwiseLeftShift),
             "BIT_RSHIFT": binary_from_function(exp.BitwiseRightShift),
+            # https://docs.exasol.com/db/latest/sql_references/functions/alphabeticallistfunctions/date_trunc.htm#DATE_TRUNC
+            "DATE_TRUNC": lambda args: exp.TimestampTrunc(
+                this=seq_get(args, 1), unit=seq_get(args, 0)
+            ),
             "EVERY": lambda args: exp.All(this=seq_get(args, 0)),
             "EDIT_DISTANCE": exp.Levenshtein.from_arg_list,
             "HASH_SHA": exp.SHA.from_arg_list,
             "HASH_SHA1": exp.SHA.from_arg_list,
             "HASH_MD5": exp.MD5.from_arg_list,
             "HASHTYPE_MD5": exp.MD5Digest.from_arg_list,
+            "REGEXP_SUBSTR": exp.RegexpExtract.from_arg_list,
             "REGEXP_REPLACE": lambda args: exp.RegexpReplace(
                 this=seq_get(args, 0),
                 expression=seq_get(args, 1),
@@ -82,6 +114,8 @@ class Exasol(Dialect):
             "HASH_SHA512": lambda args: exp.SHA2(
                 this=seq_get(args, 0), length=exp.Literal.number(512)
             ),
+            "TRUNC": _build_trunc,
+            "TRUNCATE": _build_trunc,
             "VAR_POP": exp.VariancePop.from_arg_list,
             "APPROXIMATE_COUNT_DISTINCT": exp.ApproxDistinct.from_arg_list,
             "TO_CHAR": build_formatted_time(exp.ToChar, "exasol"),
@@ -155,12 +189,18 @@ class Exasol(Dialect):
             exp.BitwiseXor: rename_func("BIT_XOR"),
             # https://docs.exasol.com/db/latest/sql_references/functions/alphabeticallistfunctions/every.htm
             exp.All: rename_func("EVERY"),
+            exp.DateTrunc: lambda self, e: self.func("TRUNC", e.this, unit_to_str(e)),
+            exp.DatetimeTrunc: timestamptrunc_sql(),
             # https://docs.exasol.com/db/latest/sql_references/functions/alphabeticallistfunctions/edit_distance.htm#EDIT_DISTANCE
             exp.Levenshtein: unsupported_args("ins_cost", "del_cost", "sub_cost", "max_dist")(
                 rename_func("EDIT_DISTANCE")
             ),
             # https://docs.exasol.com/db/latest/sql_references/functions/alphabeticallistfunctions/mod.htm
             exp.Mod: rename_func("MOD"),
+            # https://docs.exasol.com/db/latest/sql_references/functions/alphabeticallistfunctions/regexp_substr.htm
+            exp.RegexpExtract: unsupported_args("parameters", "group")(
+                rename_func("REGEXP_SUBSTR")
+            ),
             # https://docs.exasol.com/db/latest/sql_references/functions/alphabeticallistfunctions/regexp_replace.htm
             exp.RegexpReplace: unsupported_args("modifiers")(rename_func("REGEXP_REPLACE")),
             # https://docs.exasol.com/db/latest/sql_references/functions/alphabeticallistfunctions/var_pop.htm
@@ -175,6 +215,7 @@ class Exasol(Dialect):
             exp.TsOrDsToDate: lambda self, e: self.func("TO_DATE", e.this, self.format_time(e)),
             exp.TimeToStr: lambda self, e: self.func("TO_CHAR", e.this, self.format_time(e)),
             exp.TimeStrToTime: timestrtotime_sql,
+            exp.TimestampTrunc: timestamptrunc_sql(),
             exp.StrToTime: lambda self, e: self.func("TO_DATE", e.this, self.format_time(e)),
             exp.CurrentUser: lambda *_: "CURRENT_USER",
             exp.AtTimeZone: lambda self, e: self.func(

sqlglot/dialects/materialize.py

@@ -51,6 +51,7 @@ class Materialize(Postgres):
 
     class Generator(Postgres.Generator):
         SUPPORTS_CREATE_TABLE_LIKE = False
+        SUPPORTS_BETWEEN_FLAGS = False
 
         TRANSFORMS = {
             **Postgres.Generator.TRANSFORMS,

sqlglot/dialects/oracle.py

@@ -52,10 +52,6 @@ class Oracle(Dialect):
     # https://docs.oracle.com/database/121/SQLRF/sql_elements004.htm#SQLRF00212
     # https://docs.python.org/3/library/datetime.html#strftime-and-strptime-format-codes
     TIME_MAPPING = {
-        "AM": "%p",  # Meridian indicator with or without periods
-        "A.M.": "%p",  # Meridian indicator with or without periods
-        "PM": "%p",  # Meridian indicator with or without periods
-        "P.M.": "%p",  # Meridian indicator with or without periods
         "D": "%u",  # Day of week (1-7)
         "DAY": "%A",  # name of day
         "DD": "%d",  # day of month (1-31)

sqlglot/dialects/postgres.py

@@ -274,8 +274,6 @@ class Postgres(Dialect):
     TABLESAMPLE_SIZE_IS_PERCENT = True
 
     TIME_MAPPING = {
-        "AM": "%p",
-        "PM": "%p",
         "d": "%u",  # 1-based day of week
         "D": "%u",  # 1-based day of week
         "dd": "%d",  # day of month

sqlglot/dialects/redshift.py

@@ -161,6 +161,7 @@ class Redshift(Postgres):
         SUPPORTS_MEDIAN = True
         ALTER_SET_TYPE = "TYPE"
         SUPPORTS_DECODE_CASE = True
+        SUPPORTS_BETWEEN_FLAGS = False
 
         # Redshift doesn't have `WITH` as part of their with_properties so we remove it
         WITH_PROPERTIES_PREFIX = " "

sqlglot/dialects/risingwave.py

@@ -61,6 +61,7 @@ class RisingWave(Postgres):
 
     class Generator(Postgres.Generator):
         LOCKING_READS_SUPPORTED = False
+        SUPPORTS_BETWEEN_FLAGS = False
 
         TRANSFORMS = {
             **Postgres.Generator.TRANSFORMS,