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 django.db.models import Sum

from apps.inm.models import WareHouse, MaterialBatch, MIO, MIOItem
from apps.inm.serializers import (
    MaterialBatchSerializer, WareHourseSerializer, MIOListSerializer, MIOItemSerializer, MioItemAnaSerializer,
    MIODoSerializer, MIOSaleSerializer, MIOPurSerializer, MIOOtherSerializer, MIOItemCreateSerializer, MaterialBatchDetailSerializer, MIODetailSerializer, MIOItemTestSerializer, MIOItemPurInTestSerializer)
from apps.utils.viewsets import CustomGenericViewSet, CustomModelViewSet
from apps.inm.services import InmService
from apps.utils.mixins import BulkCreateModelMixin, BulkDestroyModelMixin, BulkUpdateModelMixin
from apps.utils.permission import has_perm
from .filters import MaterialBatchFilter


# 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
    retrieve_serializer_class = MaterialBatchDetailSerializer
    select_related_fields = ['warehouse', 'material']
    filterset_class = MaterialBatchFilter
    search_fields = ['material__name', 'material__number',
                     'material__model', 'material__specification', 'batch']


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(CustomModelViewSet):
    """
    list: 出入库记录

    出入库记录
    """
    queryset = MIO.objects.all()
    select_related_fields = ['create_by', 'belong_dept', 'do_user',
                             'submit_user', 'supplier', 'order', 'customer', 'pu_order']
    serializer_class = MIOListSerializer
    retrieve_serializer_class = MIODetailSerializer
    filterset_fields = {
        'state': ["exact", "in"],
        "type": ["exact", "in"],
        "pu_order": ["exact"],
        "order": ["exact"]
    }
    search_fields = ['number']
    data_filter = True

    def get_serializer_class(self):
        if self.action in ['create', 'update', 'partial_update']:
            type = self.request.data.get('type')
            user = self.request.user
            if type in [MIO.MIO_TYPE_DO_IN, MIO.MIO_TYPE_DO_OUT]:
                if not has_perm(user, ['mio.do']):
                    raise PermissionDenied
                return MIODoSerializer
            elif type in [MIO.MIO_TYPE_OTHER_IN, MIO.MIO_TYPE_OTHER_OUT]:
                if not has_perm(user, ['mio.other']):
                    raise PermissionDenied
                return MIOOtherSerializer
            elif type == MIO.MIO_TYPE_SALE_OUT:
                if not has_perm(user, ['mio.sale']):
                    raise PermissionDenied
                return MIOSaleSerializer
            elif type == MIO.MIO_TYPE_PUR_IN:
                if not has_perm(user, ['mio.pur']):
                    raise PermissionDenied
                return MIOPurSerializer
        return super().get_serializer_class()

    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(MIOListSerializer(instance=ins).data)


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

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

    def perform_destroy(self, instance):
        if instance.mio.state != MIO.MIO_CREATE:
            raise ParseError('出入库记录非创建中不可删除')
        if has_perm(self.request.user, ['mio.update']) is False and instance.mio.create_by != self.request.user:
            raise PermissionDenied('无权限删除')
        return super().perform_destroy(instance)

    @action(methods=['post'], detail=True, perms_map={'post': 'mioitem.test'}, serializer_class=MIOItemTestSerializer)
    @transaction.atomic
    def test(self, request, *args, **kwargs):
        """半成品检验

        半成品检验
        """
        ins: MIOItem = self.get_object()
        if ins.test_date:
            raise ParseError('该明细已检验')
        sr = MIOItemTestSerializer(instance=ins, data=request.data)
        sr.is_valid(raise_exception=True)
        sr.save()
        # 开始变动库存
        count_notok = ins.count_notok
        batch = ins.batch
        material = ins.material
        warehouse = ins.warehouse
        mb = MaterialBatch.objects.get(
            material=material, batch=batch, warehouse=warehouse)
        count_new = mb.count - count_notok
        if count_new < 0:
            raise ParseError('库存扣减失败,请确认!')
        mb.count = count_new
        mb.save()
        return Response()

    @action(methods=['post'], detail=True, perms_map={'post': 'mioitem.test'}, serializer_class=MIOItemPurInTestSerializer)
    @transaction.atomic
    def test_pur_in(self, request, *args, **kwargs):
        """入厂检验

        入厂检验
        """
        ins: MIOItem = self.get_object()
        if ins.test_date:
            raise ParseError('该明细已检验')
        sr = MIOItemPurInTestSerializer(instance=ins, data=request.data)
        sr.is_valid(raise_exception=True)
        sr.save()
        return Response()

    @action(methods=['post'], detail=False, perms_map={'post': '*'}, serializer_class=MioItemAnaSerializer)
    def sale_out_ana(self, request, *args, **kwargs):
        """交付统计数据

        交付统计数据
        """
        sr = MIOItemTestSerializer(data=request.data)
        sr.is_valid(raise_exception=True)
        vdata = sr.validated_data
        mioitems = MIOItem.objects.filter(
            mio__type='sale_out', mio__state=MIO.MIO_SUBMITED)
        if vdata.get('material_cate'):
            mioitems = mioitems.filter(
                material__cate=vdata['material_cate'])
        if vdata.get('start_date', ''):
            mioitems = mioitems.filter(
                mio__inout_date__gte=vdata['start_date'])
        if vdata.get('end_date', ''):
            mioitems = mioitems.filter(mio__inout_date__lte=vdata['end_date'])
        res = mioitems.aggregate(count=Sum('count'))
        for i in res:
            if res[i] is None:
                res[i] = 0
        return Response(res)