from django.shortcuts import render
from rest_framework.mixins import ListModelMixin
from rest_framework.exceptions import ParseError

from apps.inm.models import WareHouse, MaterialBatch
from apps.inm.serializers import MaterialBatchSerializer, WareHourseSerializer
from apps.utils.viewsets import CustomGenericViewSet, CustomModelViewSet


# Create your views here.
class WarehouseVIewSet(CustomModelViewSet):
    """
    list: 仓库信息

    仓库信息
    """
    queryset = WareHouse.objects.all()
    serializer_class = WareHourseSerializer
    search_fields = ['name', 'number', 'place']
    ordering = ['create_time']

    def perform_destroy(self, instance):
        if MaterialBatch.objects.filter(warehouse=instance).exclude(count=0).exists():
            raise ParseError('该仓库存在物料')
        instance.delete()


class MaterialBatchViewSet(ListModelMixin, CustomGenericViewSet):
    """
    list: 物料批次

    物料批次
    """
    perms_map = {'get': '*'}
    queryset = MaterialBatch.objects.all()
    serializer_class = MaterialBatchSerializer
    select_related_fields = ['warehouse', 'material']
    filterset_fields = ['warehouse', 'material']