chore: remove unnecessary code

Author: Achintya-Chatterjee

todo/repositories/task_repository.py

@@ -46,7 +46,8 @@ def list(
 
         if team_id:
             logger.debug(f"TaskRepository.list: team_id={team_id}")
-            team_task_ids = cls._get_assigned_task_ids_for_team(team_id)
+            team_assignments = TaskAssignmentRepository.get_by_assignee_id(team_id, "team")
+            team_task_ids = [assignment.task_id for assignment in team_assignments]
             logger.debug(f"TaskRepository.list: team_task_ids={team_task_ids}")
             query_filter = {"$and": [base_filter, {"_id": {"$in": team_task_ids}}]}
             logger.debug(f"TaskRepository.list: query_filter={query_filter}")
@@ -77,45 +78,38 @@ def _get_assigned_task_ids_for_user(cls, user_id: str) -> List[ObjectId]:
         direct_task_ids = [assignment.task_id for assignment in direct_assignments]
 
         # Get teams where user is a member
-        from todo.repositories.team_repository import UserTeamDetailsRepository
+        from todo.repositories.team_repository import UserTeamDetailsRepository, TeamRepository
 
         user_teams = UserTeamDetailsRepository.get_by_user_id(user_id)
         team_ids = [str(team.team_id) for team in user_teams]
 
-        # Get tasks assigned to those teams
+        # Get tasks assigned to those teams (only if user is POC)
         team_task_ids = []
-        for team_id in team_ids:
-            team_assignments = TaskAssignmentRepository.get_by_assignee_id(team_id, "team")
-            team_task_ids.extend([assignment.task_id for assignment in team_assignments])
+        if team_ids:
+            # Get teams where user is POC
+            poc_teams = TeamRepository.get_collection().find(
+                {"_id": {"$in": [ObjectId(team_id) for team_id in team_ids]}, "is_deleted": False, "poc_id": user_id}
+            )
+            poc_team_ids = [str(team["_id"]) for team in poc_teams]
+
+            # Get team assignments for POC teams
+            if poc_team_ids:
+                team_assignments = TaskAssignmentRepository.get_collection().find(
+                    {"assignee_id": {"$in": poc_team_ids}, "user_type": "team", "is_active": True}
+                )
+                team_task_ids = [ObjectId(assignment["task_id"]) for assignment in team_assignments]
 
         return direct_task_ids + team_task_ids
 
-    @classmethod
-    def _get_assigned_task_ids_for_team(cls, team_id: str) -> List[ObjectId]:
-        """Get task IDs where team is assigned (either directly or via team members)."""
-
-        direct_team_assignments = TaskAssignmentRepository.get_by_assignee_id(team_id, "team")
-        direct_team_task_ids = [assignment.task_id for assignment in direct_team_assignments]
-
-        from todo.repositories.team_repository import UserTeamDetailsRepository
-
-        team_member_ids = UserTeamDetailsRepository.get_users_by_team_id(team_id)
-
-        member_task_ids = []
-        for member_id in team_member_ids:
-            member_assignments = TaskAssignmentRepository.get_by_assignee_id(member_id, "user")
-            member_task_ids.extend([assignment.task_id for assignment in member_assignments])
-
-        return direct_team_task_ids + member_task_ids
-
     @classmethod
     def count(cls, user_id: str = None, team_id: str = None, status_filter: str = None) -> int:
         tasks_collection = cls.get_collection()
 
         base_filter = cls._build_status_filter(status_filter)
 
         if team_id:
-            team_task_ids = cls._get_assigned_task_ids_for_team(team_id)
+            team_assignments = TaskAssignmentRepository.get_by_assignee_id(team_id, "team")
+            team_task_ids = [assignment.task_id for assignment in team_assignments]
             query_filter = {"$and": [base_filter, {"_id": {"$in": team_task_ids}}]}
         elif user_id:
             assigned_task_ids = cls._get_assigned_task_ids_for_user(user_id)