leguadmin/xbackend
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

1

Browse Source
This commit is contained in:
wuaho 2021-09-09 15:30:27 +08:00
parent 8666c1fd04
commit 6520f89795
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
crud/crud_api_list.py
View File

@ -5,6 +5,8 @@ from crud.base import CRUDBase
__all__ = 'api_list', __all__ = 'api_list',
from utils import get_uid
class CRUDApiList(CRUDBase): class CRUDApiList(CRUDBase):
async def add_api(self, db: AsyncIOMotorDatabase, data_in: schemas.AddApi): 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): async def update_api(self, db: AsyncIOMotorDatabase, data_in: schemas.UpdateApi):
where = {'path': data_in.path} where = {'path': data_in.path}
data = {'$set': data_in.dict()} 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) return await self.update_one(db, where, data, upsert=True)
async def edit_api(self, db: AsyncIOMotorDatabase, data_in: schemas.EditApi): 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'})} data = {'$set': data_in.dict(exclude={'id'})}
return await self.update_one(db, where, data) return await self.update_one(db, where, data)
async def all_api(self, db: AsyncIOMotorDatabase): async def all_api(self, db: AsyncIOMotorDatabase):
return await self.find_many(db) return await self.find_many(db)
async def del_api(self, db: AsyncIOMotorDatabase, data_in: schemas.DelApi): async def del_api(self, db: AsyncIOMotorDatabase, data_in: schemas.DelApi):
return await self.delete_id(db, *data_in.ids) return await self.delete_id(db, *data_in.ids)