n3wt-school/Back-End/Establishment/views.py

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 EstablishmentCompetency, Competency
from django.db.models import Q
from Auth.models import Profile, ProfileRole, Directeur
from Settings.models import SMTPSettings

@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)
        try:
            establishment, data = create_establishment_with_directeur(establishment_data)
            # 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(data, safe=False, status=status.HTTP_201_CREATED)
        except Exception as e:
            return JsonResponse({"error": str(e)}, 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)

def create_establishment_with_directeur(establishment_data):
    # Extraction des sous-objets
    directeur_data = establishment_data.pop("directeur", None)
    smtp_settings_data = establishment_data.pop("smtp_settings", {})

    # Vérification de la présence du directeur
    if not directeur_data or not directeur_data.get("email"):
        raise ValueError("Le champ 'directeur.email' est obligatoire.")

    directeur_email = directeur_data.get("email")
    last_name = directeur_data.get("last_name", "")
    first_name = directeur_data.get("first_name", "")
    password = directeur_data.get("password", "Provisoire01!")

    # Création ou récupération du profil utilisateur
    profile, created = Profile.objects.get_or_create(
        email=directeur_email,
        defaults={"username": directeur_email}
    )
    if created or not profile.has_usable_password():
        profile.set_password(password)
        profile.save()

    # Création de l'établissement
    establishment_serializer = EstablishmentSerializer(data=establishment_data)
    establishment_serializer.is_valid(raise_exception=True)
    establishment = establishment_serializer.save()

    # Création ou récupération du ProfileRole ADMIN pour ce profil et cet établissement
    profile_role, _ = ProfileRole.objects.get_or_create(
        profile=profile,
        establishment=establishment,
        role_type=ProfileRole.RoleType.PROFIL_ADMIN,
        defaults={"is_active": True}
    )

    # Création ou mise à jour du Directeur lié à ce ProfileRole
    Directeur.objects.update_or_create(
        profile_role=profile_role,
        defaults={
            "last_name": last_name,
            "first_name": first_name
        }
    )

    # Création du SMTPSettings rattaché à l'établissement si des données sont fournies
    if smtp_settings_data:
        smtp_settings_data["establishment"] = establishment
        SMTPSettings.objects.create(**smtp_settings_data)

    return establishment, establishment_serializer.data