28 lines
904 B
Python
28 lines
904 B
Python
from backend.project.users.models import User
|
|
from rest_framework import status
|
|
from rest_framework.response import Response
|
|
from rest_framework.views import APIView
|
|
|
|
from teams.models import Team
|
|
|
|
from .serializers import TeamSerializer
|
|
|
|
|
|
class AddUserToTeam(APIView):
|
|
def post(self, request, team_id, user_id): # noqa: ARG002
|
|
try:
|
|
team = Team.objects.get(id=team_id)
|
|
user = User.objects.get(id=user_id)
|
|
except Team.DoesNotExist:
|
|
return Response(
|
|
{"error": "Team not found"}, status=status.HTTP_404_NOT_FOUND
|
|
)
|
|
except User.DoesNotExist:
|
|
return Response(
|
|
{"error": "User not found"}, status=status.HTTP_404_NOT_FOUND
|
|
)
|
|
|
|
team.members.add(user)
|
|
team_serializer = TeamSerializer(team)
|
|
return Response(team_serializer.data, status=status.HTTP_200_OK)
|