feat: add functionality to retrieve users by team ID and update team detail endpoint (#151)

Co-authored-by: Amit Prakash <[email protected]>

Author: amit-flx

todo/repositories/team_repository.py

@@ -89,3 +89,15 @@ def get_by_user_id(cls, user_id: str) -> list[UserTeamDetailsModel]:
             return [UserTeamDetailsModel(**data) for data in user_teams_data]
         except Exception:
             return []
+
+    @classmethod
+    def get_users_by_team_id(cls, team_id: str) -> list[str]:
+        """
+        Get all user IDs for a specific team.
+        """
+        collection = cls.get_collection()
+        try:
+            user_teams_data = collection.find({"team_id": ObjectId(team_id), "is_active": True})
+            return [str(data["user_id"]) for data in user_teams_data]
+        except Exception:
+            return []

todo/services/user_service.py

@@ -7,6 +7,7 @@
 )
 from rest_framework.exceptions import ValidationError as DRFValidationError
 from typing import List, Tuple
+from todo.dto.user_dto import UserDTO
 
 
 class UserService:
@@ -35,6 +36,21 @@ def search_users(cls, query: str, page: int = 1, limit: int = 10) -> Tuple[List[
         cls._validate_search_params(query, page, limit)
         return UserRepository.search_users(query, page, limit)
 
+    @classmethod
+    def get_users_by_ids(cls, user_ids: list[str]) -> list[UserDTO]:
+        users = []
+        for user_id in user_ids:
+            user = UserRepository.get_by_id(user_id)
+            if user:
+                users.append(UserDTO(
+                    id=str(user.id),
+                    name=user.name,
+                    email_id=user.email_id,
+                    created_at=user.created_at,
+                    updated_at=user.updated_at,
+                ))
+        return users
+
     @classmethod
     def _validate_google_user_data(cls, google_user_data: dict) -> None:
         validation_errors = {}

todo/views/team.py

@@ -14,6 +14,7 @@
 from drf_spectacular.utils import extend_schema, OpenApiParameter, OpenApiResponse
 from drf_spectacular.types import OpenApiTypes
 from todo.dto.team_dto import TeamDTO
+from todo.services.user_service import UserService
 
 
 class TeamListView(APIView):
@@ -99,7 +100,7 @@ class TeamDetailView(APIView):
     @extend_schema(
         operation_id="get_team_by_id",
         summary="Get team by ID",
-        description="Retrieve a single team by its unique identifier",
+        description="Retrieve a single team by its unique identifier. Optionally, set ?member=true to get users belonging to this team.",
         tags=["teams"],
         parameters=[
             OpenApiParameter(
@@ -108,20 +109,34 @@ class TeamDetailView(APIView):
                 location=OpenApiParameter.PATH,
                 description="Unique identifier of the team",
             ),
+            OpenApiParameter(
+                name="member",
+                type=OpenApiTypes.BOOL,
+                location=OpenApiParameter.QUERY,
+                description="If true, returns users that belong to this team instead of team details.",
+                required=False,
+            ),
         ],
         responses={
-            200: OpenApiResponse(description="Team retrieved successfully"),
+            200: OpenApiResponse(description="Team or team members retrieved successfully"),
             404: OpenApiResponse(description="Team not found"),
             500: OpenApiResponse(description="Internal server error"),
         },
     )
     def get(self, request: Request, team_id: str):
         """
-        Retrieve a single team by ID.
+        Retrieve a single team by ID, or users in the team if ?member=true.
         """
         try:
-            team_dto: TeamDTO = TeamService.get_team_by_id(team_id)
-            return Response(data=team_dto.model_dump(mode="json"), status=status.HTTP_200_OK)
+            member = request.query_params.get("member", "false").lower() == "true"
+            if member:
+                from todo.repositories.team_repository import UserTeamDetailsRepository
+                user_ids = UserTeamDetailsRepository.get_users_by_team_id(team_id)
+                users = UserService.get_users_by_ids(user_ids)
+                return Response(data=[user.model_dump(mode="json") for user in users], status=status.HTTP_200_OK)
+            else:
+                team_dto: TeamDTO = TeamService.get_team_by_id(team_id)
+                return Response(data=team_dto.model_dump(mode="json"), status=status.HTTP_200_OK)
         except ValueError as e:
             fallback_response = ApiErrorResponse(
                 statusCode=404,