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