pref: 修改授权规则

apps/perms/api/user_permission/assets/api.py

@@ -14,16 +14,15 @@ __all__ = [
     'MyFavoriteGrantedAssetsApi', 'UserDirectGrantedAssetsAsTreeApi',
     'MyUngroupAssetsAsTreeApi',
     'UserAllGrantedAssetsApi', 'MyAllGrantedAssetsApi', 'MyAllAssetsAsTreeApi',
-    'UserGrantedNodeAssetsApi',
-    'MyGrantedNodeAssetsApi',
+    'UserGrantedNodeAssetsApi', 'MyGrantedNodeAssetsApi',
 ]
 
 logger = get_logger(__name__)
 
 
 class UserDirectGrantedAssetsApi(
-    AssetRoleAdminMixin,
-    UserDirectGrantedAssetsQuerysetMixin, AssetsSerializerFormatMixin, ListAPIView
+    AssetRoleAdminMixin, UserDirectGrantedAssetsQuerysetMixin,
+    AssetsSerializerFormatMixin, ListAPIView
 ):
     """ 直接授权给用户的资产 """
     pass
@@ -35,8 +34,8 @@ class MyDirectGrantedAssetsApi(AssetRoleUserMixin, UserDirectGrantedAssetsApi):
 
 
 class UserFavoriteGrantedAssetsApi(
-    AssetRoleAdminMixin,
-    UserFavoriteGrantedAssetsMixin, AssetsSerializerFormatMixin, ListAPIView
+    AssetRoleAdminMixin, UserFavoriteGrantedAssetsMixin,
+    AssetsSerializerFormatMixin, ListAPIView
 ):
     """ 用户收藏的授权资产 """
     pass
@@ -63,8 +62,8 @@ class MyUngroupAssetsAsTreeApi(AssetRoleUserMixin, UserDirectGrantedAssetsAsTree
 
 
 class UserAllGrantedAssetsApi(
-    AssetRoleAdminMixin,
-    UserAllGrantedAssetsQuerysetMixin, AssetsSerializerFormatMixin, ListAPIView
+    AssetRoleAdminMixin, UserAllGrantedAssetsQuerysetMixin,
+    AssetsSerializerFormatMixin, ListAPIView
 ):
     """ 授权给用户的所有资产 """
     pass
@@ -81,7 +80,8 @@ class MyAllAssetsAsTreeApi(AssetsTreeFormatMixin, MyAllGrantedAssetsApi):
 
 
 class UserGrantedNodeAssetsApi(
-    AssetRoleAdminMixin, UserGrantedNodeAssetsMixin, AssetsSerializerFormatMixin, ListAPIView
+    AssetRoleAdminMixin, UserGrantedNodeAssetsMixin,
+    AssetsSerializerFormatMixin, ListAPIView
 ):
     """ 授权给用户的节点资产 """
     pass

apps/perms/api/user_permission/assets/mixin.py

@@ -1,11 +1,11 @@
 from rest_framework.response import Response
 from rest_framework.request import Request
 
+from common.utils import get_logger
 from users.models import User
 from assets.api.mixin import SerializeToTreeNodeMixin
-from common.utils import get_logger
-from perms.pagination import NodeGrantedAssetPagination, AllGrantedAssetPagination
 from assets.models import Asset, Node
+from perms.pagination import NodeGrantedAssetPagination, AllGrantedAssetPagination
 from perms import serializers
 from perms.utils.user_permission import UserGrantedAssetsQueryUtils
 
@@ -21,8 +21,7 @@ class UserDirectGrantedAssetsQuerysetMixin:
     def get_queryset(self):
         if getattr(self, 'swagger_fake_view', False):
             return Asset.objects.none()
-        user = self.user
-        assets = UserGrantedAssetsQueryUtils(user) \
+        assets = UserGrantedAssetsQueryUtils(self.user) \
             .get_direct_granted_assets() \
             .prefetch_related('platform') \
             .only(*self.only_fields)
@@ -32,7 +31,7 @@ class UserDirectGrantedAssetsQuerysetMixin:
 class UserAllGrantedAssetsQuerysetMixin:
     only_fields = serializers.AssetGrantedSerializer.Meta.only_fields
     pagination_class = AllGrantedAssetPagination
-    ordering_fields = ("name", "address", "port", "cpu_cores")
+    ordering_fields = ("name", "address")
     ordering = ('name', )
 
     user: User