from django.db import models
from backend_app.models.abstract.taggedItem import (
    TaggedItem,
    TaggedItemSerializer,
    TaggedItemViewSet,
)
from backend_app.models.university import University


class UniversityTaggedItem(TaggedItem):

    university = models.ForeignKey(
        University, on_delete=models.PROTECT, related_name="university_tagged_items"
    )

    class Meta:
        unique_together = ("university", "tag", "importance_level")


class UniversityTaggedItemSerializer(TaggedItemSerializer):
    class Meta:
        model = UniversityTaggedItem
        fields = "__all__"


class UniversityTaggedItemViewSet(TaggedItemViewSet):
    queryset = UniversityTaggedItem.objects.all()  # pylint: disable=E1101
    serializer_class = UniversityTaggedItemSerializer

    def get_queryset(self):
        univ_id = self.kwargs["univ_id"]
        return super().get_queryset().filter(university__pk=univ_id).distinct()