diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-05-04 16:13:01 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-05-04 16:13:01 +0000 |
commit | a7044b672667f2a0b48bd0b326b5a55b0815ef79 (patch) | |
tree | 4fb5238d47fb4709d47f766a74b8bbaa9c6f17d8 /sqlglot/dialects/spark.py | |
parent | Releasing debian version 23.12.1-1. (diff) | |
download | sqlglot-a7044b672667f2a0b48bd0b326b5a55b0815ef79.tar.xz sqlglot-a7044b672667f2a0b48bd0b326b5a55b0815ef79.zip |
Merging upstream version 23.13.1.
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'sqlglot/dialects/spark.py')
-rw-r--r-- | sqlglot/dialects/spark.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/sqlglot/dialects/spark.py b/sqlglot/dialects/spark.py index 9bb9a5c..69c154e 100644 --- a/sqlglot/dialects/spark.py +++ b/sqlglot/dialects/spark.py @@ -5,7 +5,7 @@ import typing as t from sqlglot import exp from sqlglot.dialects.dialect import rename_func, unit_to_var from sqlglot.dialects.hive import _build_with_ignore_nulls -from sqlglot.dialects.spark2 import Spark2, temporary_storage_provider +from sqlglot.dialects.spark2 import Spark2, temporary_storage_provider, _build_as_cast from sqlglot.helper import ensure_list, seq_get from sqlglot.transforms import ( ctas_with_tmp_tables_to_create_tmp_view, @@ -63,6 +63,8 @@ class Spark(Spark2): **Spark2.Parser.FUNCTIONS, "ANY_VALUE": _build_with_ignore_nulls(exp.AnyValue), "DATEDIFF": _build_datediff, + "TIMESTAMP_LTZ": _build_as_cast("TIMESTAMP_LTZ"), + "TIMESTAMP_NTZ": _build_as_cast("TIMESTAMP_NTZ"), "TRY_ELEMENT_AT": lambda args: exp.Bracket( this=seq_get(args, 0), expressions=ensure_list(seq_get(args, 1)), safe=True ), @@ -88,6 +90,8 @@ class Spark(Spark2): exp.DataType.Type.MONEY: "DECIMAL(15, 4)", exp.DataType.Type.SMALLMONEY: "DECIMAL(6, 4)", exp.DataType.Type.UNIQUEIDENTIFIER: "STRING", + exp.DataType.Type.TIMESTAMPLTZ: "TIMESTAMP_LTZ", + exp.DataType.Type.TIMESTAMPNTZ: "TIMESTAMP_NTZ", } TRANSFORMS = { |