from backend.models.university import University
from rest_framework import serializers, viewsets, permissions
from backend.models.university.campus import CampusSerializer
from backend.models.tools import NoDeleteIfNotAdmin
from backend.models.module import MyModelSerializer


class UniversitySerializer(MyModelSerializer):
    univ_campus = CampusSerializer(many=True, read_only=True)

    class Meta:
        model = University
        fields = '__all__'


class UniversityViewSet(viewsets.ModelViewSet):  # TODO switch to dict
    permission_classes = (
        permissions.IsAuthenticated,
        NoDeleteIfNotAdmin,
    )
    queryset = University.objects.all()\
        .prefetch_related('univ_campus', 'univ_campus__updated_by')  # pylint: disable=E1101
    serializer_class = UniversitySerializer