dagster-shared 1.11.9__tar.gz → 1.11.10__tar.gz

{dagster_shared-1.11.9/dagster_shared.egg-info → dagster_shared-1.11.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dagster_shared
-Version: 1.11.9
+Version: 1.11.10
 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

{dagster_shared-1.11.9 → dagster_shared-1.11.10}/dagster_shared/error.py

@@ -1,4 +1,5 @@
 # mypy does not support recursive types, so "cause" has to be typed `Any`
+import os
 import sys
 import traceback
 from collections.abc import Sequence
@@ -102,14 +103,14 @@ class SerializableErrorInfo(
 
 
 DAGSTER_FRAMEWORK_SUBSTRINGS = [
-    "/site-packages/dagster",
-    "/python_modules/dagster",
-    "/python_modules/libraries/dagster",
+    os.sep + os.path.join("site-packages", "dagster"),
+    os.sep + os.path.join("python_modules", "dagster"),
+    os.sep + os.path.join("python_modules", "libraries", "dagster"),
 ]
 
 IMPORT_MACHINERY_SUBSTRINGS = [
-    "importlib/__init__.py",
-    "importlib/metadata/__init__.py",
+    os.path.join("importlib", "__init__.py"),
+    os.path.join("importlib", "metadata", "__init__.py"),
     "importlib._bootstrap",
 ]
 

{dagster_shared-1.11.9 → dagster_shared-1.11.10}/dagster_shared/serdes/objects/__init__.py

@@ -1,7 +1,3 @@
-from dagster_shared.serdes.objects.defs_state_info import (
-    DefsKeyStateInfo as DefsKeyStateInfo,
-    DefsStateInfo as DefsStateInfo,
-)
 from dagster_shared.serdes.objects.package_entry import (
     ComponentFeatureData as ComponentFeatureData,
     EnvRegistryKey as EnvRegistryKey,

dagster_shared-1.11.10/dagster_shared/serdes/objects/models/__init__.py

@@ -0,0 +1,4 @@
+from dagster_shared.serdes.objects.models.defs_state_info import (
+    DefsKeyStateInfo as DefsKeyStateInfo,
+    DefsStateInfo as DefsStateInfo,
+)

dagster_shared-1.11.10/dagster_shared/serdes/objects/models/defs_state_info.py

@@ -0,0 +1,41 @@
+import time
+from collections.abc import Mapping
+from typing import Optional
+
+from dagster_shared.dagster_model import DagsterModel
+from dagster_shared.serdes import whitelist_for_serdes
+
+
+@whitelist_for_serdes
+class DefsKeyStateInfo(DagsterModel):
+    """Records information about the version of the state for a given defs key."""
+
+    version: str
+    create_timestamp: float
+
+
+@whitelist_for_serdes
+class DefsStateInfo(DagsterModel):
+    """All of the information about the state version that will be used to load a given code location."""
+
+    info_mapping: Mapping[str, Optional[DefsKeyStateInfo]]
+
+    @staticmethod
+    def empty() -> "DefsStateInfo":
+        return DefsStateInfo(info_mapping={})
+
+    @staticmethod
+    def add_version(
+        current_info: Optional["DefsStateInfo"], key: str, version: Optional[str]
+    ) -> "DefsStateInfo":
+        new_info = (
+            DefsKeyStateInfo(version=version, create_timestamp=time.time()) if version else None
+        )
+        if current_info is None:
+            return DefsStateInfo(info_mapping={key: new_info})
+        else:
+            return DefsStateInfo(info_mapping={**current_info.info_mapping, key: new_info})
+
+    def get_version(self, key: str) -> Optional[str]:
+        info = self.info_mapping.get(key)
+        return info.version if info else None

dagster_shared-1.11.10/dagster_shared/version.py

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

{dagster_shared-1.11.9 → dagster_shared-1.11.10/dagster_shared.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dagster_shared
-Version: 1.11.9
+Version: 1.11.10
 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

{dagster_shared-1.11.9 → dagster_shared-1.11.10}/dagster_shared.egg-info/SOURCES.txt

@@ -39,8 +39,9 @@ dagster_shared/serdes/serdes.py
 dagster_shared/serdes/utils.py
 dagster_shared/serdes/objects/__init__.py
 dagster_shared/serdes/objects/definition_metadata.py
-dagster_shared/serdes/objects/defs_state_info.py
 dagster_shared/serdes/objects/package_entry.py
+dagster_shared/serdes/objects/models/__init__.py
+dagster_shared/serdes/objects/models/defs_state_info.py
 dagster_shared/seven/__init__.py
 dagster_shared/seven/abc.py
 dagster_shared/seven/json.py

dagster_shared-1.11.9/dagster_shared/serdes/objects/defs_state_info.py

@@ -1,55 +0,0 @@
-import time
-from collections.abc import Mapping
-from typing import Any, Optional
-
-from dagster_shared.record import record
-from dagster_shared.serdes import whitelist_for_serdes
-
-
-@whitelist_for_serdes
-@record
-class DefsKeyStateInfo:
-    """Records information about the version of the state for a given defs key."""
-
-    version: str
-    create_timestamp: float
-
-    @staticmethod
-    def from_dict(val: dict[str, Any]) -> "DefsKeyStateInfo":
-        return DefsKeyStateInfo(version=val["version"], create_timestamp=val["create_timestamp"])
-
-    def to_dict(self) -> dict[str, Any]:
-        return {"version": self.version, "create_timestamp": self.create_timestamp}
-
-
-@whitelist_for_serdes
-@record
-class DefsStateInfo:
-    """All of the information about the state version that will be used to load a given code location."""
-
-    info_mapping: Mapping[str, DefsKeyStateInfo]
-
-    @staticmethod
-    def add_version(
-        current_info: Optional["DefsStateInfo"], key: str, version: str
-    ) -> "DefsStateInfo":
-        new_info = DefsKeyStateInfo(version=version, create_timestamp=time.time())
-        if current_info is None:
-            return DefsStateInfo(info_mapping={key: new_info})
-        else:
-            return DefsStateInfo(info_mapping={**current_info.info_mapping, key: new_info})
-
-    @staticmethod
-    def from_dict(val: dict[str, Any]) -> "DefsStateInfo":
-        """Used for converting from the user-facing dict representation."""
-        return DefsStateInfo(
-            info_mapping={key: DefsKeyStateInfo.from_dict(info) for key, info in val.items()}
-        )
-
-    def to_dict(self) -> dict[str, Any]:
-        """Used for converting to the user-facing dict representation."""
-        return {key: info.to_dict() for key, info in self.info_mapping.items()}
-
-    def get_version(self, key: str) -> Optional[str]:
-        info = self.info_mapping.get(key)
-        return info.version if info else None

dagster_shared-1.11.9/dagster_shared/version.py

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