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