34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
from typing import Union
|
|
|
|
from bson import ObjectId
|
|
|
|
|
|
class CRUDBase:
|
|
def __init__(self, coll_name):
|
|
self.coll_name = coll_name
|
|
|
|
async def get(self, db, id: Union[ObjectId, str]):
|
|
return (await db[self.coll_name].find_one({'_id': ObjectId(id)})) or dict()
|
|
|
|
async def read_have(self, db, user_id: str, **kwargs):
|
|
where = {'members': user_id}
|
|
where.update(kwargs)
|
|
cursor = db[self.coll_name].find(where)
|
|
return await cursor.to_list(length=999)
|
|
|
|
async def find_many(self, db, **kwargs):
|
|
cursor = db[self.coll_name].find(kwargs)
|
|
return await cursor.to_list(length=999)
|
|
|
|
async def delete(self, db, **kwargs):
|
|
return await db[self.coll_name].delete_many(kwargs)
|
|
|
|
async def update_one(self, db, id, **kwargs):
|
|
return await db[self.coll_name].update_one({'_id': id}, kwargs)
|
|
|
|
async def update_upsert(self, db, where: dict, **kwargs):
|
|
return await db[self.coll_name].update_one(where, {'$set': kwargs}, upsert=True)
|
|
|
|
async def distinct(self, db, key, filter=None):
|
|
return await db[self.coll_name].distinct(key, filter)
|