91 lines
3.5 KiB
Python
91 lines
3.5 KiB
Python
from datetime import timedelta
|
|
from django.shortcuts import render
|
|
from numpy import delete
|
|
from rest_framework.exceptions import APIException
|
|
from rest_framework.views import APIView
|
|
from rest_framework.viewsets import ModelViewSet, GenericViewSet
|
|
from rest_framework.mixins import CreateModelMixin, RetrieveModelMixin, ListModelMixin, DestroyModelMixin
|
|
from rest_framework import serializers, status
|
|
from rest_framework.response import Response
|
|
from apps.em.models import Equipment, ECheckRecord
|
|
from apps.em.serializers import DaqCreateSerializer, EChcekRecordCreateSerializer, ECheckRecordListSerializer, \
|
|
EquipmentCreateUpdateSerializer, EquipmentListSerializer
|
|
from apps.em.services import EmService
|
|
from apps.system.mixins import CreateUpdateModelAMixin, OptimizationMixin
|
|
from dateutil.relativedelta import relativedelta
|
|
from django.utils import timezone
|
|
from django.db import transaction
|
|
|
|
# Create your views here.
|
|
class EquipmentViewSet(CreateUpdateModelAMixin, OptimizationMixin, ModelViewSet):
|
|
"""
|
|
设备台账-增删改查
|
|
"""
|
|
perms_map = {'get': '*', 'post': 'equipment_create',
|
|
'put': 'equipment_update', 'delete': 'equipment_delete'}
|
|
queryset = Equipment.objects.all()
|
|
serializer_class = EquipmentListSerializer
|
|
search_fields = ['number', 'name','description']
|
|
filterset_fields = ['keeper', 'type']
|
|
ordering_fields = ['create_time']
|
|
ordering = ['-create_time']
|
|
|
|
def get_serializer_class(self):
|
|
if self.action in ['create', 'update']:
|
|
return EquipmentCreateUpdateSerializer
|
|
return super().get_serializer_class()
|
|
|
|
# Create your views here.
|
|
class EChcekRecordViewSet(CreateUpdateModelAMixin, OptimizationMixin,
|
|
CreateModelMixin, RetrieveModelMixin, ListModelMixin, DestroyModelMixin, GenericViewSet):
|
|
"""
|
|
设备校准-增删改查
|
|
"""
|
|
perms_map = {'get': '*', 'post': 'echeckrecord_create', 'delete': 'echeckrecord_delete'}
|
|
queryset = ECheckRecord.objects.all()
|
|
serializer_class = ECheckRecordListSerializer
|
|
filterset_fields = ['equipment']
|
|
ordering = ['-id']
|
|
|
|
def get_serializer_class(self):
|
|
if self.action in ['create']:
|
|
return EChcekRecordCreateSerializer
|
|
return super().get_serializer_class()
|
|
|
|
@transaction.atomic
|
|
def create(self, request, *args, **kwargs):
|
|
serializer = self.get_serializer(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
instance = serializer.save(create_by=request.user)
|
|
EmService.update_check_date(equip=instance)
|
|
return Response()
|
|
|
|
def destroy(self, request, *args, **kwargs):
|
|
instance = self.get_object()
|
|
EmService.update_check_date(equip=instance)
|
|
instance.delete()
|
|
return super().destroy(request, *args, **kwargs)
|
|
|
|
import uuid
|
|
import os
|
|
from django.conf import settings
|
|
from rest_framework.parsers import MultiPartParser
|
|
class DaqView(APIView):
|
|
"""
|
|
数据采集
|
|
"""
|
|
authentication_classes = []
|
|
permission_classes = []
|
|
parser_classes = [MultiPartParser]
|
|
|
|
def post(self, request, format=None):
|
|
data = request.data
|
|
file = data.get('file', None)
|
|
equip_num = data.get('equip_num', None)
|
|
filename = str(uuid.uuid4())
|
|
filepath = settings.BASE_DIR +'/temp/' + filename + os.path.splitext(file.name)[-1]
|
|
with open(filepath, 'wb') as f:
|
|
for chunk in file.chunks():
|
|
f.write(chunk)
|
|
return Response()
|
|
# raise APIException('解析失败') |