feat: add method to retrieve user information by team ID and correspo… (#155)

* feat: add method to retrieve user information by team ID and corresponding unit tests

* refactor: remove unused import of ObjectId in migrate_add_creator_to_teams command

* refactor: improve code readability in migrate_add_creator_to_teams command and user retrieval method

---------

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

Author: amit-flx

todo/management/commands/migrate_add_creator_to_teams.py

@@ -0,0 +1,35 @@
+from django.core.management.base import BaseCommand
+from todo.models.team import UserTeamDetailsModel
+from todo.repositories.team_repository import TeamRepository, UserTeamDetailsRepository
+from todo.models.common.pyobjectid import PyObjectId
+
+
+class Command(BaseCommand):
+    help = "Backfill user_team_details so every team has its creator as an active member."
+
+    def handle(self, *args, **options):
+        teams = TeamRepository.get_collection().find({"is_deleted": False})
+        updated = 0
+        for team in teams:
+            team_id = team["_id"]
+            creator_id = team["created_by"]
+            # Check if creator is already an active member
+            exists = UserTeamDetailsRepository.get_collection().find_one(
+                {
+                    "team_id": team_id,
+                    "user_id": creator_id,
+                    "is_active": True,
+                }
+            )
+            if not exists:
+                user_team = UserTeamDetailsModel(
+                    user_id=PyObjectId(creator_id),
+                    team_id=PyObjectId(team_id),
+                    role_id="1",
+                    is_active=True,
+                    created_by=PyObjectId(creator_id),
+                    updated_by=PyObjectId(creator_id),
+                )
+                UserTeamDetailsRepository.create(user_team)
+                updated += 1
+        self.stdout.write(self.style.SUCCESS(f"Added creator as member to {updated} teams."))

todo/repositories/team_repository.py

@@ -101,3 +101,18 @@ def get_users_by_team_id(cls, team_id: str) -> list[str]:
             return [str(data["user_id"]) for data in user_teams_data]
         except Exception:
             return []
+
+    @classmethod
+    def get_user_infos_by_team_id(cls, team_id: str) -> list[dict]:
+        """
+        Get all user info (user_id, name, email) for a specific team.
+        """
+        from todo.repositories.user_repository import UserRepository
+
+        user_ids = cls.get_users_by_team_id(team_id)
+        user_infos = []
+        for user_id in user_ids:
+            user = UserRepository.get_by_id(user_id)
+            if user:
+                user_infos.append({"user_id": user_id, "name": user.name, "email": user.email_id})
+        return user_infos

todo/tests/unit/repositories/test_user_repository.py

@@ -8,6 +8,7 @@
 from todo.exceptions.auth_exceptions import UserNotFoundException, APIException
 from todo.tests.fixtures.user import users_db_data
 from todo.constants.messages import RepositoryErrors
+from todo.repositories.team_repository import UserTeamDetailsRepository
 
 
 class UserRepositoryTests(TestCase):
@@ -92,3 +93,27 @@ def test_create_or_update_sets_timestamps(self, mock_db_manager):
         self.assertIn("updated_at", update_doc["$set"])
         self.assertIn("$setOnInsert", update_doc)
         self.assertIn("created_at", update_doc["$setOnInsert"])
+
+
+class UserTeamDetailsRepositoryTests(TestCase):
+    @patch("todo.repositories.user_repository.UserRepository.get_by_id")
+    @patch("todo.repositories.team_repository.UserTeamDetailsRepository.get_users_by_team_id")
+    def test_get_user_infos_by_team_id(self, mock_get_users_by_team_id, mock_get_by_id):
+        team_id = str(ObjectId())
+        user_ids = [str(ObjectId()), str(ObjectId())]
+        mock_get_users_by_team_id.return_value = user_ids
+        user1 = MagicMock()
+        user1.name = "Alice"
+        user1.email_id = "[email protected]"
+        user2 = MagicMock()
+        user2.name = "Bob"
+        user2.email_id = "[email protected]"
+        mock_get_by_id.side_effect = [user1, user2]
+
+        result = UserTeamDetailsRepository.get_user_infos_by_team_id(team_id)
+
+        self.assertEqual(len(result), 2)
+        self.assertEqual(result[0]["name"], "Alice")
+        self.assertEqual(result[0]["email"], "[email protected]")
+        self.assertEqual(result[1]["name"], "Bob")
+        self.assertEqual(result[1]["email"], "[email protected]")