from django.db import models
from rex.models.module import BasicModule, BasicModuleSerializer
from rex.models.university import University
from rex.models.module import Currency
from django.core.validators import MinValueValidator
from rest_framework import viewsets, permissions


class UniversityInfo(BasicModule):

    university = models.OneToOneField(
        University, on_delete=models.CASCADE, related_name='university_info', primary_key=True, null=False)

    cost_exchange = models.DecimalField(
        decimal_places=2,
        max_digits=20,
        validators=[MinValueValidator(0)],
        null=True
    )

    cost_double_degree = models.DecimalField(
        decimal_places=2,
        max_digits=20,
        validators=[MinValueValidator(0)],
        null=True
    )

    costs_currency = models.ForeignKey(Currency, on_delete=models.PROTECT)


class UniversityInfoSerializer(BasicModuleSerializer):

    class Meta:
        model = UniversityInfo
        fields = '__all__'


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