28 lines
1.1 KiB
Python
28 lines
1.1 KiB
Python
from django.shortcuts import render
|
|
from apps.enm.models import Mpoint, Mgroup, Team, Mrecord
|
|
from apps.utils.viewsets import CustomModelViewSet
|
|
from apps.enm.serializers import MpointSerializer, TeamSerializer, MgroupSerializer
|
|
from rest_framework.exceptions import ParseError
|
|
|
|
class MgroupViewSet(CustomModelViewSet):
|
|
queryset = Mgroup.objects.all()
|
|
serializer_class = MgroupSerializer
|
|
select_related_fields = ['create_by', 'belong_dept']
|
|
filterset_fields = ['belong_dept']
|
|
search_fields = ['number']
|
|
|
|
|
|
class TeamViewSet(CustomModelViewSet):
|
|
queryset = Team.objects.all()
|
|
serializer_class = TeamSerializer
|
|
select_related_fields = ['belong_dept', 'leader']
|
|
filterset_fields = ['belong_dept']
|
|
search_fields = ['name']
|
|
|
|
|
|
class MpointViewSet(CustomModelViewSet):
|
|
queryset = Mpoint.objects.all()
|
|
serializer_class = MpointSerializer
|
|
select_related_fields = ['create_by', 'belong_dept', 'ep_monitored', 'ep_belong', 'mgroup']
|
|
filterset_fields = ['belong_dept', 'ep_monitored', 'ep_belong', 'mgroup']
|
|
search_fields = ['number', 'code'] |