177 lines
7.4 KiB
Python
177 lines
7.4 KiB
Python
from django.shortcuts import render
|
|
from drf_yasg.utils import swagger_auto_schema
|
|
from drf_yasg import openapi
|
|
from django.utils import timezone
|
|
from apps.em.models import Equipment, EcheckRecord, EInspect, Ecate
|
|
from apps.utils.viewsets import CustomModelViewSet, CustomGenericViewSet
|
|
from apps.em.serializers import EquipmentSerializer, EcheckRecordSerializer, EInspectSerializer, EcateSerializer
|
|
from apps.em.filters import EquipFilterSet
|
|
from rest_framework.exceptions import ParseError
|
|
from rest_framework.mixins import ListModelMixin, CreateModelMixin, DestroyModelMixin
|
|
from dateutil.relativedelta import relativedelta
|
|
from rest_framework.decorators import action
|
|
from rest_framework.serializers import Serializer
|
|
from django.db import transaction
|
|
from apps.em.services import daoru_equipment
|
|
from rest_framework.response import Response
|
|
from django.conf import settings
|
|
from django.db.models import Count, Case, When, IntegerField, Max, OuterRef, Subquery
|
|
from datetime import datetime, timedelta
|
|
|
|
# Create your views here.
|
|
|
|
|
|
class EcateViewSet(CustomModelViewSet):
|
|
"""
|
|
list:设备分类
|
|
|
|
设备分类
|
|
"""
|
|
|
|
queryset = Ecate.objects.all()
|
|
serializer_class = EcateSerializer
|
|
ordering = ["id", "type", "code", "create_time"]
|
|
filterset_fields = {"type": ["exact", "in"], "code": ["exact", "in", "contains"], "is_for_safe": ["exact"], "is_for_enp": ["exact"], "is_car": ["exact"]}
|
|
|
|
|
|
class EquipmentViewSet(CustomModelViewSet):
|
|
"""
|
|
list:设备列表
|
|
|
|
设备列表
|
|
"""
|
|
|
|
queryset = Equipment.objects.get_queryset(all=True)
|
|
serializer_class = EquipmentSerializer
|
|
select_related_fields = ["create_by", "belong_dept", "keeper", "mgroup"]
|
|
search_fields = ["number", "name"]
|
|
filterset_class = EquipFilterSet
|
|
|
|
def get_queryset(self):
|
|
if self.request.method == 'GET' and (not self.request.query_params.get('is_deleted', None)):
|
|
self.queryset = Equipment.objects.all()
|
|
return super().get_queryset()
|
|
|
|
@action(methods=["post"], detail=False, perms_map={"post": "equipment.create"}, serializer_class=Serializer)
|
|
@transaction.atomic
|
|
def daoru(self, request, *args, **kwargs):
|
|
"""导入
|
|
|
|
导入
|
|
"""
|
|
daoru_equipment(settings.BASE_DIR + request.data.get("path", ""))
|
|
return Response()
|
|
|
|
@swagger_auto_schema(
|
|
manual_parameters=[
|
|
openapi.Parameter(name="has_envdata", in_=openapi.IN_QUERY, description="Include envdata in the response", type=openapi.TYPE_STRING, enum=["yes", "no"], required=False),
|
|
openapi.Parameter(name="query", in_=openapi.IN_QUERY, description="定制返回数据", type=openapi.TYPE_STRING, required=False),
|
|
]
|
|
)
|
|
def list(self, request, *args, **kwargs):
|
|
return super().list(request, *args, **kwargs)
|
|
|
|
def add_info_for_list(self, data):
|
|
if self.request.query_params.get("has_envdata", "no") == "yes":
|
|
now = timezone.localtime()
|
|
now_10_before = now - timezone.timedelta(minutes=10)
|
|
data_ids = [item["id"] for item in data]
|
|
from apps.enp.models import EnvData
|
|
from apps.enp.serializers import EnvDataSerializer
|
|
|
|
# 子查询获取每个 equipment_id 对应的最大时间戳
|
|
# 后面可以考虑从缓存里拿
|
|
last_time_subquery = EnvData.objects.filter(equipment_id=OuterRef("equipment_id"), timex__gte=now_10_before, timex__lte=now).order_by("-timex").values("timex")[:1]
|
|
|
|
# 主查询，获取每个 equipment_id 对应的完整记录
|
|
last_envdata_qs = EnvData.objects.filter(equipment_id__in=data_ids, timex=Subquery(last_time_subquery))
|
|
envdata = EnvDataSerializer(last_envdata_qs, many=True).data
|
|
envdata_dict = {item["equipment"]: item for item in envdata}
|
|
for item in data:
|
|
item["envdata"] = envdata_dict.get(item["id"], {})
|
|
return data
|
|
|
|
@action(methods=["get"], detail=False, perms_map={"get": "*"})
|
|
def count_running_state(self, request, *args, **kwargs):
|
|
"""当前运行状态统计
|
|
|
|
当前运行状态统计
|
|
"""
|
|
queryset = self.filter_queryset(self.get_queryset())
|
|
result = queryset.aggregate(
|
|
count=Count("id"),
|
|
count_running=Count(Case(When(running_state=10, then=1), output_field=IntegerField())),
|
|
count_standby=Count(Case(When(running_state=20, then=1), output_field=IntegerField())),
|
|
count_stop=Count(Case(When(running_state=30, then=1), output_field=IntegerField())),
|
|
count_fail=Count(Case(When(running_state=40, then=1), output_field=IntegerField())),
|
|
count_offline=Count(Case(When(running_state=50, then=1), output_field=IntegerField())),
|
|
count_ok=Count(Case(When(state=Equipment.EQUIP_STATE_OK, then=1), output_field=IntegerField())),
|
|
count_limit=Count(Case(When(state=Equipment.EQUIP_STATE_LIMIT, then=1), output_field=IntegerField())),
|
|
count_fix=Count(Case(When(state=Equipment.EQUIP_STATE_FIX, then=1), output_field=IntegerField())),
|
|
count_disable=Count(Case(When(state=Equipment.EQUIP_STATE_DISABLE, then=1), output_field=IntegerField())),
|
|
count_scrap=Count(Case(When(state=Equipment.EQUIP_STATE_SCRAP, then=1), output_field=IntegerField())),
|
|
)
|
|
json_result = {
|
|
"count": result["count"],
|
|
"count_running": result["count_running"],
|
|
"count_standby": result["count_standby"],
|
|
"count_stop": result["count_stop"],
|
|
"count_fail": result["count_fail"],
|
|
"count_offline": result["count_offline"],
|
|
"count_ok": result["count_ok"],
|
|
"count_limit": result["count_limit"],
|
|
"count_fix": result["count_fix"],
|
|
"count_disable": result["count_disable"],
|
|
"count_scrap": result["count_scrap"]
|
|
}
|
|
|
|
return Response(json_result)
|
|
|
|
|
|
class EcheckRecordViewSet(ListModelMixin, CreateModelMixin, DestroyModelMixin, CustomGenericViewSet):
|
|
"""
|
|
list:校准/检定记录
|
|
|
|
校准/检定记录
|
|
"""
|
|
|
|
perms_map = {"get": "*", "post": "echeckrecord.create", "put": "echeckrecord.update", "delete": "echeckrecord.delete"}
|
|
queryset = EcheckRecord.objects.all()
|
|
serializer_class = EcheckRecordSerializer
|
|
select_related_fields = ["equipment", "create_by"]
|
|
filterset_fields = ["equipment", "result"]
|
|
|
|
def perform_create(self, serializer):
|
|
instance = serializer.save()
|
|
equipment = instance.equipment
|
|
if equipment.cycle:
|
|
equipment.check_date = instance.check_date
|
|
equipment.next_check_date = instance.check_date + relativedelta(months=equipment.cycle)
|
|
equipment.save()
|
|
|
|
def perform_destroy(self, instance):
|
|
instance.delete()
|
|
equipment = instance.equipment
|
|
er = EcheckRecord.objects.filter(equipment=equipment).order_by("check_date").last()
|
|
if er:
|
|
equipment.check_date = instance.check_date
|
|
equipment.next_check_date = instance.check_date + relativedelta(months=equipment.cycle)
|
|
equipment.save()
|
|
else:
|
|
equipment.check_date = None
|
|
equipment.next_check_date = None
|
|
equipment.save()
|
|
|
|
|
|
class EInspectViewSet(CustomModelViewSet):
|
|
"""
|
|
list:巡检记录
|
|
|
|
巡检记录
|
|
"""
|
|
|
|
queryset = EInspect.objects.all()
|
|
serializer_class = EInspectSerializer
|
|
select_related_fields = ["equipment", "inspect_user"]
|
|
filterset_fields = ["equipment"]
|