dagster-shared 1.11.14__py3-none-any.whl → 1.11.16__py3-none-any.whl

dagster_shared/record/__init__.py

@@ -628,12 +628,12 @@ def __defaults_new__(cls{kw_args_str}):
     """
 
 
-def _banned_iter(*args, **kwargs):
-    raise Exception("Iteration is not allowed on `@record`s.")
+def _banned_iter(self, *args, **kwargs):
+    raise Exception(f"Iteration is not allowed on `@record` {self.__class__.__name__}.")
 
 
-def _banned_idx(*args, **kwargs):
-    raise Exception("Index access is not allowed on `@record`s.")
+def _banned_idx(self, *args, **kwargs):
+    raise Exception(f"Index access is not allowed on `@record` {self.__class__.__name__}.")
 
 
 def _true(_):

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/version.py

@@ -1 +1 @@
-__version__ = "1.11.14"
+__version__ = "1.11.16"

{dagster_shared-1.11.14.dist-info → dagster_shared-1.11.16.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dagster_shared
-Version: 1.11.14
+Version: 1.11.16
 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.dist-info → dagster_shared-1.11.16.dist-info}/RECORD

@@ -5,7 +5,7 @@ dagster_shared/match.py,sha256=o2ZzDuNEsDpX6rF6CV3GSVrfdLIpnJvr8B75bSQ_T40,3433
 dagster_shared/merger.py,sha256=c9cJTwHtgMXgR9MtB-KwZEMtQSXn13rF08F2ID5aDTg,1758
 dagster_shared/modules.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dagster_shared/py.typed,sha256=mDShSrm8qg9qjacQc2F-rI8ATllqP6EdgHuEYxuCXZ0,7
-dagster_shared/version.py,sha256=WTSxgnRywP6HIZz5yh04W6BA-_H9MLGBIFoX8gBMQUk,24
+dagster_shared/version.py,sha256=-NeVBKZrjnuM9q744VBGeKsAho-QI1WcQVgySxT1kic,24
 dagster_shared/check/README.md,sha256=UlxRVMWJEoVAlaHWDvhobUpuNQGBG5-ddqZ6OIfLYnM,1352
 dagster_shared/check/__init__.py,sha256=CHSOqJpaeWr6skJT7jD8HnN5s4bdnoX4x94sGN55KqA,3482
 dagster_shared/check/builder.py,sha256=SVu2OSL206R1HVP4qiGSA1WHt4Oyb-bDzGh9OvPlhJ8,17427
@@ -20,7 +20,7 @@ dagster_shared/plus/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSu
 dagster_shared/plus/config.py,sha256=0kAWocZFYh9NmCoOT348YUg0vQe9SEpJ_Ze9R3uyR-w,5756
 dagster_shared/plus/config_utils.py,sha256=eVvahsrGQdcYL1H3_63kXO2NKsVPtWXhlGFajvJPQ5w,6147
 dagster_shared/plus/login_server.py,sha256=zRc85NUHtxpgXGfzV1d4hq-n9ZdXTBiSmagUebwxqdE,3630
-dagster_shared/record/__init__.py,sha256=L1k0imyJZ2AQXyQJPBouFsRHq1WaZsM-NEy7Ud8nVGs,22346
+dagster_shared/record/__init__.py,sha256=khKW4ob9kiLCsC8g-NNkVwmrpj2Fjhv2RDNQqRNYF54,22410
 dagster_shared/scaffold/__init__.py,sha256=xb5-ZdbZ5wSYYJy8VA1KGDnPodM6bKTCSBPS85Wqi_w,3759
 dagster_shared/serdes/__init__.py,sha256=BInlrdN_14uwPz0t32Dl1VH4D-7_rG1vUfZ0rw5DQc0,649
 dagster_shared/serdes/errors.py,sha256=qg_VZqtuYAlAXIjOp96bGs_z309ifDoHuT4sVnxJ6nA,142
@@ -30,7 +30,7 @@ dagster_shared/serdes/objects/__init__.py,sha256=iSlP-BnIyeBaNp9kcxrJHtGO5cbpodS
 dagster_shared/serdes/objects/definition_metadata.py,sha256=ECd9dCV7EULPQlWRsvlO_eauYjkM58VhGnMPPxjGfew,2000
 dagster_shared/serdes/objects/package_entry.py,sha256=Z5PbxwvEboSlRMYJ06X2Qv_HXPsHQVaGCYLTw3sL9xQ,6957
 dagster_shared/serdes/objects/models/__init__.py,sha256=k4SeT33CVm58Gwn87BKa85nrqCwfp6cUnUMiePLrkBI,147
-dagster_shared/serdes/objects/models/defs_state_info.py,sha256=aPmiEmpQ6hehCgGnYRhdnIyE-iRZk59HwA1YjPsWrAM,1781
+dagster_shared/serdes/objects/models/defs_state_info.py,sha256=5EmJELrgm-wnWLA3eq6xz1vRKXK_imiAjPaMSMihB-Y,2826
 dagster_shared/seven/__init__.py,sha256=sWCH8lRg2Y4IPyW6wo_c963RfhwVeO8tpSYsJAIR6JU,8432
 dagster_shared/seven/abc.py,sha256=vaqd-UGzvqAADutqU4O0V7-Hihe0vQNerj2SlCPJMFQ,553
 dagster_shared/seven/json.py,sha256=TElHKCmD71QJSDKvFP4e_vLMoSAFsAllPAbmplzSN30,383
@@ -47,8 +47,8 @@ dagster_shared/utils/warnings.py,sha256=uaCngAy2aPwuuN8ktod1Z5GEIvwwBaJYMSo4inNN
 dagster_shared/yaml_utils/__init__.py,sha256=Ya7qT3aJGPC1TgD-pSkkhLX5U4QbrUkCsrUtZQeACXg,10529
 dagster_shared/yaml_utils/sample_yaml.py,sha256=YC-1h1P48F3QV-Uwo-MqzEkzlrMliXPfZM_X6ynZvTQ,5461
 dagster_shared/yaml_utils/source_position.py,sha256=yVSnHUkBYzWzLzqtyGbqRjH2b7oTeiE4K3b-uokYUIQ,9786
-dagster_shared-1.11.14.dist-info/licenses/LICENSE,sha256=hgU51ohULAFKZE2la8sartPYoUaAA14jxFAfFC4Tow0,11347
-dagster_shared-1.11.14.dist-info/METADATA,sha256=iUrWy-pkAM5BQE6ZrqT6iHiaCHekkonZx7L1xA4gMUo,1176
-dagster_shared-1.11.14.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dagster_shared-1.11.14.dist-info/top_level.txt,sha256=S8ADcO4aTW07ONhLHXDxiuSkDzucgYKOlocB2x3r_dY,15
-dagster_shared-1.11.14.dist-info/RECORD,,
+dagster_shared-1.11.16.dist-info/licenses/LICENSE,sha256=hgU51ohULAFKZE2la8sartPYoUaAA14jxFAfFC4Tow0,11347
+dagster_shared-1.11.16.dist-info/METADATA,sha256=4r8jqDY8qFRxwFSjc4BunP7M395d9Rcn0Ahqo9XzBCM,1204
+dagster_shared-1.11.16.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dagster_shared-1.11.16.dist-info/top_level.txt,sha256=S8ADcO4aTW07ONhLHXDxiuSkDzucgYKOlocB2x3r_dY,15
+dagster_shared-1.11.16.dist-info/RECORD,,