From 005b5c1057886973ca5f764114a44afcba5d5eb1 Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmail.com>
Date: Fri, 13 Dec 2024 17:05:02 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=A2=9E=E5=8A=A0enstat=20note?=
 =?UTF-8?q?=E5=AD=97=E6=AE=B5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/enm/models.py      | 1 +
 apps/enm/serializers.py | 4 ++++
 apps/enm/views.py       | 4 ++--
 3 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/apps/enm/models.py b/apps/enm/models.py
index 7a9e5db9..ff68de4c 100644
--- a/apps/enm/models.py
+++ b/apps/enm/models.py
@@ -156,6 +156,7 @@ class EnStat(BaseModel):
     shut_sec = models.PositiveIntegerField("停机时长", default=0, help_text="s")
     run_rate = models.FloatField("运转率", default=0, help_text="%")
     ammonia_consume = models.FloatField("氨水消耗", default=0, help_text="t")
+    note = models.TextField("备注", default="", blank=True)
 
 
 class EnStat2(BaseModel):
diff --git a/apps/enm/serializers.py b/apps/enm/serializers.py
index 3b1264c0..1df10584 100644
--- a/apps/enm/serializers.py
+++ b/apps/enm/serializers.py
@@ -132,6 +132,10 @@ class MpointStatSerializer(CustomModelSerializer):
                 ret[key] = "{:.2f}".format(round(ret_one_val, 2))
         return ret
 
+class EnStatUpdateSerializer(CustomModelSerializer):
+    class Meta:
+        model = EnStat
+        fields = ["id", "note"]
 
 class EnStatSerializer(CustomModelSerializer):
     mgroup_name = serializers.CharField(source="mgroup.name", read_only=True)
diff --git a/apps/enm/views.py b/apps/enm/views.py
index 71fd3e4b..0df7dca9 100644
--- a/apps/enm/views.py
+++ b/apps/enm/views.py
@@ -1,7 +1,7 @@
 from django.conf import settings
 from apps.enm.models import Mpoint, MpointStat, EnStat, EnStat2, MpLogx, Xscript
 from apps.utils.viewsets import CustomModelViewSet, CustomGenericViewSet
-from apps.utils.mixins import BulkCreateModelMixin, BulkDestroyModelMixin, CustomListModelMixin
+from apps.utils.mixins import BulkCreateModelMixin, BulkDestroyModelMixin, CustomListModelMixin, BulkUpdateModelMixin
 from apps.enm.serializers import (MpointSerializer, MpLogxSerializer, MpointStatSerializer, 
                                   EnStatSerializer, EnStat2Serializer, ReCalSerializer, 
                                   MpointStatCorrectSerializer, EnStatAnaSerializer, XscriptSerializer, XscriptDetailSerializer)
@@ -229,7 +229,7 @@ class MpointStatViewSet(BulkCreateModelMixin, BulkDestroyModelMixin, CustomListM
         return Response(result)
 
 
-class EnStatViewSet(CustomListModelMixin, CustomGenericViewSet):
+class EnStatViewSet(CustomListModelMixin,  BulkUpdateModelMixin, CustomGenericViewSet):
     """
     list:能耗统计记录