实现BaseModelSerializer，支持序列化字段过滤

Author: hhyo

sql_api/api_views/sql_workflow.py

@@ -27,7 +27,6 @@
     ExecuteCheckSerializer,
     ExecuteCheckResultSerializer,
     SqlWorkflowSerializer,
-    SqlWorkflowDetailSerializer,
 )
 
 
@@ -64,12 +63,12 @@ def post(self, request):
         summary="获取SQL工单列表",
         description="获取SQL工单列表，支持筛选、分页、检索等",
         request=SqlWorkflowSerializer,
-        responses={200: SqlWorkflowSerializer},
+        responses={200: SqlWorkflowSerializer(exclude=["sql_content", "display_content"])},
     ),
     retrieve=extend_schema(
         summary="获取SQL工单详情",
         description="通过工单ID获取工单详情",
-        responses={200: SqlWorkflowDetailSerializer},
+        responses={200: SqlWorkflowSerializer},
     ),
     rollback_sql=extend_schema(
         summary="获取SQL工单回滚语句",
@@ -86,6 +85,7 @@ def post(self, request):
 )
 class SqlWorkflowView(viewsets.ModelViewSet):
     permission_classes = [IsAuthenticated, SqlWorkFlowViewPermission]
+    serializer_class = SqlWorkflowSerializer
     pagination_class = BootStrapTablePagination
     filter_backends = [
         filters.SearchFilter,
@@ -118,10 +118,13 @@ def get_queryset(self):
         queryset = SqlWorkflow.objects.filter(**filter_dict).order_by("-id")
         return self.get_serializer_class().setup_eager_loading(queryset)
 
-    def get_serializer_class(self):
+    def get_serializer(self, *args, **kwargs):
+
+        serializer_class = self.get_serializer_class()
+        kwargs.setdefault('context', self.get_serializer_context())
         if self.action == "retrieve":
-            return SqlWorkflowDetailSerializer
-        return SqlWorkflowSerializer
+            return serializer_class(*args, **kwargs)
+        return serializer_class(exclude=["sql_content", "display_content"], *args, **kwargs)
 
     @action(methods=["get"], detail=True, pagination_class=None, filter_backends=[], search_fields=None)
     def rollback_sql(self, request, *args, **kwargs):

sql_api/serializers/__init__.py

@@ -0,0 +1,40 @@
+# -*- coding: UTF-8 -*-
+""" 
+@author: hhyo
+@license: Apache Licence
+@file: __init__.py
+@time: 2022/10/22
+"""
+__author__ = 'hhyo'
+
+from rest_framework import serializers
+
+
+class BaseModelSerializer(serializers.ModelSerializer):
+    """BaseModelSerializer，主要是引入过滤和排除字段的方法"""
+
+    def __init__(self, *args, **kwargs):
+        """
+        ``fields`` 需要保留的字段列表
+        ``exclude`` 需要排除的字段列表
+        """
+        fields = kwargs.pop("fields", None)
+        exclude = kwargs.pop("exclude", None)
+        super(BaseModelSerializer, self).__init__(*args, **kwargs)
+
+        for field_name in set(self.fields.keys()):
+            if not any([fields, exclude]):
+                break
+            if fields and field_name in fields:
+                continue
+            if exclude and field_name not in exclude:
+                continue
+            self.fields.pop(field_name, None)
+
+    @staticmethod
+    def setup_eager_loading(queryset):
+        """
+        Perform necessary eager loading of data.
+        https://ses4j.github.io/2015/11/23/optimizing-slow-django-rest-framework-performance/
+        """
+        pass

sql_api/serializers/sql_workflow.py

@@ -17,6 +17,7 @@
 from sql.engines import ReviewSet, get_engine
 from sql.engines.models import ReviewResult
 from sql.models import Instance, SqlWorkflow
+from sql_api.serializers import BaseModelSerializer
 
 logger = logging.getLogger("default")
 
@@ -69,57 +70,18 @@ class ExecuteCheckResultSerializer(serializers.Serializer):
     affected_rows = serializers.IntegerField(read_only=True)
 
 
-class SqlWorkflowSerializer(serializers.ModelSerializer):
+class SqlWorkflowSerializer(BaseModelSerializer):
     """SQL工单"""
 
     instance_name = serializers.CharField(source="instance.instance_name")
-
-    def __init__(self, *args, **kwargs):
-        """
-        ``fields`` 需要保留的字段列表
-        ``exclude`` 需要排除的字段列表
-        """
-        fields = kwargs.pop("fields", None)
-        exclude = kwargs.pop("exclude", None)
-        super(SqlWorkflowSerializer, self).__init__(*args, **kwargs)
-
-        for field_name in set(self.fields.keys()):
-            if not any([fields, exclude]):
-                break
-            if fields and field_name in fields:
-                continue
-            if exclude and field_name not in exclude:
-                continue
-            self.fields.pop(field_name, None)
+    sql_content = serializers.CharField(source="sqlworkflowcontent.sql_content")
+    display_content = serializers.SerializerMethodField()
 
     @staticmethod
     def setup_eager_loading(queryset):
-        """
-        Perform necessary eager loading of data.
-        https://ses4j.github.io/2015/11/23/optimizing-slow-django-rest-framework-performance/
-        """
         queryset = queryset.select_related("instance")
         return queryset
 
-    @staticmethod
-    def rollback_sql(obj):
-        try:
-            query_engine = get_engine(instance=obj.instance)
-            return query_engine.get_rollback(workflow=obj)
-        except Exception as msg:
-            logger.error(traceback.format_exc())
-            raise serializers.ValidationError({"errors": msg})
-
-    class Meta:
-        model = SqlWorkflow
-        fields = "__all__"
-
-
-class SqlWorkflowDetailSerializer(SqlWorkflowSerializer):
-    instance_name = serializers.CharField(source="instance.instance_name")
-    sql_content = serializers.CharField(source="sqlworkflowcontent.sql_content")
-    display_content = serializers.SerializerMethodField()
-
     @extend_schema_field(field=serializers.ListField(child=ReviewResultSerializer()))
     def get_display_content(self, obj):
         """获取工单详情用于列表展示的内容，区分不同的状态进行转换"""
@@ -161,6 +123,15 @@ def get_display_content(self, obj):
             rows = obj.sqlworkflowcontent.review_content
         return json.loads(rows)
 
+    @staticmethod
+    def rollback_sql(obj):
+        try:
+            query_engine = get_engine(instance=obj.instance)
+            return query_engine.get_rollback(workflow=obj)
+        except Exception as msg:
+            logger.error(traceback.format_exc())
+            raise serializers.ValidationError({"errors": msg})
+
     class Meta:
         model = SqlWorkflow
         fields = "__all__"