feat: create teams

.env.example

@@ -13,4 +13,4 @@ PUBLIC_KEY="generate keys and paste here"
 
 # use if required
 # ACCESS_LIFETIME="20"
-# REFRESH_LIFETIME="30"
+# REFRESH_LIFETIME="30"

todo/constants/messages.py

@@ -1,6 +1,7 @@
 # Application Messages
 class AppMessages:
     TASK_CREATED = "Task created successfully"
+    TEAM_CREATED = "Team created successfully"
     GOOGLE_LOGIN_SUCCESS = "Successfully logged in with Google"
     GOOGLE_LOGOUT_SUCCESS = "Successfully logged out"
     TOKEN_REFRESHED = "Access token refreshed successfully"

todo/dto/responses/create_team_response.py

@@ -0,0 +1,7 @@
+from pydantic import BaseModel
+from todo.dto.team_dto import TeamDTO
+
+
+class CreateTeamResponse(BaseModel):
+    team: TeamDTO
+    message: str

todo/dto/team_dto.py

@@ -0,0 +1,49 @@
+from pydantic import BaseModel, validator
+from typing import List, Optional
+from datetime import datetime
+from todo.repositories.user_repository import UserRepository
+
+
+class CreateTeamDTO(BaseModel):
+    name: str
+    description: Optional[str] = None
+    member_ids: Optional[List[str]] = None
+    poc_id: str
+
+    @validator("member_ids")
+    def validate_member_ids(cls, value):
+        """Validate that all member IDs exist in the database."""
+        if value is None:
+            return value
+
+        invalid_ids = []
+        for member_id in value:
+            user = UserRepository.get_by_id(member_id)
+            if not user:
+                invalid_ids.append(member_id)
+
+        if invalid_ids:
+            raise ValueError(f"Invalid member IDs: {invalid_ids}")
+        return value
+
+    @validator("poc_id")
+    def validate_poc_id(cls, value):
+        """Validate that the POC ID exists in the database."""
+        if value is None:
+            return value
+
+        user = UserRepository.get_by_id(value)
+        if not user:
+            raise ValueError(f"Invalid POC ID: {value}")
+        return value
+
+
+class TeamDTO(BaseModel):
+    id: str
+    name: str
+    description: Optional[str] = None
+    poc_id: str
+    created_by: str
+    updated_by: str
+    created_at: datetime
+    updated_at: datetime

todo/models/team.py

@@ -0,0 +1,58 @@
+from pydantic import Field, validator
+from typing import ClassVar
+from datetime import datetime, timezone
+
+from todo.models.common.document import Document
+from todo.models.common.pyobjectid import PyObjectId
+
+
+class TeamModel(Document):
+    """
+    Model for teams.
+    """
+
+    collection_name: ClassVar[str] = "teams"
+
+    name: str = Field(..., min_length=1, max_length=100)
+    description: str | None = None
+    poc_id: PyObjectId
+    created_by: PyObjectId
+    updated_by: PyObjectId
+    created_at: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))
+    updated_at: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))
+    is_deleted: bool = False
+
+    @validator("created_by", "updated_by")
+    def validate_user_id(cls, v):
+        """Validate that the user ID is a valid ObjectId format."""
+        if v is None:
+            raise ValueError("User ID cannot be None")
+        if not PyObjectId.is_valid(v):
+            raise ValueError(f"Invalid user ID format: {v}")
+        return v
+
+
+class UserTeamDetailsModel(Document):
+    """
+    Model for user-team relationships.
+    """
+
+    collection_name: ClassVar[str] = "user_team_details"
+
+    user_id: PyObjectId
+    team_id: PyObjectId
+    is_active: bool = True
+    role_id: str
+    created_at: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))
+    updated_at: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))
+    created_by: PyObjectId
+    updated_by: PyObjectId
+
+    @validator("user_id", "team_id", "created_by", "updated_by")
+    def validate_object_ids(cls, v):
+        """Validate that the ObjectId fields are in valid format."""
+        if v is None:
+            raise ValueError("ObjectId cannot be None")
+        if not PyObjectId.is_valid(v):
+            raise ValueError(f"Invalid ObjectId format: {v}")
+        return v

todo/repositories/team_repository.py

@@ -0,0 +1,79 @@
+from datetime import datetime, timezone
+from typing import Optional
+from bson import ObjectId
+
+from todo.models.team import TeamModel, UserTeamDetailsModel
+from todo.repositories.common.mongo_repository import MongoRepository
+
+
+class TeamRepository(MongoRepository):
+    collection_name = TeamModel.collection_name
+
+    @classmethod
+    def create(cls, team: TeamModel) -> TeamModel:
+        """
+        Creates a new team in the repository.
+        """
+        teams_collection = cls.get_collection()
+        team.created_at = datetime.now(timezone.utc)
+        team.updated_at = datetime.now(timezone.utc)
+
+        team_dict = team.model_dump(mode="json", by_alias=True, exclude_none=True)
+        insert_result = teams_collection.insert_one(team_dict)
+        team.id = insert_result.inserted_id
+        return team
+
+    @classmethod
+    def get_by_id(cls, team_id: str) -> Optional[TeamModel]:
+        """
+        Get a team by its ID.
+        """
+        teams_collection = cls.get_collection()
+        try:
+            team_data = teams_collection.find_one({"_id": ObjectId(team_id), "is_deleted": False})
+            if team_data:
+                return TeamModel(**team_data)
+            return None
+        except Exception:
+            return None
+
+
+class UserTeamDetailsRepository(MongoRepository):
+    collection_name = UserTeamDetailsModel.collection_name
+
+    @classmethod
+    def create(cls, user_team: UserTeamDetailsModel) -> UserTeamDetailsModel:
+        """
+        Creates a new user-team relationship.
+        """
+        collection = cls.get_collection()
+        user_team.created_at = datetime.now(timezone.utc)
+        user_team.updated_at = datetime.now(timezone.utc)
+
+        user_team_dict = user_team.model_dump(mode="json", by_alias=True, exclude_none=True)
+        insert_result = collection.insert_one(user_team_dict)
+        user_team.id = insert_result.inserted_id
+        return user_team
+
+    @classmethod
+    def create_many(cls, user_teams: list[UserTeamDetailsModel]) -> list[UserTeamDetailsModel]:
+        """
+        Creates multiple user-team relationships.
+        """
+        collection = cls.get_collection()
+        current_time = datetime.now(timezone.utc)
+
+        for user_team in user_teams:
+            user_team.created_at = current_time
+            user_team.updated_at = current_time
+
+        user_teams_dicts = [
+            user_team.model_dump(mode="json", by_alias=True, exclude_none=True) for user_team in user_teams
+        ]
+        insert_result = collection.insert_many(user_teams_dicts)
+
+        # Set the inserted IDs
+        for i, user_team in enumerate(user_teams):
+            user_team.id = insert_result.inserted_ids[i]
+
+        return user_teams

todo/serializers/create_team_serializer.py

@@ -0,0 +1,8 @@
+from rest_framework import serializers
+
+
+class CreateTeamSerializer(serializers.Serializer):
+    name = serializers.CharField(max_length=100)
+    description = serializers.CharField(max_length=500, required=False, allow_blank=True)
+    member_ids = serializers.ListField(child=serializers.CharField(), required=False, default=list)
+    poc_id = serializers.CharField(required=True, allow_null=False)

todo/services/team_service.py

@@ -0,0 +1,86 @@
+from todo.dto.team_dto import CreateTeamDTO, TeamDTO
+from todo.dto.responses.create_team_response import CreateTeamResponse
+from todo.models.team import TeamModel, UserTeamDetailsModel
+from todo.models.common.pyobjectid import PyObjectId
+from todo.repositories.team_repository import TeamRepository, UserTeamDetailsRepository
+from todo.constants.messages import AppMessages
+
+
+class TeamService:
+    @classmethod
+    def create_team(cls, dto: CreateTeamDTO, created_by_user_id: str) -> CreateTeamResponse:
+        """
+        Create a new team with members and POC.
+        """
+        try:
+            # Member IDs and POC ID validation is handled at DTO level
+            member_ids = dto.member_ids or []
+
+            # Create team
+            team = TeamModel(
+                name=dto.name,
+                description=dto.description,
+                poc_id=PyObjectId(dto.poc_id) if dto.poc_id else None,
+                created_by=PyObjectId(created_by_user_id),
+                updated_by=PyObjectId(created_by_user_id),
+            )
+
+            created_team = TeamRepository.create(team)
+
+            # Create user-team relationships
+            user_teams = []
+
+            # Add members to the team
+            if member_ids:
+                for user_id in member_ids:
+                    user_team = UserTeamDetailsModel(
+                        user_id=PyObjectId(user_id),
+                        team_id=created_team.id,
+                        role_id="1",
+                        created_by=PyObjectId(created_by_user_id),
+                        updated_by=PyObjectId(created_by_user_id),
+                    )
+                    user_teams.append(user_team)
+
+            # Add POC if not already in member_ids
+            if dto.poc_id and dto.poc_id not in member_ids:
+                poc_user_team = UserTeamDetailsModel(
+                    user_id=PyObjectId(dto.poc_id),
+                    team_id=created_team.id,
+                    role_id="2",  # POC role
+                    created_by=PyObjectId(created_by_user_id),
+                    updated_by=PyObjectId(created_by_user_id),
+                )
+                user_teams.append(poc_user_team)
+
+            # Add creator if not already in member_ids
+            if created_by_user_id not in member_ids:
+                creator_user_team = UserTeamDetailsModel(
+                    user_id=PyObjectId(created_by_user_id),
+                    team_id=created_team.id,
+                    role_id="1",
+                    created_by=PyObjectId(created_by_user_id),
+                    updated_by=PyObjectId(created_by_user_id),
+                )
+                user_teams.append(creator_user_team)
+
+            # Create all user-team relationships
+            if user_teams:
+                UserTeamDetailsRepository.create_many(user_teams)
+
+            # Convert to DTO
+            team_dto = TeamDTO(
+                id=str(created_team.id),
+                name=created_team.name,
+                description=created_team.description,
+                poc_id=str(created_team.poc_id) if created_team.poc_id else None,
+                created_by=str(created_team.created_by),
+                updated_by=str(created_team.updated_by),
+                created_at=created_team.created_at,
+                updated_at=created_team.updated_at,
+            )
+
+            return CreateTeamResponse(team=team_dto, message=AppMessages.TEAM_CREATED)
+
+        except Exception as e:
+            raise ValueError(str(e))

todo/tests/unit/views/test_auth.py

@@ -111,6 +111,8 @@ def setUp(self):
 
         self.test_user_data = users_db_data[0]
 
+        self.test_user_data = users_db_data[0]
+
     def test_get_redirects_for_oauth_error(self):
         error = "access_denied"
         response = self.client.get(f"{self.url}?error={error}")
@@ -148,7 +150,10 @@ def test_get_redirects_for_valid_code_and_state(self, mock_create_user, mock_han
             "email": self.test_user_data["email_id"],
             "name": self.test_user_data["name"],
         }
+<<<<<<< HEAD
+=======
 
+>>>>>>> a57ab63 (fix: lint and format)
         user_id = str(ObjectId())
         mock_user = Mock()
         mock_user.id = ObjectId(user_id)

todo/urls.py

@@ -2,6 +2,7 @@
 from todo.views.task import TaskListView, TaskDetailView
 from todo.views.label import LabelListView
 from todo.views.health import HealthView
+from todo.views.team import TeamListView
 from todo.views.auth import (
     GoogleLoginView,
     GoogleCallbackView,
@@ -12,6 +13,7 @@
 urlpatterns = [
     path("tasks", TaskListView.as_view(), name="tasks"),
     path("tasks/<str:task_id>", TaskDetailView.as_view(), name="task_detail"),
+    path("teams", TeamListView.as_view(), name="teams"),
     path("health", HealthView.as_view(), name="health"),
     path("labels", LabelListView.as_view(), name="labels"),
     path("auth/google/login/", GoogleLoginView.as_view(), name="google_login"),