university.py

from django.conf import settings
from django.core.exceptions import ValidationError
from django.db import models
from rest_framework.validators import ValidationError as RFValidationError

from backend_app.models.abstract.essentialModule import (
    EssentialModule,
    EssentialModuleSerializer,
    EssentialModuleViewSet,
)
from backend_app.validators.tags import validate_extension


def validate_extension_django(url):
    try:
        validate_extension(settings.ALLOWED_PHOTOS_EXTENSION, url)
    except RFValidationError:
        raise ValidationError("Extension in URL doesn't correspond to an image")


class University(EssentialModule):
    """
    Model storing information about universities
    """

    name = models.CharField(max_length=200)
    acronym = models.CharField(max_length=20, default="", blank=True)
    logo = models.URLField(
        default="", blank=True, validators=[validate_extension_django]
    )
    website = models.URLField(default="", blank=True, max_length=300)
    utc_id = models.IntegerField(unique=True)


class UniversitySerializer(EssentialModuleSerializer):
    class Meta:
        model = University
        exclude = ("utc_id",)


class UniversityViewSet(EssentialModuleViewSet):
    serializer_class = UniversitySerializer
    queryset = University.objects.all()  # pylint: disable=E1101
    end_point_route = "universities"