from apps.fim.models import PriceSet, FeeSet, Fee
from apps.utils.viewsets import CustomModelViewSet, CustomGenericViewSet
from rest_framework.mixins import ListModelMixin, UpdateModelMixin
from apps.fim.serializers import (PriceSetSerializer, FeeSetSerializer, FeeSerializer)

# Create your views here.
class FeeViewSet(ListModelMixin, CustomGenericViewSet):
    perms_map = {'get': '*'}
    serializer_class = FeeSerializer
    queryset = Fee.objects.all()


class PriceSetViewSet(CustomModelViewSet):
    queryset = PriceSet.objects.all()
    serializer_class = PriceSetSerializer
    select_related_fields = ['material']
    filterset_fields = ['material', 'year', 'month']
    search_fields = ['material__name']


class FeeSetViewSet(CustomModelViewSet):
    queryset = FeeSet.objects.all()
    serializer_class = FeeSetSerializer
    select_related_fields = ['fee', 'mgroup']
    filterset_fields = ['fee', 'mgroup', 'year', 'month']
    search_fields = ['fee__name', 'mgroup__name']