feat(snowflake)!: Type annotation for Snowflake LOCALTIMESTAMP

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,
@@ -1464,6 +1465,9 @@ class Generator(generator.Generator):
             exp.CurrentTimestamp: lambda self, e: self.func("SYSDATE")
             if e.args.get("sysdate")
             else self.function_fallback_sql(e),
+            exp.Localtimestamp: lambda self, e: self.func("CURRENT_TIMESTAMP", e.this)
+            if e.this
+            else "CURRENT_TIMESTAMP",
             exp.DateAdd: date_delta_sql("DATEADD"),
             exp.DateDiff: date_delta_sql("DATEDIFF"),
             exp.DatetimeAdd: date_delta_sql("TIMESTAMPADD"),

sqlglot/typing/snowflake.py

@@ -441,6 +441,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

@@ -4741,7 +4741,6 @@ def test_localtime_and_localtimestamp(self):
                     "postgres": f"SELECT {func}",
                     "duckdb": f"SELECT {func}",
                     "redshift": f"SELECT {func}",
-                    "snowflake": f"SELECT {func}",
                     "presto": f"SELECT {func}",
                     "trino": f"SELECT {func}",
                     "mysql": f"SELECT {func}",
@@ -4762,7 +4761,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

@@ -2170,6 +2170,9 @@ 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()", "CURRENT_TIMESTAMP()")
+        self.validate_identity("LOCALTIMESTAMP(3)", "CURRENT_TIMESTAMP(3)")
 
     def test_null_treatment(self):
         self.validate_all(

tests/fixtures/optimizer/annotate_functions.sql

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