49 lines
1.8 KiB
Python
49 lines
1.8 KiB
Python
from typing import List, Optional
|
|
|
|
from bson import ObjectId
|
|
from motor.motor_asyncio import AsyncIOMotorClient
|
|
|
|
from models.Character import Character
|
|
|
|
|
|
class CharacterRepo:
|
|
def __init__(self, client: AsyncIOMotorClient, db_name="bot_db"):
|
|
self.collection = client[db_name]["characters"]
|
|
|
|
async def add_character(self, character: Character) -> Character:
|
|
op = await self.collection.insert_one(character.model_dump())
|
|
character.id = str(op.inserted_id)
|
|
return character
|
|
|
|
async def get_character(self, character_id: str) -> Character | None:
|
|
res = await self.collection.find_one({"_id": ObjectId(character_id)})
|
|
if res is None:
|
|
return None
|
|
else:
|
|
res["id"] = str(res.pop("_id"))
|
|
return Character(**res)
|
|
|
|
async def get_all_characters(self, created_by: Optional[str] = None, project_id: Optional[str] = None, limit: int = 100, offset: int = 0) -> List[Character]:
|
|
filter = {}
|
|
if created_by:
|
|
filter["created_by"] = created_by
|
|
if project_id is None:
|
|
filter["project_id"] = None
|
|
if project_id:
|
|
filter["project_id"] = project_id
|
|
|
|
res = await self.collection.find(filter).skip(offset).limit(limit).to_list(None)
|
|
chars = []
|
|
for doc in res:
|
|
doc["id"] = str(doc.pop("_id"))
|
|
chars.append(Character(**doc))
|
|
return chars
|
|
|
|
async def update_char(self, char_id: str, character: Character) -> bool:
|
|
result = await self.collection.update_one({"_id": ObjectId(char_id)}, {"$set": character.model_dump()})
|
|
return result.modified_count > 0
|
|
|
|
async def delete_character(self, char_id: str) -> bool:
|
|
result = await self.collection.delete_one({"_id": ObjectId(char_id)})
|
|
return result.deleted_count > 0
|