60 lines
2.3 KiB
Python
60 lines
2.3 KiB
Python
from django.shortcuts import render
|
|
from rest_framework.viewsets import ModelViewSet
|
|
|
|
from rest_framework import serializers, status
|
|
from rest_framework.response import Response
|
|
from apps.em.models import Equipment,Equipmentrecord
|
|
from apps.em.serializers import EquipmentSerializer,EquipmentrecordSerializer
|
|
from apps.system.mixins import CreateUpdateModelAMixin, OptimizationMixin
|
|
|
|
|
|
# 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 = EquipmentSerializer
|
|
search_fields = ['number', 'name','description']
|
|
filterset_fields = ['belong_dept', 'keeper', 'type']
|
|
ordering_fields = ['create_time']
|
|
ordering = ['-create_time']
|
|
|
|
|
|
# Create your views here.
|
|
class EquipmentrecordViewSet(CreateUpdateModelAMixin, OptimizationMixin, ModelViewSet):
|
|
"""
|
|
设备台账-增删改查
|
|
"""
|
|
perms_map = {'get': '*', 'post': 'equipmentrecord_create',
|
|
'put': 'equipmentrecord_update', 'delete': 'equipmentrecord_delete'}
|
|
queryset = Equipmentrecord.objects.all()
|
|
serializer_class = EquipmentrecordSerializer
|
|
filterset_fields = ['equipment']
|
|
ordering_fields = ['create_time']
|
|
ordering = ['-create_time']
|
|
|
|
def create(self, request, *args, **kwargs):
|
|
|
|
data = request.data
|
|
if data.get('equipment', None):
|
|
equipment = Equipment.objects.get(pk=data['equipment'])
|
|
equipment.statedm = data['state']
|
|
equipment.save()
|
|
serializer = self.get_serializer(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
serializer.save()
|
|
return Response(status=status.HTTP_200_OK)
|
|
def update(self, request, *args, **kwargs):
|
|
data = request.data
|
|
if data.get('equipment', None):
|
|
equipment = Equipment.objects.get(pk=data['equipment'])
|
|
equipment.statedm = data['state']
|
|
equipment.save()
|
|
id = self.get_object()
|
|
serializer = self.get_serializer(id, data=data)
|
|
serializer.is_valid(raise_exception=True)
|
|
serializer.save()
|
|
return Response(status=status.HTTP_200_OK) |