campus.py

from django.db import models
from backend.models.module import BasicModule, BasicModuleSerializer
from backend.models.location import City
from backend.models.university import University
from django.core.validators import MinValueValidator, MaxValueValidator
from rest_framework import viewsets, permissions


class Campus(BasicModule):
    module_icon = "You forget to override the icon in the sub-class : I know"
    module_name = "Campus principal"
    module_description = "Campus de l'université."

    is_main_campus = models.BooleanField(null=False)
    name = models.CharField(max_length=200, null=True)
    comment = models.TextField(null=True, blank=True)
    city = models.ForeignKey(City, on_delete=models.PROTECT, null=False)
    university = models.ForeignKey(
        University, on_delete=models.PROTECT, null=False, related_name="univCampus")

    lat = models.DecimalField(
        max_digits=10,
        decimal_places=6,
        validators=[MinValueValidator(-85.05112878),
                    MaxValueValidator(85.05112878)]
    )

    lon = models.DecimalField(
        max_digits=10,
        decimal_places=6,
        validators=[MinValueValidator(-180),
                    MaxValueValidator(180)]
    )

    def location(self):
        return {'lat': self.lat, 'lon': self.lon}

    class Meta:
        unique_together = ('is_main_campus', 'university')


class CampusSerializer(BasicModuleSerializer):

    class Meta:
        model = Campus
        fields = '__all__'


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