mirror of
https://github.com/beak-insights/felicity-lims.git
synced 2025-02-24 08:53:00 +08:00
68 lines
2.6 KiB
Python
68 lines
2.6 KiB
Python
import logging
|
|
from typing import List
|
|
|
|
from felicity.apps.analysis.entities.analysis import Sample
|
|
from felicity.apps.analysis.enum import SampleState
|
|
from felicity.apps.analysis.services.analysis import SampleService
|
|
from felicity.apps.impress.sample import utils
|
|
from felicity.apps.iol.redis import process_tracker
|
|
from felicity.apps.iol.redis.enum import TrackableObject
|
|
from felicity.apps.job import schemas as job_schemas
|
|
from felicity.apps.job.enum import JobAction, JobCategory, JobState
|
|
from felicity.apps.job.enum import JobPriority
|
|
from felicity.apps.job.services import JobService
|
|
from felicity.apps.notification.services import NotificationService
|
|
from felicity.apps.user.entities import User
|
|
from felicity.apps.user.services import UserService
|
|
from felicity.core.config import get_settings
|
|
|
|
settings = get_settings()
|
|
logging.basicConfig(level=logging.INFO)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
async def impress_results(job_uid: str):
|
|
logger.info(f"starting impress job {job_uid} ....")
|
|
job = await JobService().get(uid=job_uid)
|
|
if not job:
|
|
return
|
|
|
|
if job.status != JobState.PENDING:
|
|
return
|
|
|
|
await JobService().change_status(job.uid, new_status=JobState.RUNNING)
|
|
|
|
user = await UserService().get(uid=job.creator_uid)
|
|
|
|
await utils.impress_samples(job.data, user)
|
|
try:
|
|
await JobService().change_status(job.uid, new_status=JobState.FINISHED)
|
|
await process_tracker.release(uid=job.uid, object_type=TrackableObject.SAMPLE)
|
|
await NotificationService().notify("Your results were successfully published", user)
|
|
except Exception as e:
|
|
await JobService().change_status(job.uid, new_status=JobState.FAILED)
|
|
logger.info(f"Failed impress job {job_uid} with errr: {str(e)}")
|
|
await NotificationService().notify(
|
|
f"Failed to publish results in job with uid: {job.uid} with error: {str(e)}",
|
|
user,
|
|
)
|
|
|
|
|
|
async def prepare_for_impress():
|
|
samples: List[Sample] = await SampleService().get_all(status__in=[SampleState.APPROVED])
|
|
sample_uids = [sample.uid for sample in samples]
|
|
system_daemon: User = await UserService().get(email=settings.SYSTEM_DAEMON_EMAIL)
|
|
|
|
job_schema = job_schemas.JobCreate(
|
|
action=JobAction.IMPRESS_REPORT,
|
|
category=JobCategory.IMPRESS,
|
|
priority=JobPriority.NORMAL,
|
|
job_id="0",
|
|
status=JobState.PENDING,
|
|
creator_uid=system_daemon.uid,
|
|
data=sample_uids,
|
|
)
|
|
|
|
await JobService().create(job_schema)
|
|
for uid in sample_uids:
|
|
await process_tracker.process(uid=uid, object_type=TrackableObject.SAMPLE)
|