fix: 抽象permissions-serializer的actions字段值设置

apps/perms/serializers/application/permission.py

@@ -6,14 +6,14 @@ from django.utils.translation import ugettext_lazy as _
 
 from orgs.mixins.serializers import BulkOrgResourceModelSerializer
 from perms.models import ApplicationPermission
-from ..base import ActionsField
+from ..base import ActionsField, BasePermissionSerializer
 
 __all__ = [
     'ApplicationPermissionSerializer'
 ]
 
 
-class ApplicationPermissionSerializer(BulkOrgResourceModelSerializer):
+class ApplicationPermissionSerializer(BasePermissionSerializer):
     actions = ActionsField(required=False, allow_null=True, label=_("Actions"))
     category_display = serializers.ReadOnlyField(source='get_category_display', label=_('Category display'))
     type_display = serializers.ReadOnlyField(source='get_type_display', label=_('Type display'))
@@ -46,15 +46,7 @@ class ApplicationPermissionSerializer(BulkOrgResourceModelSerializer):
             'applications_amount': {'label': _('Applications amount')},
         }
 
-    def __init__(self, *args, **kwargs):
-        super().__init__(*args, **kwargs)
-        self.set_actions_choices()
-
-    def set_actions_choices(self):
-        actions = self.fields.get('actions')
-        if not actions:
-            return
-        choices = actions._choices
+    def _filter_actions_choices(self, choices):
         if request := self.context.get('request'):
             category = request.query_params.get('category')
         else:
@@ -62,8 +54,7 @@ class ApplicationPermissionSerializer(BulkOrgResourceModelSerializer):
         exclude_choices = ApplicationPermission.get_exclude_actions_choices(category=category)
         for choice in exclude_choices:
             choices.pop(choice, None)
-        actions._choices = choices
-        actions.default = list(choices.keys())
+        return choices
 
     @classmethod
     def setup_eager_loading(cls, queryset):

apps/perms/serializers/asset/permission.py

@@ -9,12 +9,12 @@ from orgs.mixins.serializers import BulkOrgResourceModelSerializer
 from perms.models import AssetPermission, Action
 from assets.models import Asset, Node, SystemUser
 from users.models import User, UserGroup
-from ..base import ActionsField
+from ..base import ActionsField, BasePermissionSerializer
 
 __all__ = ['AssetPermissionSerializer']
 
 
-class AssetPermissionSerializer(BulkOrgResourceModelSerializer):
+class AssetPermissionSerializer(BasePermissionSerializer):
     actions = ActionsField(required=False, allow_null=True, label=_("Actions"))
     is_valid = serializers.BooleanField(read_only=True, label=_("Is valid"))
     is_expired = serializers.BooleanField(read_only=True, label=_('Is expired'))

apps/perms/serializers/base.py

@@ -1,7 +1,8 @@
 from rest_framework import serializers
 from perms.models import Action
+from orgs.mixins.serializers import BulkOrgResourceModelSerializer
 
-__all__ = ['ActionsDisplayField', 'ActionsField']
+__all__ = ['ActionsDisplayField', 'ActionsField', 'BasePermissionSerializer']
 
 
 class ActionsField(serializers.MultipleChoiceField):
@@ -24,3 +25,21 @@ class ActionsDisplayField(ActionsField):
         choices = dict(Action.CHOICES)
         return [choices.get(i) for i in values]
 
+
+class BasePermissionSerializer(BulkOrgResourceModelSerializer):
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.set_actions_field()
+
+    def set_actions_field(self):
+        actions = self.fields.get('actions')
+        if not actions:
+            return
+        choices = actions._choices
+        choices = self._filter_actions_choices(choices)
+        actions._choices = choices
+        actions.default = list(choices.keys())
+
+    def _filter_actions_choices(self, choices):
+        return choices