PR feedback

Author: erindru

sqlmesh/core/context.py

@@ -120,7 +120,14 @@
 from sqlmesh.utils import UniqueKeyDict, Verbosity
 from sqlmesh.utils.concurrency import concurrent_apply_to_values
 from sqlmesh.utils.dag import DAG
-from sqlmesh.utils.date import TimeLike, now_ds, to_timestamp, format_tz_datetime, now_timestamp
+from sqlmesh.utils.date import (
+    TimeLike,
+    now_ds,
+    to_timestamp,
+    format_tz_datetime,
+    now_timestamp,
+    now,
+)
 from sqlmesh.utils.errors import (
     CircuitBreakerError,
     ConfigError,
@@ -1517,7 +1524,7 @@ def plan_builder(
                 max_interval_end_per_model,
                 backfill_models,
                 modified_model_names,
-                execution_time,
+                execution_time or now(),
             )
 
             # Refresh snapshot intervals to ensure that they are up to date with values reflected in the max_interval_end_per_model.

sqlmesh/core/plan/builder.py

@@ -42,6 +42,7 @@
     yesterday_ds,
     to_timestamp,
     time_like_to_str,
+    is_relative,
 )
 from sqlmesh.utils.errors import NoChangesPlanError, PlanError
 
@@ -139,7 +140,7 @@ def __init__(
         self._include_unmodified = include_unmodified
         self._restate_models = set(restate_models) if restate_models is not None else None
         self._effective_from = effective_from
-        self._execution_time = execution_time
+        self._execution_time = execution_time or now()
         self._backfill_models = backfill_models
         self._end = end or default_end
         self._apply = apply
@@ -176,20 +177,18 @@ def is_start_and_end_allowed(self) -> bool:
 
     @property
     def start(self) -> t.Optional[TimeLike]:
-        if self._start and self._execution_time:
+        if self._start and is_relative(self._start):
+            # only do this for relative expressions otherwise inclusive date strings like '2020-01-01' can be turned into exclusive timestamps eg '2020-01-01 00:00:00'
             return to_datetime(self._start, relative_base=to_datetime(self._execution_time))
         return self._start
 
     @property
     def end(self) -> t.Optional[TimeLike]:
-        if self._end and self._execution_time:
+        if self._end and is_relative(self._end):
+            # only do this for relative expressions otherwise inclusive date strings like '2020-01-01' can be turned into exclusive timestamps eg '2020-01-01 00:00:00'
             return to_datetime(self._end, relative_base=to_datetime(self._execution_time))
         return self._end
 
-    @property
-    def execution_time(self) -> TimeLike:
-        return self._execution_time or now()
-
     def set_start(self, new_start: TimeLike) -> PlanBuilder:
         self._start = new_start
         self.override_start = True
@@ -274,7 +273,8 @@ def build(self) -> Plan:
         )
 
         restatements = self._build_restatements(
-            dag, earliest_interval_start(self._context_diff.snapshots.values(), self.execution_time)
+            dag,
+            earliest_interval_start(self._context_diff.snapshots.values(), self._execution_time),
         )
         models_to_backfill = self._build_models_to_backfill(dag, restatements)
 
@@ -307,7 +307,7 @@ def build(self) -> Plan:
             selected_models_to_backfill=self._backfill_models,
             models_to_backfill=models_to_backfill,
             effective_from=self._effective_from,
-            execution_time=self.execution_time,
+            execution_time=self._execution_time,
             end_bounded=self._end_bounded,
             ensure_finalized_snapshots=self._ensure_finalized_snapshots,
             user_provided_flags=self._user_provided_flags,
@@ -764,9 +764,9 @@ def _ensure_valid_date_range(self) -> None:
                 )
 
         if end := self.end:
-            if to_datetime(end) > to_datetime(self.execution_time):
+            if to_datetime(end) > to_datetime(self._execution_time):
                 raise PlanError(
-                    f"Plan end date: '{time_like_to_str(end)}' cannot be in the future (execution time: '{time_like_to_str(self.execution_time)}')"
+                    f"Plan end date: '{time_like_to_str(end)}' cannot be in the future (execution time: '{time_like_to_str(self._execution_time)}')"
                 )
 
     def _ensure_no_forward_only_revert(self) -> None:

sqlmesh/core/plan/definition.py

@@ -83,8 +83,9 @@ def start(self) -> TimeLike:
     def end(self) -> TimeLike:
         return self.provided_end or self.execution_time
 
-    @property
+    @cached_property
     def execution_time(self) -> TimeLike:
+        # note: property is cached so that it returns a consistent timestamp for now()
         return self.execution_time_ or now()
 
     @property

sqlmesh/utils/date.py

@@ -373,6 +373,16 @@ def is_categorical_relative_expression(expression: str) -> bool:
     return not any(k in TIME_UNITS for k in grain_kwargs)
 
 
+def is_relative(value: TimeLike) -> bool:
+    """
+    Tests a TimeLike object to see if it is a relative expression, eg '1 week ago' as opposed to an absolute timestamp
+    """
+    if isinstance(value, str):
+        return is_categorical_relative_expression(value)
+
+    return False
+
+
 def to_time_column(
     time_column: t.Union[TimeLike, exp.Null],
     time_column_type: exp.DataType,

tests/integrations/github/cicd/test_github_controller.py

@@ -19,6 +19,7 @@
     GithubCheckStatus,
     MergeStateStatus,
 )
+from sqlmesh.utils.date import to_datetime
 from tests.integrations.github.cicd.conftest import MockIssueComment
 
 pytestmark = pytest.mark.github
@@ -249,7 +250,7 @@ def test_pr_plan_auto_categorization(github_client, make_controller):
     assert not controller._context.apply.called
     assert controller._context._run_plan_tests.call_args == call(skip_tests=True)
     assert controller._pr_plan_builder._categorizer_config == custom_categorizer_config
-    assert controller.pr_plan.start == default_start
+    assert controller.pr_plan.start == to_datetime(default_start)
 
 
 def test_prod_plan(github_client, make_controller):

tests/utils/test_date.py

@@ -12,6 +12,7 @@
     date_dict,
     format_tz_datetime,
     is_categorical_relative_expression,
+    is_relative,
     make_inclusive,
     to_datetime,
     to_time_column,
@@ -324,3 +325,17 @@ def test_format_tz_datetime():
     test_datetime = to_datetime("2020-01-01 00:00:00")
     assert format_tz_datetime(test_datetime) == "2020-01-01 12:00AM UTC"
     assert format_tz_datetime(test_datetime, format_string=None) == "2020-01-01 00:00:00+00:00"
+
+
+def test_is_relative():
+    assert is_relative("1 week ago")
+    assert is_relative("1 week")
+    assert is_relative("1 day ago")
+    assert is_relative("yesterday")
+
+    assert not is_relative("2024-01-01")
+    assert not is_relative("2024-01-01 01:02:03")
+    assert not is_relative(to_datetime("2024-01-01 01:02:03"))
+    assert not is_relative(to_timestamp("2024-01-01 01:02:03"))
+    assert not is_relative(to_datetime("1 week ago"))
+    assert not is_relative(to_timestamp("1 day ago"))