factory/apps/fim/views.py

26 lines
989 B
Python

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']