felicity-lims/felicity/apps/shipment/tasks.py

import logging

from apps.iol.fhir.utils import (
    get_shipment_bundle_resource,
    get_diagnostic_report_resource,
)
from apps.iol.relay import post_data
from apps.job import models as job_models
from apps.job.conf import states as job_states
from apps.shipment import conf, models
from apps.shipment.utils import (
    shipment_assign,
    shipment_reset_assigned_count,
    shipment_receive,
    shipment_result_update,
)
from apps.user.models import User

logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)


async def populate_shipment_manually(job_uid: str):
    logger.info(f"starting job {job_uid} ....")
    job: job_models.Job = await job_models.Job.get(uid=job_uid)
    if not job:
        return

    if not job.status == job_states.PENDING:
        return

    await job.change_status(new_status=job_states.RUNNING)
    shipment_uid = job.job_id

    shipment: models.Shipment = await models.Shipment.get(uid=shipment_uid)
    if not shipment:
        await job.change_status(
            new_status=job_states.FAILED,
            change_reason=f"Failed to acquire Shipment {shipment_uid}",
        )
        logger.warning(f"Failed to acquire Shipment {shipment_uid}")
        return

    await shipment_reset_assigned_count(shipment.uid)

    # handle Empty and preperation
    if shipment.state not in [
        conf.shipment_states.EMPTY,
        conf.shipment_states.PREPERATION,
    ]:
        await job.change_status(
            new_status=job_states.FAILED,
            change_reason=f"Shipment {shipment_uid} - is already processed",
        )
        logger.warning(f"Shipment {shipment_uid} - is already processed")
        return

    await shipment_assign(shipment.uid, job.data, job.creator_uid)

    await job.change_status(new_status=job_states.FINISHED)
    logger.info(f"Done !! Job {job_uid} was executed successfully :)")


async def dispatch_shipment(job_uid: str, by_uid=None):
    logger.info(f"running shipment dispatch job: {job_uid} ....")
    job: job_models.Job = await job_models.Job.get(uid=job_uid)
    if not job:
        return

    if not job.status == job_states.PENDING:
        return None

    await job.change_status(new_status=job_states.RUNNING)
    shipment_uid = job.job_id
    shipment: models.Shipment = await models.Shipment.get(uid=shipment_uid)

    if not shipment.state == conf.shipment_states.AWAITING:
        await job.change_status(new_status=job_states.FAILED)
        return None

    resource = await get_shipment_bundle_resource(shipment_uid)
    success = await post_data(
        f"{shipment.laboratory.url}/Bundle",
        resource.model_dump(exclude_none=True),
        shipment.laboratory.username,
        shipment.laboratory.password,
    )

    await job.change_status(
        new_status=job_states.PENDING if not success else job_states.FINISHED
    )
    await shipment.change_state(
        state=conf.shipment_states.FAILED
        if not success
        else conf.shipment_states.SHIPPED,
        updated_by_uid=by_uid if by_uid else job.creator_uid,
    )
    if not success:
        await job.backoff(1, 10)

    return await models.Shipment.get(uid=shipment_uid)


async def receive_shipment(job_uid: str):
    logger.info(f"starting job receive shipment: {job_uid} ....")
    job: job_models.Job = await job_models.Job.get(uid=job_uid)
    if not job:
        return

    if not job.status == job_states.PENDING:
        return

    await job.change_status(new_status=job_states.RUNNING)
    shipment_uid = job.job_id

    shipment: models.Shipment = await models.Shipment.get(uid=shipment_uid)
    if not shipment:
        await job.change_status(
            new_status=job_states.FAILED,
            change_reason=f"Failed to acquire Shipment {shipment_uid}",
        )
        logger.warning(f"Failed to acquire Shipment {shipment_uid}")
        return

    await shipment_receive(shipment.uid, job.data, job.creator_uid)

    await job.change_status(new_status=job_states.FINISHED)
    logger.info(f"Done !! Job {job_uid} was executed successfully :)")


async def return_shipped_report(job_uid: str):
    logger.info(f"starting job return shipped report: {job_uid} ....")
    job: job_models.Job = await job_models.Job.get(uid=job_uid)
    if not job:
        return

    if not job.status == job_states.PENDING:
        return

    await job.change_status(new_status=job_states.RUNNING)

    shipped_sample_uid = job.job_id
    shipped: models.ShippedSample = await models.ShippedSample.get(
        uid=shipped_sample_uid
    )

    if not shipped:
        await job.change_status(
            new_status=job_states.FAILED,
            change_reason=f"Failed to acquire shipped sample {shipped_sample_uid}",
        )
        logger.warning(f"Failed to acquire shipped sample {shipped_sample_uid}")
        return

    shipment = await models.Shipment.get(uid=shipped.shipment_uid)
    if not shipment:
        await job.change_status(
            new_status=job_states.FAILED,
            change_reason=f"Failed to acquire Shipment {shipped.shipment_uid}",
        )
        logger.warning(f"Failed to acquire Shipment {shipped.shipment_uid}")
        return

    result_uids = []
    if job.data["target"] == "result":
        result_uids.append(job.data["uid"])

    resource = await get_diagnostic_report_resource(
        shipped.sample.analysis_request_uid, result_uids, True
    )
    success = await post_data(
        f"{shipment.laboratory.url}DiagnosticReport",
        resource.model_dump(exclude_none=True),
        shipment.laboratory.username,
        shipment.laboratory.password,
    )

    if not success:
        await job.change_status(new_status=job_states.FAILED)
    else:
        await job.change_status(new_status=job_states.FINISHED)


async def process_shipped_report(job_uid: str):
    logger.info(f"starting job return shipped report: {job_uid} ....")
    job: job_models.Job = await job_models.Job.get(uid=job_uid)
    if not job:
        return

    if not job.status == job_states.PENDING:
        return

    await job.change_status(new_status=job_states.RUNNING)
    data = job.data.get("data", None)

    assert data["resourceType"] == "DiagnosticReport"

    actor = await User.get(uid=job.creator_uid)
    await shipment_result_update(data, actor)

    await job.change_status(new_status=job_states.FINISHED)