sakhteman/app/repositories/monitor_rial_progress.py

import uuid

from app.config import TaskEnv
from app.models.monitor_rial_progress import RialProgressModel
from app.repositories.project import ProjectRepository
from core.repositories.base import BaseRepository
from core.service.mongo_db import use_collection
from core.utils.crudify import mongo_inserter


class RialProgressRepository(BaseRepository):
    def __init__(self):
        super().__init__(collection=TaskEnv.CLN_MONITOR_RIAL, model=RialProgressModel)

    async def last_record(self, project_id:uuid.UUID):
        last = await use_collection(TaskEnv.CLN_MONITOR_RIAL).find({'project_id':project_id}).sort("$natural", -1).to_list(None)
        if last:
            return last[0]
        else:
            return {}

    async def create(self,_inp:RialProgressModel):
        create = await mongo_inserter(_inp,TaskEnv.CLN_MONITOR_RIAL,getter=self.get)
        await ProjectRepository().set_sts_two_for_monitor(_inp.project_id)
        return create