from django.shortcuts import render
from rest_framework.mixins import ListModelMixin, DestroyModelMixin
from rest_framework.exceptions import ParseError, PermissionDenied
from rest_framework.decorators import action
from django.db import transaction
from rest_framework import serializers
from django.utils import timezone
from rest_framework.response import Response

from apps.inm.models import WareHouse, MaterialBatch, MIO, MIOItem
from apps.inm.serializers import (
    MaterialBatchSerializer, WareHourseSerializer, MIOListSerializer, MIOItemSerializer, MIODoSerializer, MIOSaleSerializer, MIOPurSerializer, MIOOtherSerializer)
from apps.utils.viewsets import CustomGenericViewSet, CustomModelViewSet
from apps.inm.services import InmService
from apps.utils.mixins import BulkCreateModelMixin, BulkDestroyModelMixin, BulkUpdateModelMixin


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


class MioDoViewSet(BulkCreateModelMixin, BulkUpdateModelMixin, CustomGenericViewSet):
    perms_map = {'post': '*', 'put': 'mio.do'}
    queryset = MIO.objects.filter(
        type__in=[MIO.MIO_TYPE_DO_IN, MIO.MIO_TYPE_DO_OUT])
    serializer_class = MIODoSerializer

    def create(self, request, *args, **kwargs):
        """
        生产领料/入库

        生产领料/入库
        """
        return super().create(request, *args, **kwargs)


class MioSaleViewSet(BulkCreateModelMixin, BulkUpdateModelMixin, CustomGenericViewSet):
    perms_map = {'post': '*', 'put': 'mio.sale'}
    queryset = MIO.objects.filter(
        type__in=[MIO.MIO_TYPE_DO_IN, MIO.MIO_TYPE_DO_OUT])
    serializer_class = MIOSaleSerializer

    def create(self, request, *args, **kwargs):
        """
        销售出库

        销售出库
        """
        return super().create(request, *args, **kwargs)


class MioPurViewSet(BulkCreateModelMixin, BulkUpdateModelMixin, CustomGenericViewSet):
    perms_map = {'post': '*', 'put': 'mio.pur'}
    queryset = MIO.objects.filter(
        type__in=[MIO.MIO_TYPE_PUR_IN])
    serializer_class = MIOPurSerializer

    def create(self, request, *args, **kwargs):
        """
        采购入库

        采购入库
        """
        return super().create(request, *args, **kwargs)


class MioOtherViewSet(BulkCreateModelMixin, BulkUpdateModelMixin, CustomGenericViewSet):
    perms_map = {'post': '*', 'put': 'mio.other'}
    queryset = MIO.objects.filter(
        type__in=[MIO.MIO_TYPE_OTHER_OUT, MIO.MIO_TYPE_OTHER_IN])
    serializer_class = MIOOtherSerializer

    def create(self, request, *args, **kwargs):
        """
        其他出入库

        其他出入库
        """
        return super().create(request, *args, **kwargs)


class MIOViewSet(ListModelMixin, BulkCreateModelMixin, BulkUpdateModelMixin, DestroyModelMixin, CustomGenericViewSet):
    """
    list: 出入库记录

    出入库记录
    """
    queryset = MIO.objects.all()
    select_related_fields = ['create_by']
    serializer_class = MIOListSerializer
    filterset_fields = ['state', 'type', 'pu_order', 'order']

    def get_serializer(self, *args, **kwargs):
        if self.action in ['create', 'update', 'partial_update']:
            type = self.request.data.get('type')
            if type in [MIO.MIO_TYPE_DO_IN, MIO.MIO_TYPE_DO_OUT]:
                return MIODoSerializer(*args, **kwargs)
            elif type in [MIO.MIO_TYPE_OTHER_IN, MIO.MIO_TYPE_OTHER_OUT]:
                return MIOOtherSerializer(*args, **kwargs)
            elif type == MIO.MIO_TYPE_SALE_OUT:
                return MIOSaleSerializer(*args, **kwargs)
            elif type == MIO.MIO_TYPE_PUR_IN:
                return MIOPurSerializer(*args, **kwargs)
        return super().get_serializer(*args, **kwargs)

    def perform_destroy(self, instance):
        if instance.state != MIO.MIO_CREATE:
            raise ParseError('非创建中不可删除')
        return super().perform_destroy(instance)

    @action(methods=['post'], detail=True, perms_map={'post': 'mio.submit'}, serializer_class=serializers.Serializer)
    @transaction.atomic
    def submit(self, request, *args, **kwargs):
        """提交

        提交
        """
        ins = self.get_object()
        if ins.inout_date is None:
            raise ParseError('出入库日期未填写')
        if ins.state != MIO.MIO_CREATE:
            raise ParseError('记录状态异常')
        ins.submit_time = timezone.now()
        ins.state = MIO.MIO_SUBMITED
        ins.submit_user = request.user
        ins.save()
        InmService.update_inm(ins)
        return Response()


class MIOItemViewSet(ListModelMixin, BulkCreateModelMixin, BulkDestroyModelMixin, CustomGenericViewSet):
    """
    list: 出入库明细

    出入库明细
    """
    perms_map = {'get': '*', 'post': '*', 'delete': '*'}
    queryset = MIOItem.objects.all()
    serializer_class = MIOItemSerializer
    select_related_fields = ['warehouse', 'mio', 'material']
    filterset_fields = ['warehouse', 'mio', 'material']
    ordering = ['create_time']

    def perform_destroy(self, instance):
        if instance.state != MIO.MIO_CREATE:
            raise ParseError('出入库记录非创建中不可删除')
        return super().perform_destroy(instance)