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']