From 53193650f661f86e053db53e146a36b8ac5b45dd Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmail.com>
Date: Mon, 26 Feb 2024 10:48:47 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=A2=9E=E5=8A=A0ecate=E7=AD=9B?=
 =?UTF-8?q?=E9=80=89=E6=9D=A1=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/em/filters.py |  1 +
 apps/em/views.py   | 10 ++++++++++
 2 files changed, 11 insertions(+)

diff --git a/apps/em/filters.py b/apps/em/filters.py
index c3fee8e4..57920f30 100644
--- a/apps/em/filters.py
+++ b/apps/em/filters.py
@@ -19,6 +19,7 @@ class EquipFilterSet(filters.FilterSet):
             "mgroup": ["exact"],
             "mgroup__name": ["exact"],
             "name": ["exact", "contains"],
+            "cate": ["exact", "in"],
             "cate__code": ['exact', 'in', 'contains'],
             "cate__is_for_safe": ['exact'],
             "cate__is_for_enp": ['exact'],
diff --git a/apps/em/views.py b/apps/em/views.py
index 61da31f2..3985909f 100644
--- a/apps/em/views.py
+++ b/apps/em/views.py
@@ -17,6 +17,11 @@ from django.conf import settings
 
 
 class EcateViewSet(CustomModelViewSet):
+    """
+    list:设备分类
+
+    设备分类
+    """
     queryset = Ecate.objects.all()
     serializer_class = EcateSerializer
     ordering = ['id', 'type', 'code', 'create_time']
@@ -30,6 +35,11 @@ class EcateViewSet(CustomModelViewSet):
 
 
 class EquipmentViewSet(CustomModelViewSet):
+    """
+    list:设备列表
+
+    设备列表
+    """
     queryset = Equipment.objects.all()
     serializer_class = EquipmentSerializer
     select_related_fields = ['create_by', 'belong_dept', 'keeper', 'mgroup']