51 lines
1.9 KiB
Python
51 lines
1.9 KiB
Python
import uuid
|
|
import datetime
|
|
|
|
from app.config import TaskEnv
|
|
from app.models.monito_physical_progress import PhysicalProgressModel, LastRecord
|
|
from core.repositories.base import BaseRepository
|
|
from core.service.mongo_db import use_collection
|
|
from core.utils.crudify import mongo_inserter
|
|
from app.repositories.project import ProjectRepository
|
|
|
|
COLLECTION = TaskEnv.CLN_MONITOR_PYSICAL
|
|
MODEL = PhysicalProgressModel
|
|
|
|
|
|
class PhysicalProgressRepository(BaseRepository):
|
|
def __init__(self):
|
|
super().__init__(collection=TaskEnv.CLN_MONITOR_PYSICAL, model=PhysicalProgressModel)
|
|
|
|
async def search_last(self, connection: str, _inp: LastRecord, first: int, last: int):
|
|
m_first = _inp.date_at - datetime.timedelta(days=first)
|
|
print(m_first)
|
|
m_end = _inp.date_at - datetime.timedelta(days=last)
|
|
print(m_end)
|
|
last = await use_collection(connection).find({
|
|
'project_id': _inp.project_id,
|
|
'date_at': {'$gte': m_first, '$lte': m_end}
|
|
}).sort("$natural", -1).to_list(None)
|
|
if last:
|
|
return last[0]
|
|
else:
|
|
return {}
|
|
|
|
async def last_record(self, _inp: LastRecord):
|
|
send = {'month': {}, 'three_month': {}}
|
|
last = await self.search_last(TaskEnv.CLN_MONITOR_PYSICAL, _inp, 35, 25)
|
|
if last:
|
|
send['month'] = last
|
|
three_month = await self.search_last(TaskEnv.CLN_MONITOR_PYSICAL, _inp, 95, 85)
|
|
if three_month:
|
|
send['three_month'] = three_month
|
|
return send
|
|
|
|
async def create(self, _inp: MODEL):
|
|
create = await mongo_inserter(_inp, COLLECTION, getter=self.get)
|
|
await ProjectRepository().set_sts_two_for_monitor(_inp.project_id)
|
|
return create
|
|
|
|
async def list_by_project(self,_id:uuid.UUID):
|
|
_get = await use_collection(COLLECTION).find({"project_id":_id}).to_list(None)
|
|
return _get
|