felicity-lims/felicity/apps/worksheet/workflow.py

from felicity.apps.analysis.enum import ResultState
from felicity.apps.worksheet.entities import WorkSheet
from felicity.apps.worksheet.enum import WorkSheetState
from felicity.apps.worksheet.services import WorkSheetService


class WorksheetWorkFlowException(Exception): ...


class WorkSheetWorkFlow:
    """WorksheetWorkFlow
    Defines a set of guards that allow or prevent actions taken on Worksheet
    """

    def __init__(self):
        self.worksheet_service = WorkSheetService()

    async def revert(self, uid: str, by_uid: str):
        to_status = WorkSheetState.PENDING
        _results, qc_results = await self.worksheet_service.get_analysis_results(uid)
        results = _results + qc_results
        print("arrrrrrrrrrrrrr: ", results)

        awaiting_states = [ResultState.RESULTED, ResultState.RETRACTED]
        if all([(ar.status in awaiting_states) for ar in results]):
            to_status = WorkSheetState.AWAITING
        await self.worksheet_service.change_state(uid, to_status, by_uid)

    async def submit(self, uid, submitter):
        worksheet = await self.worksheet_service.get(uid=uid)
        await self._guard_submit(worksheet)
        return self.worksheet_service.submit(worksheet.uid, submitter)

    async def _guard_submit(self, worksheet: WorkSheet) -> bool:
        if worksheet.state not in [WorkSheetState.PENDING, WorkSheetState.SUBMITTING]:
            raise WorksheetWorkFlowException(
                f"Cannot submit a {worksheet.state} WorkSheet"
            )

        result_states = [ResultState.PENDING]

        results, qc_results = await self.worksheet_service.get_analysis_results(
            worksheet.uid
        )
        if (
            len(
                list(
                    filter(lambda ar: ar.status in result_states, results + qc_results)
                )
            )
            > 0
        ):
            raise WorksheetWorkFlowException(
                "Cannot submit a Worksheet with pending results"
            )
        return True

    async def approve(self, uid, approved_by):
        worksheet = await WorkSheet.get(uid=uid)
        await self._guard_approve(worksheet)
        await self.worksheet_service.verify(worksheet.uid, approved_by)

    async def _guard_approve(self, worksheet: WorkSheet) -> bool:
        if worksheet.state not in [WorkSheetState.AWAITING]:
            raise WorksheetWorkFlowException(
                f"Cannot approve a {worksheet.state} WorkSheet"
            )

        result_states = [
            ResultState.APPROVED,
            ResultState.RETRACTED,
            ResultState.CANCELLED,
        ]

        results, qc_results = await self.worksheet_service.get_analysis_results(
            worksheet.uid
        )
        if not all([ar.status in result_states for ar in results + qc_results]):
            raise WorksheetWorkFlowException("Cannot approve this Worksheet")

        return True