This commit is contained in:
wuaho 2021-09-09 15:30:27 +08:00
parent 8666c1fd04
commit 6520f89795

View File

@ -5,6 +5,8 @@ from crud.base import CRUDBase
__all__ = 'api_list',
from utils import get_uid
class CRUDApiList(CRUDBase):
async def add_api(self, db: AsyncIOMotorDatabase, data_in: schemas.AddApi):
@ -16,7 +18,9 @@ class CRUDApiList(CRUDBase):
async def update_api(self, db: AsyncIOMotorDatabase, data_in: schemas.UpdateApi):
where = {'path': data_in.path}
data = {'$set': data_in.dict()}
is_exists = await self.find_one(db, {'path': data_in.path})
if not is_exists:
data['$set']['_id'] = get_uid()
return await self.update_one(db, where, data, upsert=True)
async def edit_api(self, db: AsyncIOMotorDatabase, data_in: schemas.EditApi):
@ -24,14 +28,9 @@ class CRUDApiList(CRUDBase):
data = {'$set': data_in.dict(exclude={'id'})}
return await self.update_one(db, where, data)
async def all_api(self, db: AsyncIOMotorDatabase):
return await self.find_many(db)
async def del_api(self, db: AsyncIOMotorDatabase, data_in: schemas.DelApi):
return await self.delete_id(db, *data_in.ids)