mirror of
https://github.com/beak-insights/felicity-lims.git
synced 2025-02-23 16:33:11 +08:00
81 lines
2.9 KiB
Python
81 lines
2.9 KiB
Python
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
|