refactor: Simplify repository classes by inheriting from BaseRepository and removing redundant methods

app/repositories/user.py

@@ -8,26 +8,17 @@ from sqlmodel.ext.asyncio.session import AsyncSession
 from app.core.logging import get_logger
 from app.models.plan import Plan
 from app.models.user import User
+from app.repositories.base import BaseRepository
 
 logger = get_logger(__name__)
 
 
-class UserRepository:
+class UserRepository(BaseRepository[User]):
     """Repository for user operations."""
 
     def __init__(self, session: AsyncSession) -> None:
         """Initialize the user repository."""
-        self.session = session
-
-    async def get_by_id(self, user_id: int) -> User | None:
-        """Get a user by ID."""
-        try:
-            statement = select(User).where(User.id == user_id)
-            result = await self.session.exec(statement)
-            return result.first()
-        except Exception:
-            logger.exception("Failed to get user by ID: %s", user_id)
-            raise
+        super().__init__(User, session)
 
     async def get_by_email(self, email: str) -> User | None:
         """Get a user by email address."""
@@ -50,7 +41,7 @@ class UserRepository:
             raise
 
     async def create(self, user_data: dict[str, Any]) -> User:
-        """Create a new user."""
+        """Create a new user with plan assignment and first user admin logic."""
 
         def _raise_plan_not_found() -> None:
             msg = "Default plan not found"
@@ -84,45 +75,11 @@ class UserRepository:
             user_data["plan_id"] = default_plan.id
             user_data["credits"] = default_plan.credits
 
-            user = User(**user_data)
-            self.session.add(user)
-            await self.session.commit()
-            await self.session.refresh(user)
+            # Use BaseRepository's create method
+            return await super().create(user_data)
         except Exception:
-            await self.session.rollback()
             logger.exception("Failed to create user")
             raise
-        else:
-            logger.info("Created new user with email: %s", user.email)
-            return user
-
-    async def update(self, user: User, update_data: dict[str, Any]) -> User:
-        """Update a user."""
-        try:
-            for field, value in update_data.items():
-                setattr(user, field, value)
-
-            await self.session.commit()
-            await self.session.refresh(user)
-        except Exception:
-            await self.session.rollback()
-            logger.exception("Failed to update user")
-            raise
-        else:
-            logger.info("Updated user: %s", user.email)
-            return user
-
-    async def delete(self, user: User) -> None:
-        """Delete a user."""
-        try:
-            await self.session.delete(user)
-            await self.session.commit()
-
-            logger.info("Deleted user: %s", user.email)
-        except Exception:
-            await self.session.rollback()
-            logger.exception("Failed to delete user")
-            raise
 
     async def email_exists(self, email: str) -> bool:
         """Check if an email address is already registered."""