from rex.models.university import University
from rest_framework import serializers, viewsets, permissions
from rex.models.university.campus import CampusSerializer


class UniversitySerializer(serializers.ModelSerializer):
    univCampus = CampusSerializer(many=True, read_only=True)

    class Meta:
        model = University
        fields = ('name', 'acronym', 'logo', 'id', 'univCampus')


class UniversityViewSet(viewsets.ModelViewSet):  # TODO switch to dict
    permission_classes = (permissions.DjangoModelPermissions,)
    queryset = University.objects.all()  # pylint: disable=E1101
    serializer_class = UniversitySerializer