69 lines
2.5 KiB
Python
69 lines
2.5 KiB
Python
from django.shortcuts import render
|
|
from apps.enm.models import Mpoint, MpLog, MpointStat, EnStat
|
|
from apps.utils.viewsets import CustomModelViewSet, CustomGenericViewSet
|
|
from rest_framework.mixins import ListModelMixin
|
|
from apps.utils.mixins import BulkCreateModelMixin, BulkDestroyModelMixin
|
|
from apps.enm.serializers import (MpointSerializer, MpLogSerializer, MpointStatSerializer, EnStatSerializer)
|
|
from apps.enm.filters import MpointStatFilter, EnStatFilter
|
|
from apps.enm.tasks import cal_mpointstat_manual
|
|
|
|
|
|
class MpointViewSet(CustomModelViewSet):
|
|
"""
|
|
list:测点
|
|
|
|
测点
|
|
"""
|
|
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']
|
|
|
|
|
|
class MpLogViewSet(ListModelMixin, CustomGenericViewSet):
|
|
"""
|
|
list:测点原始记录
|
|
|
|
测点原始记录
|
|
"""
|
|
perms_map = {'get': '*'}
|
|
queryset = MpLog.objects.all()
|
|
serializer_class = MpLogSerializer
|
|
select_related_fields = ['mpoint']
|
|
filterset_fields = ['mpoint', 'mpoint__mgroup', 'mpoint__mgroup__belong_dept']
|
|
|
|
|
|
class MpointStatViewSet(BulkCreateModelMixin, BulkDestroyModelMixin, ListModelMixin, CustomGenericViewSet):
|
|
"""
|
|
list:测点统计记录
|
|
|
|
测点统计记录
|
|
"""
|
|
perms_map = {'get': '*', 'post': 'mpointstat.create', 'delete': 'mpointstat.delete'}
|
|
queryset = MpointStat.objects.all()
|
|
serializer_class = MpointStatSerializer
|
|
select_related_fields = ['mpoint']
|
|
filterset_class = MpointStatFilter
|
|
|
|
def perform_create(self, serializer):
|
|
ins = serializer.save()
|
|
cal_mpointstat_manual.delay(ins.mpoint.id, ins.sflog.id, ins.mgroup.id, None, None, None, None, ins.year_s, ins.month_s, ins.day_s, nex_cal=1)
|
|
|
|
def perform_destroy(self, instance):
|
|
mpoint, sflog, mgroup, year_s, month_s, day_s = instance.mpoint, instance.sflog, instance.mgroup, instance.year_s, instance.month_s, instance.day_s
|
|
instance.delete()
|
|
cal_mpointstat_manual.delay(mpoint.id, sflog.id, mgroup.id, None, None, None, None, year_s, month_s, day_s)
|
|
|
|
|
|
class EnStatViewSet(ListModelMixin, CustomGenericViewSet):
|
|
"""
|
|
list:能耗统计记录
|
|
|
|
能耗统计记录
|
|
"""
|
|
perms_map = {'get': '*'}
|
|
queryset = EnStat.objects.all()
|
|
serializer_class = EnStatSerializer
|
|
select_related_fields = ['mgroup', 'team', 'mgroup__belong_dept']
|
|
filterset_class = EnStatFilter |