airflow-unicore-integration 0.2.2__tar.gz → 0.2.4__tar.gz

{airflow_unicore_integration-0.2.2/src/airflow_unicore_integration.egg-info → airflow_unicore_integration-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: airflow-unicore-integration
-Version: 0.2.2
+Version: 0.2.4
 Summary: Running Unicore Jobs from airflow DAGs.
 Author-email: Christian Böttcher <c.boettcher@fz-juelich.de>
 License-Expression: BSD-3-Clause

{airflow_unicore_integration-0.2.2 → airflow_unicore_integration-0.2.4}/pyproject.toml

@@ -7,7 +7,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "airflow-unicore-integration"
-version = "0.2.2"
+version = "0.2.4"
 authors = [
   { name="Christian Böttcher", email="c.boettcher@fz-juelich.de" },
 ]

{airflow_unicore_integration-0.2.2 → airflow_unicore_integration-0.2.4}/src/airflow_unicore_integration/util/job.py

@@ -26,6 +26,7 @@ class JobDescriptionGenerator:
     EXECUTOR_CONFIG_PRE_COMMANDS = "precommands"  # gets added to the unicore job description
     EXECUTOR_CONFIG_POST_COMMANDS = "postcommands"  # gets added to the unicore job descirption
     EXECUTOR_CONFIG_JOB_TYPE = "job_type"
+    EXECUTOR_CONFIG_LOGIN_NODE = "login_node"
     EXECUTOR_CONFIG_UNICORE_CONN_KEY = (
         "unicore_connection_id"  # alternative connection id for the Unicore connection to use
     )
@@ -61,6 +62,7 @@ class NaiveJobDescriptionGenerator(JobDescriptionGenerator):
         user_defined_python_env: str = workload.ti.executor_config.get(JobDescriptionGenerator.EXECUTOR_CONFIG_PYTHON_ENV_KEY, None)  # type: ignore
         user_added_post_commands: list[str] = executor_config.get(JobDescriptionGenerator.EXECUTOR_CONFIG_POST_COMMANDS, [])  # type: ignore
         user_defined_job_type: str = executor_config.get(JobDescriptionGenerator.EXECUTOR_CONFIG_JOB_TYPE, None)  # type: ignore
+        user_defined_login_node: str = executor_config.get(JobDescriptionGenerator.EXECUTOR_CONFIG_LOGIN_NODE, None)  # type: ignore
         # get local dag path from cmd and fix dag path in arguments
         dag_rel_path = str(workload.dag_rel_path)
         if dag_rel_path.startswith("DAG_FOLDER"):
@@ -76,6 +78,8 @@ class NaiveJobDescriptionGenerator(JobDescriptionGenerator):
         # set job type
         if user_defined_job_type:
             job_descr_dict["Job type"] = user_defined_job_type
+        if user_defined_login_node:
+            job_descr_dict["Login node"] = user_defined_login_node
 
         # check which python virtualenv to use
         if user_defined_python_env:

{airflow_unicore_integration-0.2.2 → airflow_unicore_integration-0.2.4/src/airflow_unicore_integration.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: airflow-unicore-integration
-Version: 0.2.2
+Version: 0.2.4
 Summary: Running Unicore Jobs from airflow DAGs.
 Author-email: Christian Böttcher <c.boettcher@fz-juelich.de>
 License-Expression: BSD-3-Clause