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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dagster_shared
-Version: 1.11.10
+Version: 1.11.12
 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.10 → dagster_shared-1.11.12}/dagster_shared/serdes/objects/models/defs_state_info.py

@@ -1,6 +1,6 @@
 import time
 from collections.abc import Mapping
-from typing import Optional
+from typing import Any, Optional
 
 from dagster_shared.dagster_model import DagsterModel
 from dagster_shared.serdes import whitelist_for_serdes
@@ -13,6 +13,10 @@ class DefsKeyStateInfo(DagsterModel):
     version: str
     create_timestamp: float
 
+    @classmethod
+    def from_graphql(cls, data: dict[str, Any]) -> "DefsKeyStateInfo":
+        return cls(version=data["version"], create_timestamp=data["createTimestamp"])
+
 
 @whitelist_for_serdes
 class DefsStateInfo(DagsterModel):
@@ -39,3 +43,12 @@ class DefsStateInfo(DagsterModel):
     def get_version(self, key: str) -> Optional[str]:
         info = self.info_mapping.get(key)
         return info.version if info else None
+
+    @classmethod
+    def from_graphql(cls, data: dict[str, Any]) -> "DefsStateInfo":
+        return cls(
+            info_mapping={
+                val["name"]: DefsKeyStateInfo.from_graphql(val["info"])
+                for val in data["keyStateInfo"]
+            }
+        )

{dagster_shared-1.11.10 → dagster_shared-1.11.12}/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.10 → dagster_shared-1.11.12}/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.12/dagster_shared/version.py

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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dagster_shared
-Version: 1.11.10
+Version: 1.11.12
 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.10/dagster_shared/version.py

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