Add base, user, and session repository classes for MongoDB integration

src/repositories/__init__.py

@@ -1,3 +1,12 @@
-__all__ = []
+from ._user_repository import UserRepository
+from ._base_repository import BaseRepository
+from ._session_repository import SessionRepository
+
+__all__ = [
+    "UserRepository",
+    "BaseRepository",
+    "SessionRepository",
+]
+
 __version__ = "0.1.0"
 __author__ = "Ramanjit Singh"

src/repositories/_base_repository.py

@@ -0,0 +1,48 @@
+from beanie import Document
+from typing import Type, TypeVar, List, Dict
+
+T = TypeVar("T", bound=Document)
+
+
+class BaseRepository:
+    def __init__(self, model: Type[T]):
+        self.model = model
+
+    async def insert_one(self, data: T):
+        try:
+            return await data.insert()
+        except Exception as e:
+            raise ValueError(f"Failed to insert data: {str(e)}")
+
+    async def get_all(self, page: int = 1, page_size: int = None, filter_by: Dict = None, order_by: Dict = None):
+        query = filter_by or {}
+        cursor = self.model.find(query)
+
+        if order_by:
+            cursor = cursor.sort(order_by)
+
+        if page_size is not None:
+            skip = (page - 1) * page_size
+            cursor = cursor.skip(skip).limit(page_size)
+
+        return await cursor.to_list(length=page_size if page_size else None)
+
+    async def get_by_id(self, _id: str):
+        return await self.model.get(_id)
+
+    async def update(self, _id: str, update_data: dict):
+        document = await self.model.get(_id)
+        if not document:
+            raise ValueError(f"Document with id {_id} not found")
+
+        updated_fields = {**document.model_dump(), **update_data.model_dump()}
+        return await document.set(updated_fields)
+    async def delete(self, _id: str):
+
+        document = await self.model.get(_id)
+        if not document:
+            raise ValueError(f"Document with id {_id} not found")
+        return await document.delete()
+
+    async def count_documents(self):
+        return await self.model.count()

src/repositories/_session_repository.py

@@ -0,0 +1,8 @@
+from src.models import Session
+
+from ._base_repository import BaseRepository
+
+
+class SessionRepository(BaseRepository):
+    def __init__(self):
+        super().__init__(model=Session)

src/repositories/_user_repository.py

@@ -0,0 +1,8 @@
+from src.models import User
+
+from ._base_repository import BaseRepository
+
+
+class UserRepository(BaseRepository):
+    def __init__(self):
+        super().__init__(model=User)