+ api

repos/assets_repo.py

@@ -1,4 +1,4 @@
-from typing import List
+from typing import List, Optional
 
 from bson import ObjectId
 from motor.motor_asyncio import AsyncIOMotorClient
@@ -10,10 +10,10 @@ class AssetsRepo:
     def __init__(self, client: AsyncIOMotorClient, db_name="bot_db"):
         self.collection = client[db_name]["assets"]
 
-    async def save_asset(self, asset: Asset) -> Asset:
+    async def create_asset(self, asset: Asset) -> str:
         res = await self.collection.insert_one(asset.model_dump())
-        asset.id = res.inserted_id
-        return asset
+
+        return str(res.inserted_id)
 
     async def get_assets(self, limit: int = 10, offset: int = 0) -> List[Asset]:
         res = await self.collection.find({}, {"data": 0}).sort("created_at", -1).skip(offset).limit(limit).to_list(None)
@@ -27,6 +27,7 @@ class AssetsRepo:
 
         return assets
 
+
     async def get_asset(self, asset_id: str, with_data: bool = True) -> Asset:
         projection = {"_id": 1, "name": 1, "type": 1, "tg_doc_file_id": 1}
         if with_data:
@@ -54,3 +55,17 @@ class AssetsRepo:
             doc["id"] = str(doc.pop("_id"))
             assets.append(Asset(**doc))
         return assets
+
+    async def get_asset_count(self, character_id: Optional[str] = None) -> int:
+        return await self.collection.count_documents({"linked_char_id": character_id} if character_id else {})
+
+
+
+    async def get_assets_by_ids(self, asset_ids: List[str]) -> List[Asset]:
+        object_ids = [ObjectId(asset_id) for asset_id in asset_ids]
+        res = self.collection.find({"_id": {"$in": object_ids}})
+        assets = []
+        async for doc in res:
+            doc["id"] = str(doc.pop("_id"))
+            assets.append(Asset(**doc))
+        return assets