inspirations

repos/dao.py

@@ -9,6 +9,7 @@ from repos.project_repo import ProjectRepo
 from repos.idea_repo import IdeaRepo
 from repos.post_repo import PostRepo
 from repos.environment_repo import EnvironmentRepo
+from repos.inspiration_repo import InspirationRepo
 
 
 from typing import Optional
@@ -25,3 +26,4 @@ class DAO:
         self.ideas = IdeaRepo(client, db_name)
         self.posts = PostRepo(client, db_name)
         self.environments = EnvironmentRepo(client, db_name)
+        self.inspirations = InspirationRepo(client, db_name)

repos/inspiration_repo.py

@@ -0,0 +1,54 @@
+from typing import List, Optional
+
+from bson import ObjectId
+from motor.motor_asyncio import AsyncIOMotorClient
+
+from models.Inspiration import Inspiration
+
+
+class InspirationRepo:
+    def __init__(self, client: AsyncIOMotorClient, db_name="bot_db"):
+        self.collection = client[db_name]["inspirations"]
+
+    async def create_inspiration(self, inspiration: Inspiration) -> str:
+        res = await self.collection.insert_one(inspiration.model_dump(exclude={"id"}))
+        return str(res.inserted_id)
+
+    async def get_inspiration(self, inspiration_id: str) -> Optional[Inspiration]:
+        res = await self.collection.find_one({"_id": ObjectId(inspiration_id)})
+        if res:
+            res["id"] = str(res.pop("_id"))
+            return Inspiration(**res)
+        return None
+
+    async def get_inspirations(self, project_id: Optional[str] = None, created_by: Optional[str] = None, limit: int = 20, offset: int = 0) -> List[Inspiration]:
+        query = {}
+        if project_id:
+            query["project_id"] = project_id
+        if created_by:
+            query["created_by"] = created_by
+
+        cursor = self.collection.find(query).sort("created_at", -1).skip(offset).limit(limit)
+        inspirations = []
+        async for doc in cursor:
+            doc["id"] = str(doc.pop("_id"))
+            inspirations.append(Inspiration(**doc))
+        return inspirations
+
+    async def count_inspirations(self, project_id: Optional[str] = None, created_by: Optional[str] = None) -> int:
+        query = {}
+        if project_id:
+            query["project_id"] = project_id
+        if created_by:
+            query["created_by"] = created_by
+        return await self.collection.count_documents(query)
+
+    async def update_inspiration(self, inspiration: Inspiration):
+        await self.collection.update_one(
+            {"_id": ObjectId(inspiration.id)},
+            {"$set": inspiration.model_dump(exclude={"id"})}
+        )
+
+    async def delete_inspiration(self, inspiration_id: str) -> bool:
+        res = await self.collection.delete_one({"_id": ObjectId(inspiration_id)})
+        return res.deleted_count > 0