dagster-shared 1.12.5__py3-none-any.whl → 1.12.7__py3-none-any.whl

dagster_shared/dagster_model/pydantic_compat_layer.py

@@ -60,6 +60,10 @@ class ModelFieldCompat:
         if hasattr(self.field, "discriminator"):
             return self.field.discriminator if hasattr(self.field, "discriminator") else None
 
+    @property
+    def json_schema_extra(self) -> Optional[dict[str, Any]]:
+        return getattr(self.field, "json_schema_extra", None)
+
 
 def model_fields(model: type[BaseModel]) -> dict[str, ModelFieldCompat]:
     """Returns a dictionary of fields for a given pydantic model, wrapped

dagster_shared/record/__init__.py

@@ -32,7 +32,7 @@ _NAMED_TUPLE_BASE_NEW_FIELD = "__nt_new__"
 _REMAPPING_FIELD = "__field_remap__"
 _ORIGINAL_CLASS_FIELD = "__original_class__"
 _KW_ONLY_FIELD = "__kw_only__"
-
+_REPLACE_NEW_FIELD = "__replace_new__"
 
 _sample_nt = namedtuple("_canary", "x")
 # use a sample to avoid direct private imports (_collections._tuplegetter)
@@ -173,6 +173,7 @@ def _namedtuple_record_transform(
         _REMAPPING_FIELD: field_to_new_mapping or {},
         _ORIGINAL_CLASS_FIELD: cls,
         _KW_ONLY_FIELD: kw_only,
+        _REPLACE_NEW_FIELD: generated_new if generated_new else nt_new,
         "__reduce__": _reduce,
         # functools doesn't work, so manually update_wrapper
         "__module__": cls.__module__,
@@ -489,13 +490,8 @@ def replace(obj: TVal, **kwargs) -> TVal:
     cls = obj.__class__
 
     # if we have runtime type checking, go through that to vet new field values
-    if hasattr(cls, _CHECKED_NEW):
-        target = _CHECKED_NEW
-    else:
-        target = _NAMED_TUPLE_BASE_NEW_FIELD
-
-    return getattr(cls, target)(
-        obj.__class__,
+    return getattr(cls, _REPLACE_NEW_FIELD)(
+        cls,
         **{**as_dict(obj), **kwargs},
     )
 
@@ -570,6 +566,7 @@ class JitCheckedNew:
         for c in cls.__mro__:
             if c.__new__ is self:
                 c.__new__ = compiled_fn
+        setattr(cls, _REPLACE_NEW_FIELD, compiled_fn)
 
         return compiled_fn(cls, *args, **kwargs)
 

dagster_shared/utils/__init__.py

@@ -3,8 +3,15 @@ import os
 import random
 import socket
 from collections.abc import Iterator, Mapping
+from pathlib import Path
 from types import GenericAlias
-from typing import TypeGuard, TypeVar
+from typing import Optional, TypeGuard, TypeVar
+
+try:
+    import tomllib  # pyright: ignore[reportMissingImports]
+except ImportError:
+    # Python < 3.11 fallback
+    import tomli as tomllib
 
 T = TypeVar("T")
 
@@ -88,3 +95,35 @@ def safe_is_subclass(obj, cls: type[T]) -> TypeGuard[type[T]]:
         and not isinstance(obj, GenericAlias)  # prevent exceptions on 3.9
         and issubclass(obj, cls)
     )
+
+
+def _read_pyproject_toml(directory: Path | str) -> Optional[dict]:
+    """Read and parse pyproject.toml from a directory. Returns None if not found."""
+    if isinstance(directory, str):
+        directory = Path(directory)
+    pyproject_path = directory / "pyproject.toml"
+    if not pyproject_path.exists():
+        return None
+    with open(pyproject_path, "rb") as f:
+        return tomllib.load(f)
+
+
+def find_uv_workspace_root(start_path: Path | str) -> Optional[tuple[Path, dict]]:
+    """Walk up directories to find a uv workspace root.
+
+    Returns (workspace_root_path, workspace_config) or None if not found.
+    A uv workspace is identified by [tool.uv.workspace] in pyproject.toml.
+    """
+    if isinstance(start_path, str):
+        start_path = Path(start_path)
+    current = start_path.resolve()
+    while True:
+        data = _read_pyproject_toml(current)
+        if data:
+            workspace_config = data.get("tool", {}).get("uv", {}).get("workspace", {})
+            if workspace_config:
+                return current, workspace_config
+        parent = current.parent
+        if parent == current:
+            return None
+        current = parent

dagster_shared/version.py

@@ -1 +1 @@
-__version__ = "1.12.5"
+__version__ = "1.12.7"

{dagster_shared-1.12.5.dist-info → dagster_shared-1.12.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dagster_shared
-Version: 1.12.5
+Version: 1.12.7
 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
@@ -10,6 +10,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
 License-File: LICENSE
@@ -21,7 +22,6 @@ Requires-Dist: typing_extensions<5,>=4.11.0
 Requires-Dist: tomlkit
 Provides-Extra: test
 Requires-Dist: pytest; extra == "test"
-Requires-Dist: buildkite-test-collector; extra == "test"
 Requires-Dist: flaky; extra == "test"
 Dynamic: author
 Dynamic: author-email

{dagster_shared-1.12.5.dist-info → dagster_shared-1.12.7.dist-info}/RECORD

@@ -5,7 +5,7 @@ dagster_shared/match.py,sha256=P9wQSvzb7a03jXlC-0w9msoX5bV_BE1uSvdYvavNO6M,3444
 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=4v8TGQDepA1SecAP1bl6pAHVU5O905enxAg_EyVjJZw,23
+dagster_shared/version.py,sha256=2tyzZYgcForADG3tBrJC4WeSTZ1kx005eqIpHCIE71M,23
 dagster_shared/check/README.md,sha256=UlxRVMWJEoVAlaHWDvhobUpuNQGBG5-ddqZ6OIfLYnM,1352
 dagster_shared/check/__init__.py,sha256=CHSOqJpaeWr6skJT7jD8HnN5s4bdnoX4x94sGN55KqA,3482
 dagster_shared/check/builder.py,sha256=2Ne2EM_2tGm2zYMg0vRxqC5r-KLc746UNnXe1Fk_vRc,17311
@@ -14,13 +14,13 @@ dagster_shared/check/functions.py,sha256=PhFZkMag2q5MKulQjgyIs_O9_t0XqOnzNRdQFZT
 dagster_shared/check/record.py,sha256=vzwbmxQNQl3_oG74NRKgjlIDM2xpL5ZbKtKhoFNhNq4,412
 dagster_shared/cli/__init__.py,sha256=y2q8V4gn5IRxq962uGjMkgWqpMK0B3pongr2mMB2vvg,8781
 dagster_shared/dagster_model/__init__.py,sha256=MoUreIsyzBIjn1w2noiyR0CmBg2kiP2VhUGkXvFI5E8,1296
-dagster_shared/dagster_model/pydantic_compat_layer.py,sha256=9c2BmeucYvvfnmeSffOqadKX0Kitx1rBu-WW32A116Y,2892
+dagster_shared/dagster_model/pydantic_compat_layer.py,sha256=nA3s7rOStkjNAOGNBlS3VhukBawRt1FAXkgLarDtxqY,3030
 dagster_shared/libraries/__init__.py,sha256=VBOVSB2MU2HZzUTarK-3JE-7k7xkhiggPZnaAsEcwNg,4669
 dagster_shared/plus/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 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=bOQHMcYGHKe5d3tasHNIPfXJAJUPvb7B1puZPjzdlFA,22576
+dagster_shared/record/__init__.py,sha256=yfmDqHKt1fgwAzz9ACKicF61eRqcYqBLGY49F6JtliQ,22621
 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
@@ -37,7 +37,7 @@ dagster_shared/seven/json.py,sha256=TElHKCmD71QJSDKvFP4e_vLMoSAFsAllPAbmplzSN30,
 dagster_shared/seven/temp_dir.py,sha256=L03xfE14yxlcmjCI9NrViY36r4MMjaahfexJygMPaJo,354
 dagster_shared/seven/compat/__init__.py,sha256=8BqkAab5uZGmSyvBFpePjAaXR9seNDobs3TYI45nt5k,105
 dagster_shared/telemetry/__init__.py,sha256=_SLCyZSLGFmaZLujii-z7FK642ewI78Nm4eV8n-3Xxs,11335
-dagster_shared/utils/__init__.py,sha256=GlE5oHRzfRBD1AFV8MGPgSBexYNkVwh0-ODGSQ36D9E,2962
+dagster_shared/utils/__init__.py,sha256=m4QRhlvU_XCN3BWcUr_mead-7Ui4XvQJo3SIDgljirU,4310
 dagster_shared/utils/cached_method.py,sha256=WQj_xtSTO5qIejFqxnLXpjIMRsiWe8aJ0IcZkkWBWxY,7183
 dagster_shared/utils/config.py,sha256=np_6pJ6r3xRKfV-cPeR6qVXebhj-Oznw3rhWP67miNM,3536
 dagster_shared/utils/hash.py,sha256=WA0FegijIAhMphC0gt30tTqpR6BecRk1GhIMtV71qbU,2060
@@ -47,8 +47,8 @@ dagster_shared/utils/warnings.py,sha256=yxiG8ghc3RwyomRItwIKQe6TrUdTlJAUX3BssOwe
 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.12.5.dist-info/licenses/LICENSE,sha256=hgU51ohULAFKZE2la8sartPYoUaAA14jxFAfFC4Tow0,11347
-dagster_shared-1.12.5.dist-info/METADATA,sha256=NWO8Qvp5Nwj6N58b5p6TKsZzaerPfhw9TAkmTybDD78,1153
-dagster_shared-1.12.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dagster_shared-1.12.5.dist-info/top_level.txt,sha256=S8ADcO4aTW07ONhLHXDxiuSkDzucgYKOlocB2x3r_dY,15
-dagster_shared-1.12.5.dist-info/RECORD,,
+dagster_shared-1.12.7.dist-info/licenses/LICENSE,sha256=hgU51ohULAFKZE2la8sartPYoUaAA14jxFAfFC4Tow0,11347
+dagster_shared-1.12.7.dist-info/METADATA,sha256=FMaGpagh34RNh_wzmfeEFdOIbaRaV0s0kNkikIQrdSg,1147
+dagster_shared-1.12.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dagster_shared-1.12.7.dist-info/top_level.txt,sha256=S8ADcO4aTW07ONhLHXDxiuSkDzucgYKOlocB2x3r_dY,15
+dagster_shared-1.12.7.dist-info/RECORD,,