305 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			Python
		
	
	
	
			
		
		
	
	
			305 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			Python
		
	
	
	
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 django.conf import settings
 | 
						|
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, daoru_mb
 | 
						|
from apps.utils.mixins import BulkCreateModelMixin, BulkDestroyModelMixin, BulkUpdateModelMixin
 | 
						|
from apps.utils.permission import has_perm
 | 
						|
from .filters import MaterialBatchFilter, MioFilter
 | 
						|
 | 
						|
 | 
						|
# 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']
 | 
						|
    ordering = ['-update_time']
 | 
						|
 | 
						|
    @action(methods=['post'], detail=False, serializer_class=serializers.Serializer, perms_map={'post': 'materialbatch.daoru'})
 | 
						|
    @transaction.atomic
 | 
						|
    def daoru(self, request, *args, **kwargs):
 | 
						|
        """导入物料批次
 | 
						|
 | 
						|
        导入物料
 | 
						|
        """
 | 
						|
        daoru_mb(settings.BASE_DIR + request.data.get('path', ''))
 | 
						|
        return Response()
 | 
						|
 | 
						|
 | 
						|
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']
 | 
						|
    prefetch_related_fields = ['materials']
 | 
						|
    serializer_class = MIOListSerializer
 | 
						|
    retrieve_serializer_class = MIODetailSerializer
 | 
						|
    filterset_class = MioFilter
 | 
						|
    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)
 | 
						|
 | 
						|
    @action(methods=['post'], detail=True, perms_map={'post': 'mio.submit'}, serializer_class=serializers.Serializer)
 | 
						|
    @transaction.atomic
 | 
						|
    def revert(self, request, *args, **kwargs):
 | 
						|
        """撤回
 | 
						|
 | 
						|
        撤回
 | 
						|
        """
 | 
						|
        ins = self.get_object()
 | 
						|
        user = self.request.user
 | 
						|
        if ins.state != MIO.MIO_SUBMITED:
 | 
						|
            raise ParseError('记录状态异常')
 | 
						|
        if ins.submit_user != user:
 | 
						|
            raise ParseError('非提交人不可撤回')
 | 
						|
        ins.submit_time = None
 | 
						|
        ins.state = MIO.MIO_CREATE
 | 
						|
        ins.save()
 | 
						|
        InmService.update_inm(ins, is_reverse=True)
 | 
						|
        return Response()
 | 
						|
 | 
						|
 | 
						|
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"],
 | 
						|
        "mio__inout_date": ["gte", "lte", "exact"],
 | 
						|
        "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)
 |