fractal-server 2.13.0__py3-none-any.whl → 2.14.0__py3-none-any.whl

fractal_server/app/runner/v2/runner_functions_low_level.py

@@ -2,20 +2,15 @@ import json
 import logging
 import shutil
 import subprocess  # nosec
-from pathlib import Path
-from shlex import split as shlex_split
+from shlex import split
 from typing import Any
-from typing import Optional
 
-from ..components import _COMPONENT_KEY_
-from ..exceptions import JobExecutionError
-from ..exceptions import TaskExecutionError
-from fractal_server.app.models.v2 import WorkflowTaskV2
-from fractal_server.app.runner.task_files import get_task_file_paths
+from fractal_server.app.runner.exceptions import JobExecutionError
+from fractal_server.app.runner.exceptions import TaskExecutionError
 from fractal_server.string_tools import validate_cmd
 
 
-def _call_command_wrapper(cmd: str, log_path: Path) -> None:
+def _call_command_wrapper(cmd: str, log_path: str) -> None:
     """
     Call a command and write its stdout and stderr to files
 
@@ -32,9 +27,9 @@ def _call_command_wrapper(cmd: str, log_path: Path) -> None:
         raise TaskExecutionError(f"Invalid command. Original error: {str(e)}")
 
     # Verify that task command is executable
-    if shutil.which(shlex_split(cmd)[0]) is None:
+    if shutil.which(split(cmd)[0]) is None:
         msg = (
-            f'Command "{shlex_split(cmd)[0]}" is not valid. '
+            f'Command "{split(cmd)[0]}" is not valid. '
             "Hint: make sure that it is executable."
         )
         raise TaskExecutionError(msg)
@@ -42,7 +37,7 @@ def _call_command_wrapper(cmd: str, log_path: Path) -> None:
     with open(log_path, "w") as fp_log:
         try:
             result = subprocess.run(  # nosec
-                shlex_split(cmd),
+                split(cmd),
                 stderr=fp_log,
                 stdout=fp_log,
             )
@@ -50,7 +45,7 @@ def _call_command_wrapper(cmd: str, log_path: Path) -> None:
             raise e
 
     if result.returncode > 0:
-        with log_path.open("r") as fp_stderr:
+        with open(log_path, "r") as fp_stderr:
             err = fp_stderr.read()
         raise TaskExecutionError(err)
     elif result.returncode < 0:
@@ -60,58 +55,60 @@ def _call_command_wrapper(cmd: str, log_path: Path) -> None:
 
 
 def run_single_task(
-    args: dict[str, Any],
+    # COMMON to all parallel tasks
     command: str,
-    wftask: WorkflowTaskV2,
-    workflow_dir_local: Path,
-    workflow_dir_remote: Optional[Path] = None,
-    logger_name: Optional[str] = None,
+    workflow_task_order: int,
+    workflow_task_id: int,
+    task_name: str,
+    # SPECIAL for each parallel task
+    parameters: dict[str, Any],
+    remote_files: dict[str, str],
 ) -> dict[str, Any]:
     """
-    Runs within an executor.
+    Runs within an executor (AKA on the SLURM cluster).
     """
 
-    logger = logging.getLogger(logger_name)
-    logger.debug(f"Now start running {command=}")
-
-    if not workflow_dir_remote:
-        workflow_dir_remote = workflow_dir_local
-
-    task_name = wftask.task.name
+    try:
+        args_file_remote = remote_files["args_file_remote"]
+        metadiff_file_remote = remote_files["metadiff_file_remote"]
+        log_file_remote = remote_files["log_file_remote"]
+    except KeyError:
+        raise TaskExecutionError(
+            f"Invalid {remote_files=}",
+            workflow_task_order=workflow_task_order,
+            workflow_task_id=workflow_task_id,
+            task_name=task_name,
+        )
 
-    component = args.pop(_COMPONENT_KEY_, None)
-    task_files = get_task_file_paths(
-        workflow_dir_local=workflow_dir_local,
-        workflow_dir_remote=workflow_dir_remote,
-        task_order=wftask.order,
-        task_name=task_name,
-        component=component,
-    )
+    logger = logging.getLogger(None)
+    logger.debug(f"Now start running {command=}")
 
     # Write arguments to args.json file
-    with task_files.args.open("w") as f:
-        json.dump(args, f, indent=2)
+    # NOTE: see issue 2346
+    with open(args_file_remote, "w") as f:
+        json.dump(parameters, f, indent=2)
 
     # Assemble full command
+    # NOTE: this could be assembled backend-side
     full_command = (
         f"{command} "
-        f"--args-json {task_files.args.as_posix()} "
-        f"--out-json {task_files.metadiff.as_posix()}"
+        f"--args-json {args_file_remote} "
+        f"--out-json {metadiff_file_remote}"
     )
 
     try:
         _call_command_wrapper(
             full_command,
-            log_path=task_files.log,
+            log_path=log_file_remote,
         )
     except TaskExecutionError as e:
-        e.workflow_task_order = wftask.order
-        e.workflow_task_id = wftask.id
-        e.task_name = wftask.task.name
+        e.workflow_task_order = workflow_task_order
+        e.workflow_task_id = workflow_task_id
+        e.task_name = task_name
         raise e
 
     try:
-        with task_files.metadiff.open("r") as f:
+        with open(metadiff_file_remote, "r") as f:
             out_meta = json.load(f)
     except FileNotFoundError as e:
         logger.debug(

fractal_server/app/runner/v2/submit_workflow.py

@@ -0,0 +1,358 @@
+"""
+Runner backend subsystem root V2
+
+This module is the single entry point to the runner backend subsystem V2.
+Other subsystems should only import this module and not its submodules or
+the individual backends.
+"""
+import os
+import traceback
+from pathlib import Path
+from typing import Optional
+
+from sqlalchemy.orm import Session as DBSyncSession
+
+from ....config import get_settings
+from ....logger import get_logger
+from ....logger import reset_logger_handlers
+from ....logger import set_logger
+from ....ssh._fabric import FractalSSH
+from ....syringe import Inject
+from ....utils import get_timestamp
+from ....zip_tools import _zip_folder_to_file_and_remove
+from ...db import DB
+from ...models.v2 import DatasetV2
+from ...models.v2 import JobV2
+from ...models.v2 import WorkflowV2
+from ...schemas.v2 import JobStatusTypeV2
+from ..exceptions import JobExecutionError
+from ..exceptions import TaskExecutionError
+from ..filenames import WORKFLOW_LOG_FILENAME
+from ._local import process_workflow as local_process_workflow
+from ._slurm_ssh import process_workflow as slurm_ssh_process_workflow
+from ._slurm_sudo import process_workflow as slurm_sudo_process_workflow
+from fractal_server import __VERSION__
+from fractal_server.app.models import UserSettings
+
+
+_backends = {}
+_backends["local"] = local_process_workflow
+_backends["slurm"] = slurm_sudo_process_workflow
+_backends["slurm_ssh"] = slurm_ssh_process_workflow
+
+
+def fail_job(
+    *,
+    db: DBSyncSession,
+    job: JobV2,
+    log_msg: str,
+    logger_name: str,
+    emit_log: bool = False,
+) -> None:
+    logger = get_logger(logger_name=logger_name)
+    if emit_log:
+        logger.error(log_msg)
+    reset_logger_handlers(logger)
+    job.status = JobStatusTypeV2.FAILED
+    job.end_timestamp = get_timestamp()
+    job.log = log_msg
+    db.merge(job)
+    db.commit()
+    db.close()
+    return
+
+
+def submit_workflow(
+    *,
+    workflow_id: int,
+    dataset_id: int,
+    job_id: int,
+    user_id: int,
+    user_settings: UserSettings,
+    worker_init: Optional[str] = None,
+    slurm_user: Optional[str] = None,
+    user_cache_dir: Optional[str] = None,
+    fractal_ssh: Optional[FractalSSH] = None,
+) -> None:
+    """
+    Prepares a workflow and applies it to a dataset
+
+    This function wraps the process_workflow one, which is different for each
+    backend (e.g. local or slurm backend).
+
+    Args:
+        workflow_id:
+            ID of the workflow being applied
+        dataset_id:
+            Dataset ID
+        job_id:
+            Id of the job record which stores the state for the current
+            workflow application.
+        user_id:
+            User ID.
+        worker_init:
+            Custom executor parameters that get parsed before the execution of
+            each task.
+        user_cache_dir:
+            Cache directory (namely a path where the user can write); for the
+            slurm backend, this is used as a base directory for
+            `job.working_dir_user`.
+        slurm_user:
+            The username to impersonate for the workflow execution, for the
+            slurm backend.
+    """
+    # Declare runner backend and set `process_workflow` function
+    settings = Inject(get_settings)
+    FRACTAL_RUNNER_BACKEND = settings.FRACTAL_RUNNER_BACKEND
+    logger_name = f"WF{workflow_id}_job{job_id}"
+    logger = set_logger(logger_name=logger_name)
+
+    with next(DB.get_sync_db()) as db_sync:
+        try:
+            job: Optional[JobV2] = db_sync.get(JobV2, job_id)
+            dataset: Optional[DatasetV2] = db_sync.get(DatasetV2, dataset_id)
+            workflow: Optional[WorkflowV2] = db_sync.get(
+                WorkflowV2, workflow_id
+            )
+        except Exception as e:
+            logger.error(
+                f"Error connecting to the database. Original error: {str(e)}"
+            )
+            reset_logger_handlers(logger)
+            return
+
+        if job is None:
+            logger.error(f"JobV2 {job_id} does not exist")
+            reset_logger_handlers(logger)
+            return
+        if dataset is None or workflow is None:
+            log_msg = ""
+            if not dataset:
+                log_msg += f"Cannot fetch dataset {dataset_id} from database\n"
+            if not workflow:
+                log_msg += (
+                    f"Cannot fetch workflow {workflow_id} from database\n"
+                )
+            fail_job(
+                db=db_sync, job=job, log_msg=log_msg, logger_name=logger_name
+            )
+            return
+
+        # Declare runner backend and set `process_workflow` function
+        settings = Inject(get_settings)
+        FRACTAL_RUNNER_BACKEND = settings.FRACTAL_RUNNER_BACKEND
+        try:
+            process_workflow = _backends[settings.FRACTAL_RUNNER_BACKEND]
+        except KeyError as e:
+            fail_job(
+                db=db_sync,
+                job=job,
+                log_msg=(
+                    f"Invalid {FRACTAL_RUNNER_BACKEND=}.\n"
+                    f"Original KeyError: {str(e)}"
+                ),
+                logger_name=logger_name,
+                emit_log=True,
+            )
+            return
+
+        # Define and create server-side working folder
+        WORKFLOW_DIR_LOCAL = Path(job.working_dir)
+        if WORKFLOW_DIR_LOCAL.exists():
+            fail_job(
+                db=db_sync,
+                job=job,
+                log_msg=f"Workflow dir {WORKFLOW_DIR_LOCAL} already exists.",
+                logger_name=logger_name,
+                emit_log=True,
+            )
+            return
+
+        try:
+            # Create WORKFLOW_DIR_LOCAL and define WORKFLOW_DIR_REMOTE
+            if FRACTAL_RUNNER_BACKEND == "local":
+                WORKFLOW_DIR_LOCAL.mkdir(parents=True)
+                WORKFLOW_DIR_REMOTE = WORKFLOW_DIR_LOCAL
+            elif FRACTAL_RUNNER_BACKEND == "slurm":
+                original_umask = os.umask(0)
+                WORKFLOW_DIR_LOCAL.mkdir(parents=True, mode=0o755)
+                os.umask(original_umask)
+                WORKFLOW_DIR_REMOTE = (
+                    Path(user_cache_dir) / WORKFLOW_DIR_LOCAL.name
+                )
+            elif FRACTAL_RUNNER_BACKEND == "slurm_ssh":
+                WORKFLOW_DIR_LOCAL.mkdir(parents=True)
+                WORKFLOW_DIR_REMOTE = (
+                    Path(user_settings.ssh_jobs_dir) / WORKFLOW_DIR_LOCAL.name
+                )
+            else:
+                raise ValueError(
+                    "Invalid FRACTAL_RUNNER_BACKEND="
+                    f"{settings.FRACTAL_RUNNER_BACKEND}."
+                )
+        except Exception as e:
+            error_type = type(e).__name__
+            fail_job(
+                db=db_sync,
+                job=job,
+                log_msg=(
+                    f"{error_type} error occurred while creating job folder "
+                    f"and subfolders.\nOriginal error: {str(e)}"
+                ),
+                logger_name=logger_name,
+                emit_log=True,
+            )
+            return
+
+        # After Session.commit() is called, either explicitly or when using a
+        # context manager, all objects associated with the Session are expired.
+        # https://docs.sqlalchemy.org/en/14/orm/
+        #   session_basics.html#opening-and-closing-a-session
+        # https://docs.sqlalchemy.org/en/14/orm/
+        #   session_state_management.html#refreshing-expiring
+
+        # See issue #928:
+        # https://github.com/fractal-analytics-platform/
+        #   fractal-server/issues/928
+
+        db_sync.refresh(dataset)
+        db_sync.refresh(workflow)
+        for wftask in workflow.task_list:
+            db_sync.refresh(wftask)
+
+        # Write logs
+        log_file_path = WORKFLOW_DIR_LOCAL / WORKFLOW_LOG_FILENAME
+        logger = set_logger(
+            logger_name=logger_name,
+            log_file_path=log_file_path,
+        )
+        logger.info(
+            f'Start execution of workflow "{workflow.name}"; '
+            f"more logs at {str(log_file_path)}"
+        )
+        logger.debug(f"fractal_server.__VERSION__: {__VERSION__}")
+        logger.debug(f"FRACTAL_RUNNER_BACKEND: {FRACTAL_RUNNER_BACKEND}")
+        if FRACTAL_RUNNER_BACKEND == "slurm":
+            logger.debug(f"slurm_user: {slurm_user}")
+            logger.debug(f"slurm_account: {job.slurm_account}")
+            logger.debug(f"worker_init: {worker_init}")
+        elif FRACTAL_RUNNER_BACKEND == "slurm_ssh":
+            logger.debug(f"ssh_user: {user_settings.ssh_username}")
+            logger.debug(f"base dir: {user_settings.ssh_tasks_dir}")
+            logger.debug(f"worker_init: {worker_init}")
+        logger.debug(f"job.id: {job.id}")
+        logger.debug(f"job.working_dir: {job.working_dir}")
+        logger.debug(f"job.working_dir_user: {job.working_dir_user}")
+        logger.debug(f"job.first_task_index: {job.first_task_index}")
+        logger.debug(f"job.last_task_index: {job.last_task_index}")
+        logger.debug(f'START workflow "{workflow.name}"')
+
+    try:
+        if FRACTAL_RUNNER_BACKEND == "local":
+            process_workflow = local_process_workflow
+            backend_specific_kwargs = {}
+        elif FRACTAL_RUNNER_BACKEND == "slurm":
+            process_workflow = slurm_sudo_process_workflow
+            backend_specific_kwargs = dict(
+                slurm_user=slurm_user,
+                slurm_account=job.slurm_account,
+                user_cache_dir=user_cache_dir,
+            )
+        elif FRACTAL_RUNNER_BACKEND == "slurm_ssh":
+            process_workflow = slurm_ssh_process_workflow
+            backend_specific_kwargs = dict(fractal_ssh=fractal_ssh)
+        else:
+            raise RuntimeError(
+                f"Invalid runner backend {FRACTAL_RUNNER_BACKEND=}"
+            )
+
+        # "The Session.close() method does not prevent the Session from being
+        # used again. The Session itself does not actually have a distinct
+        # “closed” state; it merely means the Session will release all database
+        # connections and ORM objects."
+        # (https://docs.sqlalchemy.org/en/20/orm/session_api.html#sqlalchemy.orm.Session.close).
+        #
+        # We close the session before the (possibly long) process_workflow
+        # call, to make sure all DB connections are released. The reason why we
+        # are not using a context manager within the try block is that we also
+        # need access to db_sync in the except branches.
+        db_sync = next(DB.get_sync_db())
+        db_sync.close()
+
+        process_workflow(
+            workflow=workflow,
+            dataset=dataset,
+            job_id=job_id,
+            user_id=user_id,
+            workflow_dir_local=WORKFLOW_DIR_LOCAL,
+            workflow_dir_remote=WORKFLOW_DIR_REMOTE,
+            logger_name=logger_name,
+            worker_init=worker_init,
+            first_task_index=job.first_task_index,
+            last_task_index=job.last_task_index,
+            job_attribute_filters=job.attribute_filters,
+            job_type_filters=job.type_filters,
+            **backend_specific_kwargs,
+        )
+
+        logger.info(
+            f'End execution of workflow "{workflow.name}"; '
+            f"more logs at {str(log_file_path)}"
+        )
+        logger.debug(f'END workflow "{workflow.name}"')
+
+        # Update job DB entry
+        job.status = JobStatusTypeV2.DONE
+        job.end_timestamp = get_timestamp()
+        with log_file_path.open("r") as f:
+            logs = f.read()
+        job.log = logs
+        db_sync.merge(job)
+        db_sync.commit()
+
+    except TaskExecutionError as e:
+        logger.debug(f'FAILED workflow "{workflow.name}", TaskExecutionError.')
+        logger.info(f'Workflow "{workflow.name}" failed (TaskExecutionError).')
+
+        exception_args_string = "\n".join(e.args)
+        log_msg = (
+            f"TASK ERROR: "
+            f"Task name: {e.task_name}, "
+            f"position in Workflow: {e.workflow_task_order}\n"
+            f"TRACEBACK:\n{exception_args_string}"
+        )
+        fail_job(db=db_sync, job=job, log_msg=log_msg, logger_name=logger_name)
+
+    except JobExecutionError as e:
+        logger.debug(f'FAILED workflow "{workflow.name}", JobExecutionError.')
+        logger.info(f'Workflow "{workflow.name}" failed (JobExecutionError).')
+
+        fail_job(
+            db=db_sync,
+            job=job,
+            log_msg=(
+                f"JOB ERROR in Fractal job {job.id}:\n"
+                f"TRACEBACK:\n{e.assemble_error()}"
+            ),
+            logger_name=logger_name,
+        )
+
+    except Exception:
+        logger.debug(f'FAILED workflow "{workflow.name}", unknown error.')
+        logger.info(f'Workflow "{workflow.name}" failed (unkwnon error).')
+
+        current_traceback = traceback.format_exc()
+        fail_job(
+            db=db_sync,
+            job=job,
+            log_msg=(
+                f"UNKNOWN ERROR in Fractal job {job.id}\n"
+                f"TRACEBACK:\n{current_traceback}"
+            ),
+            logger_name=logger_name,
+        )
+
+    finally:
+        reset_logger_handlers(logger)
+        db_sync.close()
+        _zip_folder_to_file_and_remove(folder=job.working_dir)

fractal_server/app/runner/v2/task_interface.py

@@ -1,11 +1,14 @@
 from typing import Any
+from typing import Optional
 
 from pydantic import BaseModel
 from pydantic import ConfigDict
 from pydantic import Field
 from pydantic import field_validator
+from pydantic import ValidationError
 
 from ....images import SingleImageTaskOutput
+from fractal_server.app.runner.exceptions import TaskOutputValidationError
 from fractal_server.urls import normalize_url
 
 
@@ -61,3 +64,31 @@ class InitTaskOutput(BaseModel):
     model_config = ConfigDict(extra="forbid")
 
     parallelization_list: list[InitArgsModel] = Field(default_factory=list)
+
+
+def _cast_and_validate_TaskOutput(
+    task_output: dict[str, Any]
+) -> Optional[TaskOutput]:
+    try:
+        validated_task_output = TaskOutput(**task_output)
+        return validated_task_output
+    except ValidationError as e:
+        raise TaskOutputValidationError(
+            "Validation of task output failed.\n"
+            f"Original error: {str(e)}\n"
+            f"Original data: {task_output}."
+        )
+
+
+def _cast_and_validate_InitTaskOutput(
+    init_task_output: dict[str, Any],
+) -> Optional[InitTaskOutput]:
+    try:
+        validated_init_task_output = InitTaskOutput(**init_task_output)
+        return validated_init_task_output
+    except ValidationError as e:
+        raise TaskOutputValidationError(
+            "Validation of init-task output failed.\n"
+            f"Original error: {str(e)}\n"
+            f"Original data: {init_task_output}."
+        )

fractal_server/app/schemas/_validators.py

@@ -1,43 +1,32 @@
 import os
+from typing import Annotated
 from typing import Any
 from typing import Optional
 
+from pydantic.types import StringConstraints
 
-def valstr(attribute: str, accept_none: bool = False):
-    """
-    Check that a string attribute is not an empty string, and remove the
-    leading and trailing whitespace characters.
 
-    If `accept_none`, the validator also accepts `None`.
-    """
+def cant_set_none(value: Any) -> Any:
+    if value is None:
+        raise ValueError("Field cannot be set to 'None'.")
+    return value
 
-    def val(cls, string: Optional[str]) -> Optional[str]:
-        if string is None:
-            if accept_none:
-                return string
-            else:
-                raise ValueError(
-                    f"String attribute '{attribute}' cannot be None"
-                )
-        s = string.strip()
-        if not s:
-            raise ValueError(f"String attribute '{attribute}' cannot be empty")
-        return s
 
-    return val
+NonEmptyString = Annotated[
+    str, StringConstraints(min_length=1, strip_whitespace=True)
+]
 
 
 def valdict_keys(attribute: str):
     def val(cls, d: Optional[dict[str, Any]]) -> Optional[dict[str, Any]]:
         """
-        Apply valstr to every key of the dictionary, and fail if there are
-        identical keys.
+        Strip every key of the dictionary, and fail if there are identical keys
         """
         if d is not None:
             old_keys = list(d.keys())
-            new_keys = [
-                valstr(f"{attribute}[{key}]")(cls, key) for key in old_keys
-            ]
+            new_keys = [key.strip() for key in old_keys]
+            if any(k == "" for k in new_keys):
+                raise ValueError(f"Empty string in {new_keys}.")
             if len(new_keys) != len(set(new_keys)):
                 raise ValueError(
                     f"Dictionary contains multiple identical keys: '{d}'."

fractal_server/app/schemas/user.py

@@ -7,8 +7,8 @@ from pydantic import Field
 from pydantic import field_validator
 from pydantic import ValidationInfo
 
+from ._validators import NonEmptyString
 from ._validators import val_unique_list
-from ._validators import valstr
 
 __all__ = (
     "UserRead",
@@ -57,12 +57,12 @@ class UserUpdate(schemas.BaseUserUpdate):
 
     model_config = ConfigDict(extra="forbid")
 
-    username: Optional[str] = None
+    username: Optional[NonEmptyString] = None
 
     # Validators
-    _username = field_validator("username")(classmethod(valstr("username")))
 
     @field_validator(
+        "username",
         "is_active",
         "is_verified",
         "is_superuser",
@@ -94,11 +94,14 @@ class UserCreate(schemas.BaseUserCreate):
         username:
     """
 
-    username: Optional[str] = None
+    username: Optional[NonEmptyString] = None
 
-    # Validators
-
-    _username = field_validator("username")(classmethod(valstr("username")))
+    @field_validator("username")
+    @classmethod
+    def cant_set_none(cls, v, info: ValidationInfo):
+        if v is None:
+            raise ValueError(f"Cannot set {info.field_name}=None")
+        return v
 
 
 class UserUpdateGroups(BaseModel):