mainsequence 4.0.13__tar.gz → 4.0.14__tar.gz

{mainsequence-4.0.13 → mainsequence-4.0.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mainsequence
-Version: 4.0.13
+Version: 4.0.14
 Summary: Main Sequence SDK 
 Author-email: Main Sequence GmbH <dev@main-sequence.io>
 License: MainSequence GmbH SDK License Agreement

{mainsequence-4.0.13 → mainsequence-4.0.14}/mainsequence/client/models_tdag.py

@@ -2866,7 +2866,7 @@ class UpdateStatistics(BaseUpdateStatistics):
         return dimension_range_map
 
     def get_last_update_for_identity(self, identity_value):
-        return self._progress_stats()[identity_value] or self._initial_fallback_date
+        return self._progress_stats().get(identity_value)
 
     def get_earliest_update_for_identity(self, identity_value):
         stats = self._progress_stats().get(identity_value)

{mainsequence-4.0.13 → mainsequence-4.0.14}/mainsequence/tdag/data_nodes/data_nodes.py

@@ -987,8 +987,13 @@ class DataNode(DataAccessMixin, ABC):
         """
         return self.data_node_storage.sourcetableconfiguration.get_data_updates()
 
+    def prepare_update_statistics(self, update_statistics: UpdateStatistics) -> UpdateStatistics:
+        """Hook for subclasses to scope or enrich update statistics before update()."""
+        return update_statistics
+
     def _set_update_statistics(self, update_statistics: UpdateStatistics) -> UpdateStatistics:
         """Attach generic update statistics."""
+        update_statistics = self.prepare_update_statistics(update_statistics)
         self.update_statistics = update_statistics
         return update_statistics
 

{mainsequence-4.0.13 → mainsequence-4.0.14}/mainsequence.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mainsequence
-Version: 4.0.13
+Version: 4.0.14
 Summary: Main Sequence SDK 
 Author-email: Main Sequence GmbH <dev@main-sequence.io>
 License: MainSequence GmbH SDK License Agreement

{mainsequence-4.0.13 → mainsequence-4.0.14}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "mainsequence"
-version = "4.0.13"
+version = "4.0.14"
 description = "Main Sequence SDK "
 readme = "README.md"
 requires-python = ">=3.11"

{mainsequence-4.0.13 → mainsequence-4.0.14}/tests/test_update_statistics.py

@@ -5,6 +5,7 @@ import pytest
 from pydantic import ValidationError
 
 from mainsequence.client.models_tdag import UpdateStatistics
+from mainsequence.tdag.data_nodes.data_nodes import DataNode
 
 
 def _dt(hour: int) -> datetime.datetime:
@@ -81,6 +82,57 @@ def test_update_statistics_two_index_normalizes_progress():
     assert stats.index_progress["asset-3"] == _dt(3)
 
 
+def test_update_statistics_missing_identity_has_no_last_update():
+    stats = UpdateStatistics(index_progress={"asset-1": "2026-05-01T02:00:00Z"})
+    stats._initial_fallback_date = _dt(0)
+
+    assert stats.get_last_update_for_identity("asset-2") is None
+
+
+def test_base_data_node_attaches_update_statistics_without_forcing_fallback_date():
+    class ExampleNode(DataNode):
+        OFFSET_START = _dt(0)
+
+        def dependencies(self):
+            return {}
+
+        def update(self):
+            return None
+
+    node = object.__new__(ExampleNode)
+    stats = UpdateStatistics(index_progress={})
+
+    returned = node._set_update_statistics(stats)
+
+    assert returned is stats
+    assert node.update_statistics is stats
+    assert stats._initial_fallback_date is None
+
+
+def test_base_data_node_update_statistics_prepare_hook_can_set_fallback_date():
+    class ExampleNode(DataNode):
+        OFFSET_START = _dt(0)
+
+        def prepare_update_statistics(self, update_statistics):
+            update_statistics._initial_fallback_date = self.get_offset_start()
+            return update_statistics
+
+        def dependencies(self):
+            return {}
+
+        def update(self):
+            return None
+
+    node = object.__new__(ExampleNode)
+    stats = UpdateStatistics(index_progress={})
+
+    returned = node._set_update_statistics(stats)
+
+    assert returned is stats
+    assert node.update_statistics is stats
+    assert stats._initial_fallback_date == _dt(0)
+
+
 def test_update_statistics_builds_canonical_dimension_range_map_for_nested_progress():
     stats = UpdateStatistics(
         index_progress={