25 lines
964 B
Python
25 lines
964 B
Python
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 |