dagster-shared 1.11.14__tar.gz → 1.11.15__tar.gz

{dagster_shared-1.11.14/dagster_shared.egg-info → dagster_shared-1.11.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dagster_shared
-Version: 1.11.14
+Version: 1.11.15
 Summary: Shared code between dagster and dagster-dg-core.
 Home-page: https://github.com/dagster-io/dagster/tree/master/python_modules/libraries/dagster-shared
 Author: Dagster Labs
@@ -16,6 +16,7 @@ Classifier: Operating System :: OS Independent
 License-File: LICENSE
 Requires-Dist: PyYAML>=5.1
 Requires-Dist: packaging>=20.9
+Requires-Dist: platformdirs
 Requires-Dist: pydantic<3.0.0,>=2
 Requires-Dist: typing_extensions<5,>=4.11.0
 Requires-Dist: tomlkit

{dagster_shared-1.11.14 → dagster_shared-1.11.15}/dagster_shared/serdes/objects/models/defs_state_info.py

@@ -1,10 +1,20 @@
 import time
 from collections.abc import Mapping
+from enum import Enum
 from typing import Any, Optional
 
 from dagster_shared.dagster_model import DagsterModel
 from dagster_shared.serdes import whitelist_for_serdes
 
+LOCAL_STATE_VERSION = "__local__"
+CODE_SERVER_STATE_VERSION = "__code_server__"
+
+
+class DefsStateManagementType(Enum):
+    VERSIONED_STATE_STORAGE = "VERSIONED_STATE_STORAGE"
+    LOCAL_FILESYSTEM = "LOCAL_FILESYSTEM"
+    LEGACY_CODE_SERVER_SNAPSHOTS = "LEGACY_CODE_SERVER_SNAPSHOTS"
+
 
 @whitelist_for_serdes
 class DefsKeyStateInfo(DagsterModel):
@@ -17,6 +27,15 @@ class DefsKeyStateInfo(DagsterModel):
     def from_graphql(cls, data: dict[str, Any]) -> "DefsKeyStateInfo":
         return cls(version=data["version"], create_timestamp=data["createTimestamp"])
 
+    @property
+    def management_type(self) -> DefsStateManagementType:
+        if self.version == LOCAL_STATE_VERSION:
+            return DefsStateManagementType.LOCAL_FILESYSTEM
+        elif self.version == CODE_SERVER_STATE_VERSION:
+            return DefsStateManagementType.LEGACY_CODE_SERVER_SNAPSHOTS
+        else:
+            return DefsStateManagementType.VERSIONED_STATE_STORAGE
+
 
 @whitelist_for_serdes
 class DefsStateInfo(DagsterModel):
@@ -30,10 +49,15 @@ class DefsStateInfo(DagsterModel):
 
     @staticmethod
     def add_version(
-        current_info: Optional["DefsStateInfo"], key: str, version: Optional[str]
+        current_info: Optional["DefsStateInfo"],
+        key: str,
+        version: Optional[str],
+        create_timestamp: Optional[float] = None,
     ) -> "DefsStateInfo":
         new_info = (
-            DefsKeyStateInfo(version=version, create_timestamp=time.time()) if version else None
+            DefsKeyStateInfo(version=version, create_timestamp=create_timestamp or time.time())
+            if version
+            else None
         )
         if current_info is None:
             return DefsStateInfo(info_mapping={key: new_info})
@@ -44,6 +68,10 @@ class DefsStateInfo(DagsterModel):
         info = self.info_mapping.get(key)
         return info.version if info else None
 
+    def for_keys(self, keys: set[str]) -> "DefsStateInfo":
+        """Subsets the DefsStateInfo to only include the keys in the set."""
+        return DefsStateInfo(info_mapping={k: v for k, v in self.info_mapping.items() if k in keys})
+
     @classmethod
     def from_graphql(cls, data: dict[str, Any]) -> "DefsStateInfo":
         return cls(

dagster_shared-1.11.15/dagster_shared/version.py

@@ -0,0 +1 @@
+__version__ = "1.11.15"

{dagster_shared-1.11.14 → dagster_shared-1.11.15/dagster_shared.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dagster_shared
-Version: 1.11.14
+Version: 1.11.15
 Summary: Shared code between dagster and dagster-dg-core.
 Home-page: https://github.com/dagster-io/dagster/tree/master/python_modules/libraries/dagster-shared
 Author: Dagster Labs
@@ -16,6 +16,7 @@ Classifier: Operating System :: OS Independent
 License-File: LICENSE
 Requires-Dist: PyYAML>=5.1
 Requires-Dist: packaging>=20.9
+Requires-Dist: platformdirs
 Requires-Dist: pydantic<3.0.0,>=2
 Requires-Dist: typing_extensions<5,>=4.11.0
 Requires-Dist: tomlkit

{dagster_shared-1.11.14 → dagster_shared-1.11.15}/dagster_shared.egg-info/requires.txt

@@ -1,5 +1,6 @@
 PyYAML>=5.1
 packaging>=20.9
+platformdirs
 pydantic<3.0.0,>=2
 typing_extensions<5,>=4.11.0
 tomlkit

{dagster_shared-1.11.14 → dagster_shared-1.11.15}/setup.py

@@ -38,6 +38,7 @@ setup(
     install_requires=[
         "PyYAML>=5.1",
         "packaging>=20.9",
+        "platformdirs",
         "pydantic>=2,<3.0.0",
         "typing_extensions>=4.11.0,<5",
         "tomlkit",

dagster_shared-1.11.14/dagster_shared/version.py

@@ -1 +0,0 @@
-__version__ = "1.11.14"