dagster-k8s 0.26.18rc2__py3-none-any.whl → 0.26.19__py3-none-any.whl

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

@@ -1 +1 @@
-__version__ = "0.26.18rc2"
+__version__ = "0.26.19"

{dagster_k8s-0.26.18rc2.dist-info → dagster_k8s-0.26.19.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dagster-k8s
-Version: 0.26.18rc2
+Version: 0.26.19
 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.19
 Requires-Dist: kubernetes<33
 Requires-Dist: google-auth!=2.23.1
 Dynamic: author

{dagster_k8s-0.26.18rc2.dist-info → dagster_k8s-0.26.19.dist-info}/RECORD

@@ -2,20 +2,20 @@ dagster_k8s/__init__.py,sha256=1Vi9HXEW9fatVuNzWTr6Oq-JEHmyDrVkonZuELsQ1C0,751
 dagster_k8s/client.py,sha256=-Ns2saJri4Xxu-OpQ_T93e2K6aVnZvU19tj0iaMmTg8,40901
 dagster_k8s/component.py,sha256=-66T-IHktXQ8FC0INNPFWsrxpoBsEEtRAiDuhWxCTKk,1999
 dagster_k8s/container_context.py,sha256=399bdqmV3Ce6lNbfuxnhdyJSCy9keRRPJfVjDGn2U4I,22642
-dagster_k8s/executor.py,sha256=yOvA5KuGi9O5C3KGz6IP8HVPwt2CSD9NU7ITcWxe53M,14872
+dagster_k8s/executor.py,sha256=hRwzohUFfusEc84Iz9Q7N7NN5lHoCyODdARh-kpU9ec,15126
 dagster_k8s/job.py,sha256=v7RXYLZHzy2sTC7HYdz_INqbLWyZxUz9QIMwORArjJg,43065
 dagster_k8s/kubernetes_version.py,sha256=T7MDA5pOOVE_woRUkKWxtiRj4RCLAVfjSn6ZXUr6sAA,38
-dagster_k8s/launcher.py,sha256=LlbXu3WYemmt_H7kOmbbKPNLnsQHszipQz_JmTMXdEA,16691
+dagster_k8s/launcher.py,sha256=iNvrxhF7DP0sNxydg3hU9OzVGILYUcgL9LxeXGtaXqI,16936
 dagster_k8s/models.py,sha256=REKzAY103pZw0pQFqvHaM6lL02hgbi0rxmm71qds4W8,6338
 dagster_k8s/pipes.py,sha256=kb1So6erUgjTDkicALpU1T5mG9xa9n8aeI4lPUVeU2o,34031
 dagster_k8s/py.typed,sha256=la67KBlbjXN-_-DfGNcdOcjYumVpKG_Tkw-8n5dnGB4,8
 dagster_k8s/test.py,sha256=cNtcbzxytiZtd01wY5ip7KPi01y0BUQuQhohoIfAFUM,684
-dagster_k8s/utils.py,sha256=c1bHqh5f1p5RZ0JCT6WEbPPjDvbgUp3pl4nYZRaaI4s,786
-dagster_k8s/version.py,sha256=3jY34xulRH13mQY1DvMy3EUkXLpEhCUHGkuiPuWl5a4,27
+dagster_k8s/utils.py,sha256=NS9HRyy67-jF6n4loEPd6xuU33t7Y27IovK92WCdgu0,1296
+dagster_k8s/version.py,sha256=hFTnREFUNV6k-r7gNk8pbgoXT_sSxa-pE2ccr7Zysn4,24
 dagster_k8s/ops/__init__.py,sha256=ur-9GrE_DRfnsFCpYan03qOY9cWbjagC8KHZFZuiCmc,113
-dagster_k8s/ops/k8s_job_op.py,sha256=CQHtTowGGLAHH5EUGNfqZxDT1MYMqslAJ9u11SzCf-I,22036
-dagster_k8s-0.26.18rc2.dist-info/licenses/LICENSE,sha256=4lsMW-RCvfVD4_F57wrmpe3vX1xwUk_OAKKmV_XT7Z0,11348
-dagster_k8s-0.26.18rc2.dist-info/METADATA,sha256=RS5JL5vn6VKrt-rA-5pDLCVgKnrObfyJ-1mSg_fwYJw,918
-dagster_k8s-0.26.18rc2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dagster_k8s-0.26.18rc2.dist-info/top_level.txt,sha256=wFPjskoWPlk2hOLugYCaoZhSiZdUcbCA1QZe9I4dals,12
-dagster_k8s-0.26.18rc2.dist-info/RECORD,,
+dagster_k8s/ops/k8s_job_op.py,sha256=xiBxpuLnIMDd0BqhEGtfl7Z8y0bFHuFmcfep5PuyYqk,22375
+dagster_k8s-0.26.19.dist-info/licenses/LICENSE,sha256=4lsMW-RCvfVD4_F57wrmpe3vX1xwUk_OAKKmV_XT7Z0,11348
+dagster_k8s-0.26.19.dist-info/METADATA,sha256=aryVgUGF93L1c50jgVAj4UFB92edVayuk0Irau5yKac,912
+dagster_k8s-0.26.19.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dagster_k8s-0.26.19.dist-info/top_level.txt,sha256=wFPjskoWPlk2hOLugYCaoZhSiZdUcbCA1QZe9I4dals,12
+dagster_k8s-0.26.19.dist-info/RECORD,,