35 lines
1012 B
Python
35 lines
1012 B
Python
from motor.motor_asyncio import AsyncIOMotorDatabase
|
|
import schemas
|
|
from crud.base import CRUDBase
|
|
|
|
__all__ = 'owner_info',
|
|
|
|
|
|
class CRUDOwnerInfo(CRUDBase):
|
|
# 获取所有渠道信息
|
|
async def all_owner(self, db: AsyncIOMotorDatabase):
|
|
return await self.find_many(db, {}, {'_id': 0})
|
|
|
|
# 获取指定字段
|
|
async def find_owner_some(self, db: AsyncIOMotorDatabase, where={}, findlist=[]):
|
|
findWhere = {'_id': 0}
|
|
if findlist:
|
|
for key in findlist:
|
|
findWhere.update({
|
|
key: 1
|
|
})
|
|
return await self.find_many(db, where, findWhere)
|
|
|
|
# 更新一条渠道信息
|
|
async def update_owner(self, db: AsyncIOMotorDatabase, where, updatedata):
|
|
|
|
await self.update_one(db, where, {'$set': updatedata})
|
|
|
|
# 插入一条新的渠道数据
|
|
async def insert_owner(self, db: AsyncIOMotorDatabase, insertdata):
|
|
|
|
await self.insert_one(db, insertdata)
|
|
|
|
|
|
owner_info = CRUDOwnerInfo('owner_info')
|