pref: 修改授权 api，去掉不用的

apps/tickets/serializers/ticket/apply_asset.py

@@ -1,40 +1,40 @@
 from django.utils.translation import ugettext_lazy as _
 from rest_framework import serializers
 
-from perms.serializers.permission import ActionChoicesField
-from perms.models import AssetPermission
-from orgs.utils import tmp_to_org
 from assets.models import Asset, Node
-
+from common.drf.fields import ObjectRelatedField
+from perms.models import AssetPermission
+from perms.serializers.permission import ActionChoicesField
 from tickets.models import ApplyAssetTicket
+from .common import BaseApplyAssetSerializer
 from .ticket import TicketApplySerializer
-from .common import BaseApplyAssetApplicationSerializer
 
-__all__ = ['ApplyAssetSerializer', 'ApplyAssetDisplaySerializer', 'ApproveAssetSerializer']
+__all__ = ['ApplyAssetSerializer']
 
 asset_or_node_help_text = _("Select at least one asset or node")
 
 
-class ApplyAssetSerializer(BaseApplyAssetApplicationSerializer, TicketApplySerializer):
-    apply_actions = ActionChoicesField(required=True, allow_empty=False)
+class ApplyAssetSerializer(BaseApplyAssetSerializer, TicketApplySerializer):
+    apply_assets = ObjectRelatedField(queryset=Asset.objects, many=True, required=False, label=_('Apply assets'))
+    apply_nodes = ObjectRelatedField(queryset=Node.objects, many=True, required=False, label=_('Apply nodes'))
+    apply_actions = ActionChoicesField(required=False, allow_null=True, label=_("Apply actions"))
     permission_model = AssetPermission
 
-    class Meta:
+    class Meta(TicketApplySerializer.Meta):
         model = ApplyAssetTicket
+        fields_mini = ['id', 'title']
         writeable_fields = [
-            'id', 'title', 'type', 'apply_nodes', 'apply_assets',
+            'id', 'title', 'apply_nodes', 'apply_assets',
             'apply_accounts', 'apply_actions', 'org_id', 'comment',
             'apply_date_start', 'apply_date_expired'
         ]
-        fields = TicketApplySerializer.Meta.fields + writeable_fields + [
-            'apply_permission_name', 'apply_actions_display'
-        ]
+        fields = TicketApplySerializer.Meta.fields + writeable_fields + ['apply_permission_name', ]
         read_only_fields = list(set(fields) - set(writeable_fields))
         ticket_extra_kwargs = TicketApplySerializer.Meta.extra_kwargs
         extra_kwargs = {
-            'apply_nodes': {'required': False, 'allow_empty': True},
-            'apply_assets': {'required': False, 'allow_empty': True},
-            'apply_accounts': {'required': False, 'allow_empty': True},
+            'apply_nodes': {'required': False},
+            'apply_assets': {'required': False},
+            'apply_accounts': {'required': False},
         }
         extra_kwargs.update(ticket_extra_kwargs)
 
@@ -45,9 +45,11 @@ class ApplyAssetSerializer(BaseApplyAssetApplicationSerializer, TicketApplySeria
         return self.filter_many_to_many_field(Asset, assets)
 
     def validate(self, attrs):
+        attrs['type'] = 'apply_asset'
         attrs = super().validate(attrs)
         if self.is_final_approval and (
-                not attrs.get('apply_nodes') and not attrs.get('apply_assets')
+                not attrs.get('apply_nodes')
+                and not attrs.get('apply_assets')
         ):
             raise serializers.ValidationError({
                 'apply_nodes': asset_or_node_help_text,
@@ -56,29 +58,7 @@ class ApplyAssetSerializer(BaseApplyAssetApplicationSerializer, TicketApplySeria
 
         return attrs
 
-
-class ApproveAssetSerializer(ApplyAssetSerializer):
-    class Meta(ApplyAssetSerializer.Meta):
-        read_only_fields = ApplyAssetSerializer.Meta.read_only_fields + [
-            'title', 'type'
-        ]
-
-
-class ApplyAssetDisplaySerializer(ApplyAssetSerializer):
-    apply_nodes = serializers.SerializerMethodField()
-    apply_assets = serializers.SerializerMethodField()
-
-    class Meta:
-        model = ApplyAssetSerializer.Meta.model
-        fields = ApplyAssetSerializer.Meta.fields
-        read_only_fields = fields
-
-    @staticmethod
-    def get_apply_nodes(instance):
-        with tmp_to_org(instance.org_id):
-            return instance.apply_nodes.values_list('id', flat=True)
-
-    @staticmethod
-    def get_apply_assets(instance):
-        with tmp_to_org(instance.org_id):
-            return instance.apply_assets.values_list('id', flat=True)
+    @classmethod
+    def setup_eager_loading(cls, queryset):
+        queryset = queryset.prefetch_related('apply_nodes', 'apply_assets')
+        return queryset