mirror of
https://git.v0id.ovh/n3wt-innov/n3wt-school.git
synced 2026-01-29 07:53:23 +00:00
63 lines
3.2 KiB
Python
63 lines
3.2 KiB
Python
from django.http.response import JsonResponse
|
|
from django.views.decorators.csrf import ensure_csrf_cookie, csrf_protect
|
|
from django.utils.decorators import method_decorator
|
|
from rest_framework.parsers import JSONParser
|
|
from rest_framework.views import APIView
|
|
from rest_framework import status
|
|
from .models import Establishment
|
|
from .serializers import EstablishmentSerializer
|
|
from N3wtSchool.bdd import delete_object, getAllObjects
|
|
from School.models import Competency, EstablishmentCompetency
|
|
from django.db.models import Q
|
|
|
|
@method_decorator(csrf_protect, name='dispatch')
|
|
@method_decorator(ensure_csrf_cookie, name='dispatch')
|
|
class EstablishmentListCreateView(APIView):
|
|
def get(self, request):
|
|
establishments = getAllObjects(Establishment)
|
|
establishments_serializer = EstablishmentSerializer(establishments, many=True)
|
|
return JsonResponse(establishments_serializer.data, safe=False, status=status.HTTP_200_OK)
|
|
|
|
def post(self, request):
|
|
establishment_data = JSONParser().parse(request)
|
|
establishment_serializer = EstablishmentSerializer(data=establishment_data)
|
|
if establishment_serializer.is_valid():
|
|
establishment = establishment_serializer.save()
|
|
# Création des EstablishmentCompetency pour chaque compétence existante
|
|
competencies = Competency.objects.filter(
|
|
Q(end_of_cycle=True) | ~Q(level=None)
|
|
)
|
|
for competency in competencies:
|
|
EstablishmentCompetency.objects.get_or_create(
|
|
establishment=establishment,
|
|
competency=competency,
|
|
defaults={'is_required': True}
|
|
)
|
|
return JsonResponse(establishment_serializer.data, safe=False, status=status.HTTP_201_CREATED)
|
|
return JsonResponse(establishment_serializer.errors, safe=False, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
@method_decorator(csrf_protect, name='dispatch')
|
|
@method_decorator(ensure_csrf_cookie, name='dispatch')
|
|
class EstablishmentDetailView(APIView):
|
|
def get(self, request, id=None):
|
|
try:
|
|
establishment = Establishment.objects.get(id=id)
|
|
establishment_serializer = EstablishmentSerializer(establishment)
|
|
return JsonResponse(establishment_serializer.data, safe=False)
|
|
except Establishment.DoesNotExist:
|
|
return JsonResponse({'error': 'No object found'}, status=status.HTTP_404_NOT_FOUND)
|
|
|
|
def put(self, request, id):
|
|
establishment_data = JSONParser().parse(request)
|
|
try:
|
|
establishment = Establishment.objects.get(id=id)
|
|
except Establishment.DoesNotExist:
|
|
return JsonResponse({'error': 'No object found'}, status=status.HTTP_404_NOT_FOUND)
|
|
establishment_serializer = EstablishmentSerializer(establishment, data=establishment_data, partial=True)
|
|
if establishment_serializer.is_valid():
|
|
establishment_serializer.save()
|
|
return JsonResponse(establishment_serializer.data, safe=False)
|
|
return JsonResponse(establishment_serializer.errors, safe=False, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
def delete(self, request, id):
|
|
return delete_object(Establishment, id) |