31 lines
937 B
Python
31 lines
937 B
Python
from django.shortcuts import render
|
|
from apps.utils.viewsets import CustomModelViewSet, GenericViewSet, ListModelMixin
|
|
from apps.utils.mixins import BulkCreateModelMixin, BulkDestroyModelMixin
|
|
from .serializers import DrainSerializer, DrainEquipSerializer
|
|
from .models import Drain, DrainEquip
|
|
|
|
# Create your views here.
|
|
|
|
|
|
class DrainViewSet(CustomModelViewSet):
|
|
"""
|
|
list: 排口/污染源
|
|
|
|
排口/污染源
|
|
"""
|
|
queryset = Drain.objects.all()
|
|
serializer_class = DrainSerializer
|
|
filterset_fields = ['type', 'cate', 'mgroup']
|
|
|
|
|
|
class DrainEquipViewSet(ListModelMixin, BulkCreateModelMixin, BulkDestroyModelMixin, GenericViewSet):
|
|
"""
|
|
list: 排口/设备关系
|
|
|
|
排口/设备关系
|
|
"""
|
|
perms_map = {'get': '*', 'post:': 'drain.update', 'delete': 'drain.update'}
|
|
queryset = DrainEquip.objects.all()
|
|
serializer_class = DrainEquipSerializer
|
|
filterset_fields = ['drain', 'equipment']
|