fractal-server 2.18.5__py3-none-any.whl → 2.19.0__py3-none-any.whl

fractal_server/app/routes/api/v2/status_legacy.py

@@ -1,156 +0,0 @@
-from fastapi import APIRouter
-from fastapi import Depends
-
-from fractal_server.app.db import AsyncSession
-from fractal_server.app.db import get_async_db
-from fractal_server.app.models import UserOAuth
-from fractal_server.app.models.v2 import JobV2
-from fractal_server.app.routes.auth import current_user_act_ver_prof
-from fractal_server.app.schemas.v2.sharing import ProjectPermissions
-from fractal_server.app.schemas.v2.status_legacy import LegacyStatusRead
-from fractal_server.app.schemas.v2.status_legacy import WorkflowTaskStatusType
-from fractal_server.logger import set_logger
-
-from ._aux_functions import _get_dataset_check_access
-from ._aux_functions import _get_submitted_jobs_statement
-from ._aux_functions import _get_workflow_check_access
-
-router = APIRouter()
-
-logger = set_logger(__name__)
-
-
-@router.get(
-    "/project/{project_id}/status-legacy/",
-    response_model=LegacyStatusRead,
-)
-async def get_workflowtask_status(
-    project_id: int,
-    dataset_id: int,
-    workflow_id: int,
-    user: UserOAuth = Depends(current_user_act_ver_prof),
-    db: AsyncSession = Depends(get_async_db),
-) -> LegacyStatusRead | None:
-    """
-    Extract the status of all `WorkflowTaskV2` of a given `WorkflowV2` that ran
-    on a given `DatasetV2`.
-
-    *NOTE*: the current endpoint is not guaranteed to provide consistent
-    results if the workflow task list is modified in a non-trivial way
-    (that is, by adding intermediate tasks, removing tasks, or changing their
-    order). See fractal-server GitHub issues: 793, 1083.
-    """
-    # Get the dataset DB entry
-    output = await _get_dataset_check_access(
-        project_id=project_id,
-        dataset_id=dataset_id,
-        user_id=user.id,
-        required_permissions=ProjectPermissions.READ,
-        db=db,
-    )
-    dataset = output["dataset"]
-
-    # Get the workflow DB entry
-    workflow = await _get_workflow_check_access(
-        project_id=project_id,
-        workflow_id=workflow_id,
-        user_id=user.id,
-        required_permissions=ProjectPermissions.READ,
-        db=db,
-    )
-
-    # Check whether there exists a submitted job associated to this
-    # workflow/dataset pair. If it does exist, it will be used later.
-    # If there are multiple jobs, raise an error.
-    res = await db.execute(
-        _get_submitted_jobs_statement()
-        .where(JobV2.dataset_id == dataset_id)
-        .where(JobV2.workflow_id == workflow_id)
-    )
-    running_job = res.scalars().one_or_none()
-
-    # Initialize empty dictionary for WorkflowTaskV2 status
-    workflow_tasks_status_dict: dict = {}
-
-    # Lowest priority: read status from DB, which corresponds to jobs that are
-    # not running
-    history = dataset.history
-    for history_item in history:
-        wftask_id = history_item["workflowtask"]["id"]
-        wftask_status = history_item["status"]
-        workflow_tasks_status_dict[wftask_id] = wftask_status
-
-    if running_job is None:
-        # If no job is running, the chronological-last history item is also the
-        # positional-last workflow task to be included in the response.
-        if len(history) > 0:
-            last_valid_wftask_id = history[-1]["workflowtask"]["id"]
-        else:
-            last_valid_wftask_id = None
-    else:
-        # If a job is running, then gather more up-to-date information
-
-        # Mid priority: Set all WorkflowTask's that are part of the running job
-        # as "submitted"
-        start = running_job.first_task_index
-        end = running_job.last_task_index + 1
-
-        running_job_wftasks = workflow.task_list[start:end]
-        running_job_statuses = [
-            workflow_tasks_status_dict.get(wft.id, None)
-            for wft in running_job_wftasks
-        ]
-        try:
-            first_submitted_index = running_job_statuses.index(
-                WorkflowTaskStatusType.SUBMITTED
-            )
-        except ValueError:
-            logger.warning(
-                f"Job {running_job.id} is submitted but its task list does not"
-                f" contain a {WorkflowTaskStatusType.SUBMITTED} task."
-            )
-            first_submitted_index = 0
-
-        for wftask in running_job_wftasks[first_submitted_index:]:
-            workflow_tasks_status_dict[wftask.id] = (
-                WorkflowTaskStatusType.SUBMITTED
-            )
-
-        # The last workflow task that is included in the submitted job is also
-        # the positional-last workflow task to be included in the response.
-        try:
-            last_valid_wftask_id = workflow.task_list[end - 1].id
-        except IndexError as e:
-            logger.warning(
-                f"Handled IndexError in `get_workflowtask_status` ({str(e)})."
-            )
-            logger.warning(
-                "Additional information: "
-                f"{running_job.first_task_index=}; "
-                f"{running_job.last_task_index=}; "
-                f"{len(workflow.task_list)=}; "
-                f"{dataset_id=}; "
-                f"{workflow_id=}."
-            )
-            last_valid_wftask_id = None
-            logger.warning(f"Now setting {last_valid_wftask_id=}.")
-
-    # Based on previously-gathered information, clean up the response body
-    clean_workflow_tasks_status_dict = {}
-    for wf_task in workflow.task_list:
-        wf_task_status = workflow_tasks_status_dict.get(wf_task.id)
-        if wf_task_status is None:
-            # If a wftask ID was not found, ignore it and continue
-            continue
-        clean_workflow_tasks_status_dict[str(wf_task.id)] = wf_task_status
-        if wf_task_status == WorkflowTaskStatusType.FAILED:
-            # Starting from the beginning of `workflow.task_list`, stop the
-            # first time that you hit a failed job
-            break
-        if wf_task.id == last_valid_wftask_id:
-            # Starting from the beginning of `workflow.task_list`, stop the
-            # first time that you hit `last_valid_wftask_id``
-            break
-
-    response_body = LegacyStatusRead(status=clean_workflow_tasks_status_dict)
-    return response_body