191 lines
7.5 KiB
Python
191 lines
7.5 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, CdSerializer
|
|
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
|
|
from apps.enp.services import get_last_envdata
|
|
from rest_framework.views import APIView
|
|
from apps.utils.mixins import MyLoggingMixin
|
|
import importlib
|
|
|
|
# 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)
|
|
edata_dict = {}
|
|
if data:
|
|
data_ids = [item["id"] for item in data]
|
|
edata = get_last_envdata(data_ids, now_10_before, now)
|
|
edata_dict = {item["equipment"]: item for item in edata}
|
|
for item in data:
|
|
item["envdata"] = edata_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"]
|
|
|
|
|
|
class CdView(MyLoggingMixin, APIView):
|
|
|
|
@swagger_auto_schema(request_body=CdSerializer)
|
|
def post(self, request, format=None):
|
|
"""执行采集数据方法
|
|
|
|
执行采集数据方法
|
|
"""
|
|
method = request.data.get("method", None)
|
|
if not method:
|
|
raise ParseError("请传入method参数")
|
|
m = method.split("(")[0]
|
|
args = method.split("(")[1].split(")")[0].split(",")
|
|
module, func = m.rsplit(".", 1)
|
|
m = importlib.import_module(module)
|
|
f = getattr(m, func)
|
|
return Response(f(*args)) |