feat(sync task): support filter category (#1087)

TencentBlueKing · Jul 21, 2023 · e86facc · e86facc
1 parent 3d1ce2e
commit e86facc
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/src/api/bkuser_core/api/web/sync_task/serializers.py b/src/api/bkuser_core/api/web/sync_task/serializers.py
@@ -22,6 +22,10 @@ class Meta:
         fields = ("id", "display_name")
 
 
+class SyncTaskInputSLZ(serializers.Serializer):
+    category_id = serializers.IntegerField(required=False)
+
+
 class SyncTaskOutputSLZ(serializers.Serializer):
     id = serializers.CharField()
     category = CategoryForSyncTaskSerializer()

diff --git a/src/api/bkuser_core/api/web/sync_task/views.py b/src/api/bkuser_core/api/web/sync_task/views.py
@@ -12,7 +12,7 @@
 from django.conf import settings
 from rest_framework import generics
 
-from .serializers import SyncTaskOutputSLZ, SyncTaskProcessOutputSLZ
+from .serializers import SyncTaskInputSLZ, SyncTaskOutputSLZ, SyncTaskProcessOutputSLZ
 from bkuser_core.api.web.utils import get_operator
 from bkuser_core.api.web.viewset import CustomPagination
 from bkuser_core.bkiam.constants import IAMAction
@@ -27,8 +27,15 @@ class SyncTaskListApi(generics.ListAPIView):
 
     def get_queryset(self):
         operator = get_operator(self.request)
+        slz = SyncTaskInputSLZ(data=self.request.query_params)
+        slz.is_valid(raise_exception=True)
+        data = slz.validated_data
 
         queryset = SyncTask.objects.all().order_by("-create_time")
+        category_id = data.get("category_id")
+        if category_id:
+            queryset = queryset.filter(category_id=category_id)
+
         if settings.ENABLE_IAM:
             fs = Permission().make_filter_of_category(operator, IAMAction.VIEW_CATEGORY)
             queryset = queryset.filter(fs)