diff --git a/app/repositories/project.py b/app/repositories/project.py index a587b17..e7027ca 100644 --- a/app/repositories/project.py +++ b/app/repositories/project.py @@ -45,4 +45,12 @@ class ProjectRepository(BaseRepository[ProjectModel]): update.monitor.code = 'S' + code update.u_at() await use_collection(COLLECTION).update_one({"_id":_id},{"$set":update.mongo_dict()}) - return await self.get(_id) \ No newline at end of file + return await self.get(_id) + + async def list_project_design(self): + project = await use_collection(COLLECTION).find({'monitor':{"$exists":0}}).to_list(None) + return project + + async def list_project_monitor(self): + project = await use_collection(COLLECTION).find({'monitor':{"$exists":1}}).to_list(None) + return project \ No newline at end of file diff --git a/app/urls/project.py b/app/urls/project.py index 9e61f7a..ace6dcf 100644 --- a/app/urls/project.py +++ b/app/urls/project.py @@ -20,6 +20,16 @@ repo = ProjectRepository() def create(coding: ThisModel = Body(...)): pass +@router.post('/list/design', response_model=ThisModel, tags=_tag_base) +@shrink(repo.list_project_design) +def list_project_design(): + pass + +@router.post('/list/monitor', response_model=ThisModel, tags=_tag_base) +@shrink(repo.list_project_monitor) +def list_project_monitor(): + pass + @router.put('/{_id}', response_model=ThisModel, tags=_tag_base) @shrink(repo.update)