Add Interval expression to Snowpark API

src/snowflake/snowpark/_internal/analyzer/analyzer.py

@@ -58,6 +58,7 @@
     Expression,
     FunctionExpression,
     InExpression,
+    Interval,
     Like,
     ListAgg,
     Literal,
@@ -339,6 +340,9 @@ def analyze(
                 sql = sql.upper()
             return sql
 
+        if isinstance(expr, Interval):
+            return expr.sql
+
         if isinstance(expr, Attribute):
             assert self.alias_maps_to_use is not None
             name = self.alias_maps_to_use.get(expr.expr_id, expr.name)

src/snowflake/snowpark/_internal/analyzer/expression.py

@@ -225,6 +225,47 @@ def __init__(self, value: Any, datatype: Optional[DataType] = None) -> None:
             self.datatype = infer_type(value)
 
 
+class Interval(Expression):
+    def __init__(
+        self,
+        year: int = 0,
+        quarter: int = 0,
+        month: int = 0,
+        week: int = 0,
+        day: int = 0,
+        hour: int = 0,
+        minute: int = 0,
+        second: int = 0,
+        millisecond: int = 0,
+        microsecond: int = 0,
+        nanosecond: int = 0,
+    ) -> None:
+        super().__init__()
+        self.year = year
+        self.quarter = quarter
+        self.month = month
+        self.week = week
+        self.day = day
+        self.hour = hour
+        self.minute = minute
+        self.second = second
+        self.millisecond = millisecond
+        self.microsecond = microsecond
+        self.nanosecond = nanosecond
+
+    @property
+    def sql(self) -> str:
+        return (
+            f"INTERVAL '{self.year} YEAR, {self.quarter} QUARTER, {self.month} MONTH, "
+            f"{self.week} WEEK, {self.day} DAY, {self.hour} HOUR, {self.minute} MINUTE, "
+            f"{self.second} SECOND, {self.millisecond} MILLISECOND, {self.microsecond} MICROSECOND, "
+            f"{self.nanosecond} NANOSECOND'"
+        )
+
+    def __str__(self) -> str:
+        return self.sql
+
+
 class Like(Expression):
     def __init__(self, expr: Expression, pattern: Expression) -> None:
         super().__init__(expr)

tests/integ/test_column_names.py

@@ -2,11 +2,13 @@
 # Copyright (c) 2012-2023 Snowflake Computing Inc. All rights reserved.
 #
 
+import datetime
 import math
 
 import pytest
 
-from snowflake.snowpark import Window
+from snowflake.snowpark import Column, Window
+from snowflake.snowpark._internal.analyzer.expression import Interval
 from snowflake.snowpark._internal.utils import TempObjectType, quote_name
 from snowflake.snowpark.functions import (
     any_value,
@@ -286,6 +288,42 @@ def test_literal(session):
     )
 
 
+def test_interval(session):
+    df1 = session.create_dataframe(
+        [
+            [datetime.datetime(2010, 1, 1), datetime.datetime(2011, 1, 1)],
+            [datetime.datetime(2012, 1, 1), datetime.datetime(2013, 1, 1)],
+        ],
+        schema=["a", "b"],
+    )
+    df2 = df1.select(
+        df1["a"]
+        + Column(
+            Interval(
+                year=1,
+                quarter=1,
+                month=1,
+                week=2,
+                day=2,
+                hour=2,
+                minute=3,
+                second=3,
+                millisecond=3,
+                microsecond=4,
+                nanosecond=4,
+            )
+        )
+    )
+    assert (
+        [x.name for x in df2._output]
+        == df2.columns
+        == get_metadata_names(session, df2)
+        == [
+            '"(""A"" + INTERVAL \'1 YEAR, 1 QUARTER, 1 MONTH, 2 WEEK, 2 DAY, 2 HOUR, 3 MINUTE, 3 SECOND, 3 MILLISECOND, 4 MICROSECOND, 4 NANOSECOND\')"',
+        ]
+    )
+
+
 @pytest.mark.localtest
 def test_attribute(session):
     df1 = session.create_dataframe([[1, 2]], schema=[" a", "a"])

tests/integ/test_dataframe.py

@@ -28,7 +28,7 @@
 from snowflake.connector import IntegrityError
 from snowflake.snowpark import Column, Row, Window
 from snowflake.snowpark._internal.analyzer.analyzer_utils import result_scan_statement
-from snowflake.snowpark._internal.analyzer.expression import Attribute, Star
+from snowflake.snowpark._internal.analyzer.expression import Attribute, Interval, Star
 from snowflake.snowpark._internal.utils import TempObjectType, warning_dict
 from snowflake.snowpark.exceptions import (
     SnowparkColumnException,
@@ -3440,3 +3440,47 @@ def test_drop_columns_special_names(session):
         Utils.check_answer(df2, [Row(1), Row(2)])
     finally:
         Utils.drop_table(session, table_name)
+
+
+def test_dataframe_interval_operation(session):
+    df = session.create_dataframe(
+        [
+            [datetime.datetime(2010, 1, 1), datetime.datetime(2011, 1, 1)],
+            [datetime.datetime(2012, 1, 1), datetime.datetime(2013, 1, 1)],
+        ],
+        schema=["a", "b"],
+    )
+    df2 = df.with_column(
+        "TWO_DAYS_AHEAD",
+        df["a"]
+        + Column(
+            Interval(
+                year=1,
+                quarter=1,
+                month=1,
+                week=2,
+                day=2,
+                hour=2,
+                minute=3,
+                second=3,
+                millisecond=3,
+                microsecond=4,
+                nanosecond=4,
+            )
+        ),
+    )
+    Utils.check_answer(
+        df2,
+        [
+            Row(
+                datetime.datetime(2010, 1, 1, 0, 0, 0),
+                datetime.datetime(2011, 1, 1, 0, 0, 0),
+                datetime.datetime(2011, 5, 17, 2, 3, 3, 3004),
+            ),
+            Row(
+                datetime.datetime(2012, 1, 1, 0, 0, 0),
+                datetime.datetime(2013, 1, 1, 0, 0, 0),
+                datetime.datetime(2013, 5, 17, 2, 3, 3, 3004),
+            ),
+        ],
+    )