factory/apps/inm/views.py

83 lines
2.6 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 rest_framework import serializers
from django.utils import timezone
from rest_framework.response import Response
from apps.inm.models import WareHouse, MaterialBatch, MIO
from apps.inm.serializers import (MaterialBatchSerializer, WareHourseSerializer, MIOSerializer)
from apps.utils.viewsets import CustomGenericViewSet, CustomModelViewSet
from apps.inm.services import InmService
# 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 MIOViewSet(ListModelMixin, DestroyModelMixin, CustomGenericViewSet):
"""
list: 出入库记录
出入库记录
"""
perms_map = {'get': '*', 'delete': 'mio.delete'}
queryset = MIO.objects.all()
select_related_fields = ['create_by']
serializer_class = MIOSerializer
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.update'}, serializer_class=serializers.Serializer)
@transaction.atomic
def submit(self, request, *args, **kwargs):
"""提交
提交
"""
ins = self.get_object()
user = request.user
if ins.create_by != user:
raise PermissionDenied('非创建人不可提交')
if ins.state != MIO.MIO_CREATE:
raise ParseError('订单非创建中')
ins.submit_time = timezone.now()
ins.state = MIO.MIO_SUBMITED
ins.save()
InmService.update_inm(ins)
return Response()
class MIOItemViewSet(CustomModelViewSet):
pass