factory/apps/pm/views.py

255 lines
9.9 KiB
Python

from django.db import transaction
from rest_framework.decorators import action
from rest_framework.exceptions import ParseError
from rest_framework.response import Response
from rest_framework.serializers import Serializer
from apps.utils.serializers import PkSerializer
from apps.utils.viewsets import CustomModelViewSet
from .filters import MtaskFilter, UtaskFilter
from .models import Mtask, Utask, Mtaskb
from .serializers import (MtaskSerializer, SchedueSerializer, UtaskSerializer,
MtaskDaySerializer, MtaskAddInfoSerializer,
SchedueMtasksSerializer, MtaskbSerializer,
MtaskbUpdateSerializer, MtaskbAddSerializer)
from .services import PmService
from django.utils import timezone
# Create your views here.
class UtaskViewSet(CustomModelViewSet):
"""
list: 生产大任务
生产大任务
"""
queryset = Utask.objects.all()
serializer_class = UtaskSerializer
filterset_class = UtaskFilter
select_related_fields = ['material']
ordering_fields = ['priority', 'start_date']
ordering = ["priority", '-start_date']
def perform_destroy(self, instance):
if instance.state >= Utask.UTASK_WORKING:
raise ParseError('该任务状态不可删除')
return super().perform_destroy(instance)
@action(methods=['post'], detail=True, perms_map={'post': 'utask.submit'}, serializer_class=Serializer)
@transaction.atomic
def submit(self, request, *args, **kwargs):
"""提交任务
提交任务
"""
obj = self.get_object()
PmService.utask_submit(obj)
return Response()
@action(methods=['post'], detail=True, perms_map={'post': 'utask.toggle'}, serializer_class=Serializer)
@transaction.atomic
def toggle(self, request, *args, **kwargs):
"""停止/恢复任务
停止/恢复任务
"""
obj = self.get_object()
user = request.user
now = timezone.now()
if obj.state == Utask.UTASK_WORKING:
obj.state = Utask.UTASK_STOP
obj.update_by = user
obj.update_time = now
obj.save()
# 停止所有小任务
Mtask.objects.filter(
utask=obj, state=Mtask.MTASK_ASSGINED).update(state=Mtask.MTASK_STOP, update_by=user, update_time=now)
elif obj.state == Utask.UTASK_STOP:
obj.state = Utask.UTASK_WORKING
obj.update_by = user
obj.update_time = now
obj.save()
# 恢复所有小任务
Mtask.objects.filter(
utask=obj, state=Mtask.MTASK_STOP).update(state=Mtask.MTASK_ASSGINED, update_by=user, update_time=now)
else:
raise ParseError('该状态下不可变更')
return Response()
@action(methods=['post'], detail=False, perms_map={'post': 'utask.schedue'}, serializer_class=SchedueSerializer)
@transaction.atomic
def schedue_utasks(self, request, *args, **kwargs):
"""从多个订单明细生成大任务
从多个订单明细生成大任务
"""
sr = SchedueSerializer(data=request.data)
sr.is_valid(raise_exception=True)
vdata = sr.validated_data
PmService.make_utasks_from_orderitems(request.user,
request.data['orderitems'], vdata['start_date'], vdata['end_date'], vdata.get('belong_dept', None))
return Response()
@action(methods=['post'], detail=False, perms_map={'post': 'utask.schedue'}, serializer_class=SchedueMtasksSerializer)
@transaction.atomic
def schedue_mtasks(self, request, *args, **kwargs):
"""任务分解(支持分解到天或仅仅到工段)
任务分解
"""
sr = SchedueMtasksSerializer(data=request.data)
sr.is_valid(raise_exception=True)
vdata = sr.validated_data
utasks = Utask.objects.filter(
id__in=vdata['ids'], state=Utask.UTASK_CREATED)
for i in utasks:
PmService.schedue_mtasks(request.user, i, vdata.get("schedule_type", "to_day"))
return Response()
@action(methods=['post'], detail=False, perms_map={'post': 'utask.assgin'}, serializer_class=PkSerializer)
@transaction.atomic
def assgin(self, request, *args, **kwargs):
"""下达任务
下达任务
"""
ids = request.data.get('ids', [])
user = request.user
now = timezone.now()
utasks = Utask.objects.filter(
id__in=ids, state=Utask.UTASK_DECOMPOSE) # 已分解的任务
Mtask.objects.filter(utask__in=utasks, state=Mtask.MTASK_CREATED).update(
state=Mtask.MTASK_ASSGINED, update_by=user, update_time=now)
utasks.update(
state=Utask.UTASK_ASSGINED, update_by=user, update_time=now)
# 此处要更新订单状态
from apps.sam.models import OrderItem, Order
orderIds = OrderItem.objects.filter(
utask__in=utasks).values_list('order', flat=True).distinct()
Order.objects.filter(id__in=orderIds, state=Order.ORDER_SUBMITED).update(
state=Order.ORDER_DOING)
return Response()
class MtaskViewSet(CustomModelViewSet):
"""
list: 生产任务
生产任务
"""
queryset = Mtask.objects.all()
serializer_class = MtaskSerializer
filterset_class = MtaskFilter
select_related_fields = ['material_in', 'material_out', 'mgroup']
prefetch_related_fields = ['mlog_mtask', 'b_mtask']
ordering_fields = ["utask__priority", 'start_date', 'mgroup__process__sort', 'create_time']
ordering = ["utask__priority", '-start_date', 'route__sort', 'mgroup__process__sort', '-create_time']
@action(methods=['post'], detail=False, perms_map={'post': '*'}, serializer_class=MtaskDaySerializer)
@transaction.atomic
def day(self, request):
"""获取指定车间指定日期的代表工序生产任务
获取指定车间指定日期的代表工序生产任务
"""
sr = MtaskDaySerializer(data=request.data)
sr.is_valid(raise_exception=True)
vdata = sr.validated_data
mtasks = Mtask.objects.filter(
is_count_utask=True,
mgroup__belong_dept__name=vdata['belong_dept_name'], start_date__lte=vdata['date'], end_date__gte=vdata['date']).order_by(
'utask', 'mgroup__process__sort', 'create_time'
)
data = MtaskSerializer(instance=mtasks, many=True).data
return Response(data)
@action(methods=['get'], detail=True, perms_map={'get': '*'})
@transaction.atomic
def related(self, request, *args, **kwargs):
"""获取已下达的所有的兄弟小任务
获取已下达的所有的兄弟小任务
"""
obj = self.get_object()
rdata = MtaskSerializer(instance=obj.related, many=True).data
return Response(rdata)
def perform_destroy(self, instance):
if instance.state != Mtask.MTASK_CREATED:
raise ParseError('该任务非创建中不可删除')
return super().perform_destroy(instance)
@action(methods=['post'], detail=True, perms_map={'post': 'mtask.submit'}, serializer_class=PkSerializer)
@transaction.atomic
def submit(self, request, *args, **kwargs):
"""提交任务(根据任务ID)
提交任务后不可更新日志
"""
mtask: Mtask = self.get_object()
user = request.user
PmService.mtask_submit(mtask, user)
return Response()
@action(methods=['post'], detail=True, perms_map={'post': 'mtask.submit'}, serializer_class=Serializer)
@transaction.atomic
def submit_related_same_day(self, request, *args, **kwargs):
"""提交同一天该部门下大任务的所有兄弟小任务
提交同一天的该部门下大任务的所有兄弟小任务
"""
mtask = self.get_object()
mtasks = mtask.related
for mtask in mtasks:
PmService.mtask_submit(mtask, self.request.user)
return Response()
@action(methods=['post'], detail=True, perms_map={'post': 'mlog.create'}, serializer_class=MtaskAddInfoSerializer)
@transaction.atomic
def add_info(self, request, *args, **kwargs):
"""添加与任务相关的日志信息
添加与任务相关的日志信息
"""
obj = self.get_object()
sr = MtaskAddInfoSerializer(instance=obj, data=request.data)
sr.is_valid(raise_exception=True)
sr.save()
return Response()
@action(methods=['post'], detail=True, perms_map={'post': 'mtaskb.create'}, serializer_class=MtaskbAddSerializer)
@transaction.atomic
def add_mtaskb(self, request, *args, **kwargs):
mtask = self.get_object()
if mtask.state != Mtask.MTASK_ASSGINED:
raise ParseError('该任务不可分配')
sr = MtaskbAddSerializer(data=request.data, many=True)
sr.is_valid(raise_exception=True)
vdata = sr.validated_data
total = 0
userIds = []
for v in vdata:
total += v['count']
userIds.append(v['handle_user'].id)
if total != mtask.count:
raise ParseError('任务数量与分配数量不匹配')
for v in vdata:
mtaskb, _ = Mtaskb.objects.get_or_create(mtask=mtask, handle_user=v['handle_user'])
mtaskb.count = v['count']
mtaskb.save()
dobjs = Mtaskb.objects.filter(mtask=mtask).exclude(handle_user__in=userIds)
from apps.wpm.models import Mlog
if Mlog.objects.filter(mtaskb__in=dobjs).exists():
raise ParseError('已存在工作日志,无法修改分配')
else:
dobjs.delete()
return Response()
class MtaskbViewSet(CustomModelViewSet):
queryset = Mtaskb.objects.all()
serializer_class = MtaskbSerializer
update_serializer_class = MtaskbUpdateSerializer
filterset_fields = {"mtask": ["exact"], "handle_user": ["exact"], "mtask__mgroup": ["exact"]}
ordering = ['id']