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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dagster_shared
-Version: 1.11.9
+Version: 1.11.11
 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.11}/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.11}/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.11/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.11/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.9 → dagster_shared-1.11.11}/dagster_shared/utils/cached_method.py

@@ -105,9 +105,11 @@ def cached_method(method: Callable[Concatenate[S, P], T]) -> Callable[Concatenat
                 cache_dict[method.__name__] = {}
 
             cache = cache_dict[method.__name__]
-
-            canonical_kwargs = get_canonical_kwargs(*args, **kwargs)
-            key = make_cached_method_cache_key(canonical_kwargs)
+            if not args and not kwargs:
+                key = NO_ARGS_HASH_VALUE
+            else:
+                canonical_kwargs = get_canonical_kwargs(*args, **kwargs)
+                key = make_cached_method_cache_key(canonical_kwargs)
 
             if key not in cache:
                 result = await method(self, *args, **kwargs)
@@ -128,9 +130,12 @@ def cached_method(method: Callable[Concatenate[S, P], T]) -> Callable[Concatenat
                 cache_dict[method.__name__] = {}
 
             cache = cache_dict[method.__name__]
+            if not args and not kwargs:
+                key = NO_ARGS_HASH_VALUE
+            else:
+                canonical_kwargs = get_canonical_kwargs(*args, **kwargs)
+                key = make_cached_method_cache_key(canonical_kwargs)
 
-            canonical_kwargs = get_canonical_kwargs(*args, **kwargs)
-            key = make_cached_method_cache_key(canonical_kwargs)
             if key not in cache:
                 result = method(self, *args, **kwargs)
                 cache[key] = result

{dagster_shared-1.11.9 → dagster_shared-1.11.11}/dagster_shared/utils/timing.py

@@ -1,3 +1,8 @@
+import os
+import time
+from contextlib import contextmanager
+
+
 def format_duration(milliseconds: float) -> str:
     """Given milliseconds, return human readable duration string such as:
     533ms, 2.1s, 4m52s, 34m12s, 1h4m.
@@ -31,3 +36,29 @@ def format_duration(milliseconds: float) -> str:
         hours = int(milliseconds // (1000 * 60 * 60))
         minutes = int(milliseconds % (1000 * 60 * 60) // (1000 * 60))
         return f"{hours}h{minutes}m"
+
+
+@contextmanager
+def fixed_timezone(tzname: str = "UTC"):
+    """Temporarily set the process-wide timezone to `tzname`.
+    Temporarily set the process-wide timezone to `tzname`.
+
+    Example:
+        with fixed_timezone("America/New_York"):
+            dt = datetime.fromtimestamp(0)
+            print(dt.strftime("%Y-%m-%d %H:%M %Z"))
+            # 1969-12-31 19:00 EST
+    """
+    prev = os.environ.get("TZ")
+    os.environ["TZ"] = tzname
+    if hasattr(time, "tzset"):
+        time.tzset()
+    try:
+        yield
+    finally:
+        if prev is None:
+            os.environ.pop("TZ", None)
+        else:
+            os.environ["TZ"] = prev
+        if hasattr(time, "tzset"):
+            time.tzset()

dagster_shared-1.11.11/dagster_shared/version.py

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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dagster_shared
-Version: 1.11.9
+Version: 1.11.11
 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.11}/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"