factory/apps/mtm/views.py

115 lines
3.4 KiB
Python

from django.shortcuts import render
from rest_framework.decorators import action
from rest_framework.mixins import ListModelMixin
from rest_framework.response import Response
from rest_framework.exceptions import ParseError
from apps.mtm.filters import GoalFilter, MaterialFilter
from apps.mtm.models import Goal, Material, Mgroup, Shift, Team, Process, Route
from apps.mtm.serializers import (GoalSerializer, MaterialSerializer,
MgroupGoalYearSerializer, MgroupSerializer,
ShiftSerializer, TeamSerializer, ProcessSerializer, RouteSerializer)
from apps.mtm.services import get_mgroup_goals
from apps.utils.viewsets import CustomGenericViewSet, CustomModelViewSet
# Create your views here.
class MaterialViewSet(CustomModelViewSet):
"""
list:产品
产品
"""
queryset = Material.objects.all()
serializer_class = MaterialSerializer
filterset_class = MaterialFilter
search_fields = ['name', 'code', 'number', 'specification']
ordering = ['sort', 'id']
class ShiftViewSet(ListModelMixin, CustomGenericViewSet):
"""
list:班次
班次
"""
perms_map = {'get': '*'}
queryset = Shift.objects.all()
serializer_class = ShiftSerializer
search_fields = ['name']
ordering = ['id']
class MgroupViewSet(CustomModelViewSet):
"""
list:测点集
测点集
"""
queryset = Mgroup.objects.all()
serializer_class = MgroupSerializer
select_related_fields = ['create_by', 'belong_dept', 'process']
filterset_fields = ['belong_dept', 'process']
search_fields = ['number']
class TeamViewSet(CustomModelViewSet):
"""
list:班组
班组
"""
queryset = Team.objects.all()
serializer_class = TeamSerializer
select_related_fields = ['belong_dept', 'leader']
filterset_fields = ['belong_dept']
search_fields = ['name']
class GoalViewSet(CustomModelViewSet):
"""
list: 目标
目标
"""
queryset = Goal.objects.all()
serializer_class = GoalSerializer
select_related_fields = ['mgroup', 'goal_cate']
filterset_class = GoalFilter
search_fields = ['name']
ordering = ['mgroup__sort', 'goal_cate__sort', 'goal_cate__create_time']
@action(methods=['post'], detail=False, perms_map={'post': '*'}, serializer_class=MgroupGoalYearSerializer)
def mgroup_goals_year(self, request, pk=None):
"""
获取工段全年目标
获取工段全年目标
"""
sr = MgroupGoalYearSerializer(data=request.data)
sr.is_valid(raise_exception=True)
vdata = sr.validated_data
res = get_mgroup_goals(vdata['mgroup'], vdata['year'], True)
return Response(res)
class ProcessViewSet(CustomModelViewSet):
queryset = Process.objects.all()
serializer_class = ProcessSerializer
select_related_fields = ['belong_dept']
search_fields = ['name', 'cate']
filterset_fields = ['cate']
ordering = ['sort', 'create_time']
def perform_destroy(self, instance):
if Route.objects.filter(process=instance).exists():
raise ParseError('存在使用的工艺路线!')
return super().perform_destroy(instance)
class RouteViewSet(CustomModelViewSet):
queryset = Route.objects.all()
serializer_class = RouteSerializer
filterset_fields = ['material', 'process', 'is_autotask']
ordering = ['sort', 'create_time']