dagster-k8s 0.26.18rc2__tar.gz → 0.26.20__tar.gz

{dagster_k8s-0.26.18rc2/dagster_k8s.egg-info → dagster_k8s-0.26.20}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dagster-k8s
-Version: 0.26.18rc2
+Version: 0.26.20
 Summary: A Dagster integration for k8s
 Home-page: https://github.com/dagster-io/dagster/tree/master/python_modules/libraries/dagster-k8s
 Author: Dagster Labs
@@ -14,7 +14,7 @@ Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.9,<3.13
 License-File: LICENSE
-Requires-Dist: dagster==1.10.18rc2
+Requires-Dist: dagster==1.10.20
 Requires-Dist: kubernetes<33
 Requires-Dist: google-auth!=2.23.1
 Dynamic: author

{dagster_k8s-0.26.18rc2 → dagster_k8s-0.26.20}/dagster_k8s/executor.py

@@ -37,6 +37,7 @@ from dagster_k8s.job import (
     get_user_defined_k8s_config,
 )
 from dagster_k8s.launcher import K8sRunLauncher
+from dagster_k8s.utils import get_deployment_id_label
 
 _K8S_EXECUTOR_CONFIG_SCHEMA = merge_dicts(
     DagsterK8sJobConfig.config_type_job(),
@@ -297,6 +298,9 @@ class K8sStepHandler(StepHandler):
             labels["dagster/code-location"] = (
                 run.remote_job_origin.repository_origin.code_location_origin.location_name
             )
+        deployment_name_env_var = get_deployment_id_label(container_context.run_k8s_config)
+        if deployment_name_env_var:
+            labels["dagster/deployment-name"] = deployment_name_env_var
         job = construct_dagster_k8s_job(
             job_config=job_config,
             args=args,

{dagster_k8s-0.26.18rc2 → dagster_k8s-0.26.20}/dagster_k8s/launcher.py

@@ -18,6 +18,7 @@ from dagster._utils.error import serializable_error_info_from_exc_info
 from dagster_k8s.client import DagsterKubernetesClient
 from dagster_k8s.container_context import K8sContainerContext
 from dagster_k8s.job import DagsterK8sJobConfig, construct_dagster_k8s_job, get_job_name_from_run_id
+from dagster_k8s.utils import get_deployment_id_label
 
 
 class K8sRunLauncher(RunLauncher, ConfigurableClass):
@@ -231,6 +232,9 @@ class K8sRunLauncher(RunLauncher, ConfigurableClass):
             "dagster/job": job_origin.job_name,
             "dagster/run-id": run.run_id,
         }
+        deployment_name_env_var = get_deployment_id_label(user_defined_k8s_config)
+        if deployment_name_env_var:
+            labels["dagster/deployment-name"] = deployment_name_env_var
         if run.remote_job_origin:
             labels["dagster/code-location"] = (
                 run.remote_job_origin.repository_origin.code_location_origin.location_name

{dagster_k8s-0.26.18rc2 → dagster_k8s-0.26.20}/dagster_k8s/ops/k8s_job_op.py

@@ -313,6 +313,14 @@ def execute_k8s_job(
         labels["dagster/code-location"] = (
             context.dagster_run.remote_job_origin.repository_origin.code_location_origin.location_name
         )
+    env = user_defined_k8s_config.container_config.get("env")
+    deployment_name_env_var = (
+        next((entry for entry in env if entry["name"] == "DAGSTER_CLOUD_DEPLOYMENT_NAME"), None)
+        if env
+        else None
+    )
+    if deployment_name_env_var:
+        labels["dagster/deployment-name"] = deployment_name_env_var["value"]
 
     job = construct_dagster_k8s_job(
         job_config=k8s_job_config,

{dagster_k8s-0.26.18rc2 → dagster_k8s-0.26.20}/dagster_k8s/utils.py

@@ -1,7 +1,11 @@
 import re
+from typing import TYPE_CHECKING
 
 from dagster import __version__ as dagster_version
 
+if TYPE_CHECKING:
+    from dagster_k8s.job import UserDefinedDagsterK8sConfig
+
 
 def sanitize_k8s_label(label_name: str):
     # Truncate too long label values to fit into 63-characters limit and avoid invalid characters.
@@ -18,3 +22,13 @@ def get_common_labels():
         "app.kubernetes.io/version": sanitize_k8s_label(dagster_version),
         "app.kubernetes.io/part-of": "dagster",
     }
+
+
+def get_deployment_id_label(user_defined_k8s_config: "UserDefinedDagsterK8sConfig"):
+    env = user_defined_k8s_config.container_config.get("env")
+    deployment_name_env_var = (
+        next((entry for entry in env if entry["name"] == "DAGSTER_CLOUD_DEPLOYMENT_NAME"), None)
+        if env
+        else None
+    )
+    return deployment_name_env_var["value"] if deployment_name_env_var else None

dagster_k8s-0.26.20/dagster_k8s/version.py

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

{dagster_k8s-0.26.18rc2 → dagster_k8s-0.26.20/dagster_k8s.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dagster-k8s
-Version: 0.26.18rc2
+Version: 0.26.20
 Summary: A Dagster integration for k8s
 Home-page: https://github.com/dagster-io/dagster/tree/master/python_modules/libraries/dagster-k8s
 Author: Dagster Labs
@@ -14,7 +14,7 @@ Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.9,<3.13
 License-File: LICENSE
-Requires-Dist: dagster==1.10.18rc2
+Requires-Dist: dagster==1.10.20
 Requires-Dist: kubernetes<33
 Requires-Dist: google-auth!=2.23.1
 Dynamic: author

{dagster_k8s-0.26.18rc2 → dagster_k8s-0.26.20}/dagster_k8s.egg-info/requires.txt

@@ -1,3 +1,3 @@
-dagster==1.10.18rc2
+dagster==1.10.20
 kubernetes<33
 google-auth!=2.23.1

{dagster_k8s-0.26.18rc2 → dagster_k8s-0.26.20}/setup.py

@@ -42,7 +42,7 @@ setup(
     include_package_data=True,
     python_requires=">=3.9,<3.13",
     install_requires=[
-        "dagster==1.10.18rc2",
+        "dagster==1.10.20",
         f"kubernetes<{KUBERNETES_VERSION_UPPER_BOUND}",
         # exclude a google-auth release that added an overly restrictive urllib3 pin that confuses dependency resolvers
         "google-auth!=2.23.1",

dagster_k8s-0.26.18rc2/dagster_k8s/version.py

@@ -1 +0,0 @@
-__version__ = "0.26.18rc2"