feat(launcher): add new API endpoint /v1/launcher/time-limit and update LauncherDialog

antarest/core/config.py

@@ -308,7 +308,7 @@ def _autodetect_nb_cores(cls) -> Dict[str, int]:
 
 class InvalidConfigurationError(Exception):
     """
-    Exception raised when an attempt is made to retrieve the number of cores
+    Exception raised when an attempt is made to retrieve a property
     of a launcher that doesn't exist in the configuration.
     """
 
@@ -371,6 +371,29 @@ def get_nb_cores(self, launcher: str) -> "NbCoresConfig":
             raise InvalidConfigurationError(launcher)
         return launcher_config.nb_cores
 
+    def get_time_limit(self, launcher: str) -> int:
+        """
+        Retrieve the time limit for a job of the given launcher: "local" or "slurm".
+        If "default" is specified, retrieve the configuration of the default launcher.
+
+        Args:
+            launcher: type of launcher "local", "slurm" or "default".
+
+        Returns:
+            Time limit for a job of the given launcher.
+
+        Raises:
+            InvalidConfigurationError: Exception raised when an attempt is made to retrieve
+                a property of a launcher that doesn't exist in the configuration.
+        """
+        config_map = {"local": self.local, "slurm": self.slurm}
+        config_map["default"] = config_map[self.default]
+        launcher_config = config_map.get(launcher)
+        if launcher_config is None:
+            raise InvalidConfigurationError(launcher)
+        # Default to 172800 (48 hours) for local launcher.
+        return getattr(launcher_config, "default_time_limit", 172800)
+
 
 @dataclass(frozen=True)
 class LoggingConfig:

antarest/launcher/web.py

@@ -41,6 +41,25 @@ def __init__(self, solver: str) -> None:
         )
 
 
+LauncherQuery = Query(
+    "default",
+    examples={
+        "Default launcher": {
+            "description": "Default solver (auto-detected)",
+            "value": "default",
+        },
+        "SLURM launcher": {
+            "description": "SLURM solver configuration",
+            "value": "slurm",
+        },
+        "Local launcher": {
+            "description": "Local solver configuration",
+            "value": "local",
+        },
+    },
+)
+
+
 def create_launcher_api(service: LauncherService, config: Config) -> APIRouter:
     bp = APIRouter(prefix="/v1/launcher")
 
@@ -214,25 +233,7 @@ def get_load() -> LauncherLoadDTO:
         summary="Get list of supported solver versions",
         response_model=List[str],
     )
-    def get_solver_versions(
-        solver: str = Query(
-            "default",
-            examples={
-                "Default solver": {
-                    "description": "Get the solver versions of the default configuration",
-                    "value": "default",
-                },
-                "SLURM solver": {
-                    "description": "Get the solver versions of the SLURM server if available",
-                    "value": "slurm",
-                },
-                "Local solver": {
-                    "description": "Get the solver versions of the Local server if available",
-                    "value": "local",
-                },
-            },
-        ),
-    ) -> List[str]:
+    def get_solver_versions(solver: str = LauncherQuery) -> List[str]:
         """
         Get list of supported solver versions defined in the configuration.
 
@@ -251,25 +252,7 @@ def get_solver_versions(
         summary="Retrieving Min, Default, and Max Core Count",
         response_model=Dict[str, int],
     )
-    def get_nb_cores(
-        launcher: str = Query(
-            "default",
-            examples={
-                "Default launcher": {
-                    "description": "Min, Default, and Max Core Count",
-                    "value": "default",
-                },
-                "SLURM launcher": {
-                    "description": "Min, Default, and Max Core Count",
-                    "value": "slurm",
-                },
-                "Local launcher": {
-                    "description": "Min, Default, and Max Core Count",
-                    "value": "local",
-                },
-            },
-        ),
-    ) -> Dict[str, int]:
+    def get_nb_cores(launcher: str = LauncherQuery) -> Dict[str, int]:
         """
         Retrieve the numer of cores of the launcher.
 
@@ -288,4 +271,29 @@ def get_nb_cores(
         except InvalidConfigurationError:
             raise UnknownSolverConfig(launcher)
 
+    # noinspection SpellCheckingInspection
+    @bp.get(
+        "/time-limit",
+        tags=[APITag.launcher],
+        summary="Retrieve the time limit for a job (in seconds)",
+    )
+    def get_time_limit(launcher: str = LauncherQuery) -> int:
+        """
+        Retrieve the time limit for a job (in seconds) of the given launcher: "local" or "slurm".
+
+        If a jobs exceed this time limit, SLURM kills the job and it is considered failed.
+
+        Args:
+        - `launcher`: name of the configuration to read: "slurm" or "local".
+          If "default" is specified, retrieve the configuration of the default launcher.
+
+        Returns:
+        - time limit in seconds
+        """
+        logger.info(f"Fetching the time limit for the '{launcher}' configuration")
+        try:
+            return service.config.launcher.get_time_limit(launcher)
+        except InvalidConfigurationError:
+            raise UnknownSolverConfig(launcher)
+
     return bp

tests/integration/launcher_blueprint/test_launcher_local.py

@@ -68,3 +68,57 @@ def test_get_launcher_nb_cores(
             "description": "Unknown solver configuration: 'unknown'",
             "exception": "UnknownSolverConfig",
         }
+
+    def test_get_time_limit(
+        self,
+        client: TestClient,
+        user_access_token: str,
+    ) -> None:
+        nb_cores_expected = 3600
+        res = client.get(
+            "/v1/launcher/time-limit",
+            headers={"Authorization": f"Bearer {user_access_token}"},
+        )
+        res.raise_for_status()
+        actual = res.json()
+        assert actual == nb_cores_expected
+
+        res = client.get(
+            "/v1/launcher/time-limit?launcher=default",
+            headers={"Authorization": f"Bearer {user_access_token}"},
+        )
+        res.raise_for_status()
+        actual = res.json()
+        assert actual == nb_cores_expected
+
+        res = client.get(
+            "/v1/launcher/time-limit?launcher=local",
+            headers={"Authorization": f"Bearer {user_access_token}"},
+        )
+        res.raise_for_status()
+        actual = res.json()
+        assert actual == nb_cores_expected
+
+        # Check that the endpoint raise an exception when the "slurm" launcher is requested.
+        res = client.get(
+            "/v1/launcher/time-limit?launcher=slurm",
+            headers={"Authorization": f"Bearer {user_access_token}"},
+        )
+        assert res.status_code == http.HTTPStatus.UNPROCESSABLE_ENTITY, res.json()
+        actual = res.json()
+        assert actual == {
+            "description": "Unknown solver configuration: 'slurm'",
+            "exception": "UnknownSolverConfig",
+        }
+
+        # Check that the endpoint raise an exception when an unknown launcher is requested.
+        res = client.get(
+            "/v1/launcher/time-limit?launcher=unknown",
+            headers={"Authorization": f"Bearer {user_access_token}"},
+        )
+        assert res.status_code == http.HTTPStatus.UNPROCESSABLE_ENTITY, res.json()
+        actual = res.json()
+        assert actual == {
+            "description": "Unknown solver configuration: 'unknown'",
+            "exception": "UnknownSolverConfig",
+        }

webapp/public/locales/en/main.json

@@ -564,6 +564,7 @@
   "study.error.listOutputs": "Failed to retrieve output list",
   "study.error.launcherVersions": "Failed to retrieve launcher versions",
   "study.error.launcherCores": "Failed to retrieve launcher number of cores",
+  "study.error.launcherTimeLimit": "Failed to retrieve launcher time limit",
   "study.error.fetchComments": "Failed to fetch comments",
   "study.error.commentsNotSaved": "Comments not saved",
   "study.error.studyIdCopy": "Failed to copy study ID",

webapp/public/locales/fr/main.json

@@ -564,6 +564,7 @@
   "study.error.listOutputs": "Échec de la récupération des sorties",
   "study.error.launcherVersions": "Échec lors de la récupération des versions du launcher",
   "study.error.launcherCores": "Échec lors de la récupération du nombre de cœurs du launcher",
+  "study.error.launcherTimeLimit": "Échec lors de la récupération de la limite de temps du launcher",
   "study.error.fetchComments": "Échec lors de la récupération des commentaires",
   "study.error.commentsNotSaved": "Erreur lors de l'enregistrement des commentaires",
   "study.error.studyIdCopy": "Erreur lors de la copie de l'identifiant de l'étude",