Data-Name-ID
32 lines
1.1 KiB
Python
32 lines
1.1 KiB
Python
from rest_framework import status
|
|
from rest_framework.response import Response
|
|
from rest_framework.views import APIView
|
|
from teams.models import Team
|
|
from users.models import User
|
|
from rest_framework.generics import CreateAPIView
|
|
from api.teams.serializers import TeamSerializer, VacancySerializer
|
|
|
|
|
|
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)
|
|
|
|
|
|
class CreateVacancy(CreateAPIView):
|
|
http_method_names = ("post",)
|
|
serializer_class = VacancySerializer
|