diff --git a/apps/wpm/datax.py b/apps/wpm/datax.py index 96628cde..611241c0 100644 --- a/apps/wpm/datax.py +++ b/apps/wpm/datax.py @@ -9,6 +9,7 @@ from datetime import datetime, timedelta from rest_framework.response import Response from rest_framework.serializers import Serializer from apps.inm.models import MIOItem, MIO +from rest_framework.exceptions import ParseError def tran_time_to_mstate(mstate_json, reminder_interval_list, work_start_time: datetime, now: datetime): if len(reminder_interval_list) != len(mstate_json): @@ -30,7 +31,10 @@ class AnaViewSet(GenericViewSet): """ data = request.data now = timezone.now() - mgroup: Mgroup = Mgroup.objects.get(id=data['mgroup']) + try: + mgroup: Mgroup = Mgroup.objects.get(id=data['mgroup']) + except Mgroup.DoesNotExist: + raise ParseError("该工段不存在,请传入工段ID") # 子状态 mstate_json = mgroup.process.mstate_json if mgroup.process.mstate_json else [] # 生产设备