mirror of
https://github.com/jumpserver/jumpserver.git
synced 2026-07-16 00:10:55 +00:00
caa5060ecd
* [Update] 控制组织管理员不允许编辑（更新、删除）超级用户 - 待续（控制批量更新API） * [Update] 修改方法名称 * [Update] 控制组织管理员不允许批量更新包含超级用户的用户列表 * [Bugfix] 修复所有ViewSet API进行批量更新时rest_framework_bulk库内部的bug * [Update] 修改 OpenID Middleware 日志输出模式 info => debug
74 lines
1.9 KiB
Python
74 lines
1.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
from django.core.cache import cache
|
|
from rest_framework import serializers
|
|
|
|
from common.serializers import AdaptedBulkListSerializer
|
|
|
|
from ..models import Node, AdminUser
|
|
from ..const import ADMIN_USER_CONN_CACHE_KEY
|
|
|
|
from .base import AuthSerializer
|
|
|
|
|
|
class AdminUserSerializer(serializers.ModelSerializer):
|
|
"""
|
|
管理用户
|
|
"""
|
|
assets_amount = serializers.SerializerMethodField()
|
|
unreachable_amount = serializers.SerializerMethodField()
|
|
reachable_amount = serializers.SerializerMethodField()
|
|
|
|
class Meta:
|
|
list_serializer_class = AdaptedBulkListSerializer
|
|
model = AdminUser
|
|
fields = '__all__'
|
|
|
|
def get_field_names(self, declared_fields, info):
|
|
fields = super().get_field_names(declared_fields, info)
|
|
return [f for f in fields if not f.startswith('_')]
|
|
|
|
@staticmethod
|
|
def get_unreachable_amount(obj):
|
|
data = cache.get(ADMIN_USER_CONN_CACHE_KEY.format(obj.name))
|
|
if data:
|
|
return len(data.get('dark'))
|
|
else:
|
|
return 0
|
|
|
|
@staticmethod
|
|
def get_reachable_amount(obj):
|
|
data = cache.get(ADMIN_USER_CONN_CACHE_KEY.format(obj.name))
|
|
if data:
|
|
return len(data.get('contacted'))
|
|
else:
|
|
return 0
|
|
|
|
@staticmethod
|
|
def get_assets_amount(obj):
|
|
return obj.assets_amount
|
|
|
|
|
|
class AdminUserAuthSerializer(AuthSerializer):
|
|
|
|
class Meta:
|
|
model = AdminUser
|
|
fields = ['password', 'private_key']
|
|
|
|
|
|
class ReplaceNodeAdminUserSerializer(serializers.ModelSerializer):
|
|
"""
|
|
管理用户更新关联到的集群
|
|
"""
|
|
nodes = serializers.PrimaryKeyRelatedField(
|
|
many=True, queryset=Node.objects.all()
|
|
)
|
|
|
|
class Meta:
|
|
model = AdminUser
|
|
fields = ['id', 'nodes']
|
|
|
|
|
|
class TaskIDSerializer(serializers.Serializer):
|
|
task = serializers.CharField(read_only=True)
|