type anntoation for LOCALTIMESTAMP

Author: fivetran-kwoodbeck

sqlglot/dialects/snowflake.py

@@ -786,6 +786,7 @@ class Parser(parser.Parser):
             ),
             "LEN": lambda args: exp.Length(this=seq_get(args, 0), binary=True),
             "LENGTH": lambda args: exp.Length(this=seq_get(args, 0), binary=True),
+            "LOCALTIMESTAMP": exp.CurrentTimestamp.from_arg_list,
             "NULLIFZERO": _build_if_from_nullifzero,
             "OBJECT_CONSTRUCT": _build_object_construct,
             "OCTET_LENGTH": exp.ByteLength.from_arg_list,

sqlglot/typing/snowflake.py

@@ -438,6 +438,13 @@ def _annotate_str_to_time(self: TypeAnnotator, expression: exp.StrToTime) -> exp
             exp.TsOrDsToTime,
         }
     },
+    **{
+        expr_type: {"returns": exp.DataType.Type.TIMESTAMPLTZ}
+        for expr_type in {
+            exp.CurrentTimestamp,
+            exp.Localtimestamp,
+        }
+    },
     **{
         expr_type: {"returns": exp.DataType.Type.VARCHAR}
         for expr_type in {

tests/dialects/test_dialect.py

@@ -4762,7 +4762,6 @@ def test_localtime_and_localtimestamp(self):
                     "postgres": f"SELECT {func}(2)",
                     "duckdb": f"SELECT {func}(2)",
                     "redshift": f"SELECT {func}(2)",
-                    "snowflake": f"SELECT {func}(2)",
                     "presto": f"SELECT {func}(2)",
                     "trino": f"SELECT {func}(2)",
                     "mysql": f"SELECT {func}(2)",

tests/dialects/test_snowflake.py

@@ -2168,6 +2168,8 @@ def test_snowflake(self):
         self.validate_identity("SYSDATE()")
         self.validate_identity("SYSTIMESTAMP()", "CURRENT_TIMESTAMP()")
         self.validate_identity("GETDATE()", "CURRENT_TIMESTAMP()")
+        self.validate_identity("LOCALTIMESTAMP()", "CURRENT_TIMESTAMP()")
+        self.validate_identity("LOCALTIMESTAMP(3)", "CURRENT_TIMESTAMP(3)")
 
     def test_null_treatment(self):
         self.validate_all(

tests/fixtures/optimizer/annotate_functions.sql

@@ -2868,6 +2868,14 @@ INT;
 LENGTH(tbl.bin_col);
 INT;
 
+# dialect: snowflake
+LOCALTIMESTAMP();
+TIMESTAMPLTZ;
+
+# dialect: snowflake
+LOCALTIMESTAMP(3);
+TIMESTAMPLTZ;
+
 # dialect: snowflake
 OCTET_LENGTH(tbl.str_col);
 INT;