mirror of
https://github.com/jumpserver/jumpserver.git
synced 2025-09-03 16:35:10 +00:00
395b868dcf
* perf: swagger upgrade * perf: upgrade to drf-spectacular * perf: 添加部分注解 * perf: swagger done --------- Co-authored-by: ibuler <ibuler@qq.com>
114 lines
3.2 KiB
Python
114 lines
3.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
from rest_framework import serializers
|
|
|
|
from common.serializers import BulkSerializerMixin
|
|
from perms.models import AssetPermission
|
|
|
|
__all__ = [
|
|
'AssetPermissionUserRelationSerializer',
|
|
'AssetPermissionUserGroupRelationSerializer',
|
|
"AssetPermissionAssetRelationSerializer",
|
|
'AssetPermissionNodeRelationSerializer',
|
|
'AssetPermissionAllAssetSerializer',
|
|
'AssetPermissionAllUserSerializer',
|
|
]
|
|
|
|
|
|
class CurrentAssetPermission(object):
|
|
permission = None
|
|
|
|
def set_context(self, serializer_field):
|
|
self.permission = serializer_field.context['permission']
|
|
|
|
def __call__(self):
|
|
return self.permission
|
|
|
|
|
|
class RelationMixin(BulkSerializerMixin, serializers.Serializer):
|
|
assetpermission_display = serializers.SerializerMethodField()
|
|
|
|
def get_field_names(self, declared_fields, info):
|
|
fields = super().get_field_names(declared_fields, info)
|
|
fields.extend(['assetpermission', "assetpermission_display"])
|
|
return fields
|
|
|
|
def get_assetpermission_display(self, obj) -> str:
|
|
return str(obj.assetpermission)
|
|
|
|
|
|
class AssetPermissionUserRelationSerializer(RelationMixin, serializers.ModelSerializer):
|
|
user_display = serializers.SerializerMethodField()
|
|
|
|
class Meta:
|
|
model = AssetPermission.users.through
|
|
fields = [
|
|
'id', 'user', 'user_display',
|
|
]
|
|
|
|
def get_user_display(self, obj) -> str:
|
|
return str(obj.user)
|
|
|
|
|
|
class AssetPermissionAllUserSerializer(serializers.Serializer):
|
|
user = serializers.UUIDField(read_only=True, source='id')
|
|
user_display = serializers.SerializerMethodField()
|
|
|
|
class Meta:
|
|
only_fields = ['id', 'username', 'name']
|
|
|
|
@staticmethod
|
|
def get_user_display(obj) -> str:
|
|
return str(obj)
|
|
|
|
|
|
class AssetPermissionUserGroupRelationSerializer(RelationMixin, serializers.ModelSerializer):
|
|
usergroup_display = serializers.SerializerMethodField()
|
|
|
|
class Meta:
|
|
model = AssetPermission.user_groups.through
|
|
fields = [
|
|
'id', 'usergroup', "usergroup_display",
|
|
]
|
|
|
|
def get_usergroup_display(self, obj) -> str:
|
|
return str(obj.usergroup)
|
|
|
|
|
|
class AssetPermissionAssetRelationSerializer(RelationMixin, serializers.ModelSerializer):
|
|
asset_display = serializers.SerializerMethodField()
|
|
|
|
class Meta:
|
|
model = AssetPermission.assets.through
|
|
fields = [
|
|
'id', "asset", "asset_display",
|
|
]
|
|
|
|
def get_asset_display(self, obj) -> str:
|
|
return str(obj.asset)
|
|
|
|
|
|
class AssetPermissionAllAssetSerializer(serializers.Serializer):
|
|
asset = serializers.UUIDField(read_only=True, source='id')
|
|
asset_display = serializers.SerializerMethodField()
|
|
|
|
class Meta:
|
|
only_fields = ['id', 'name', 'address']
|
|
|
|
@staticmethod
|
|
def get_asset_display(obj) -> str:
|
|
return str(obj)
|
|
|
|
|
|
class AssetPermissionNodeRelationSerializer(RelationMixin, serializers.ModelSerializer):
|
|
node_display = serializers.CharField(source='node.full_value', read_only=True)
|
|
|
|
class Meta:
|
|
model = AssetPermission.nodes.through
|
|
fields = [
|
|
'id', 'node', "node_display",
|
|
]
|
|
|
|
def get_node_display(self, obj) -> str:
|
|
return str(obj.node)
|