• 八千流's avatar
    Dev csv (#2640) · 22f362aa
    八千流 authored
    * [Update] 封装JMSCSVRender和JMSCSVParser
    
    * [Update] 更改JMSCSVRender,根据请求参数控制导出csv的字段和下载csv模板的字段
    
    * [Update] 导入空数据,提示错误消息
    
    * [Update] 修改用户导入和导出功能代码
    
    * [Update] 修改导入路由为动态反向解析
    
    * [Update] 修改JMSCSVRender和JMSCSVParser以及用户导入导出代码
    
    * [Update] 优化parsers逻辑
    
    * [Update] 优化parsers csv代码结构
    
    * [Update] 优化renders csv代码逻辑
    
    * [Update] 删除parsers csv多余代码
    
    * [Update] 删除parsers csv多余变量
    
    * [Update] 优化renders csv代码结构
    
    * [Update] 优化renders csv代码结构2
    
    * [Update] 优化renders csv获取header逻辑
    
    * [Update] 优化Cache Resources ID View逻辑
    
    * [Update] 优化ViewSet IDCacheFilterMixin逻辑
    
    * [Update] csv: parser render 添加异常捕获逻辑
    
    * [Update] 删除多余代码
    
    * [Update] 优化前端代码
    
    * [Update] 修改小问题
    
    * [Update] 修改前端导出用户的问题
    
    * [Update] 前端 - 优化数据导出逻辑 APIExportData
    
    * [Update] 修复批量创建用户时发送created信号的bug
    
    * [Update] 优化导入时错误信息展示
    
    * [Update] 优化parser、render时,对于多对多字段的处理
    
    * [Update] 修改前端上传空文件问题
    
    * [Update] 添加IDExportFilter,控制下载模版时的queryset
    
    * [Update] 修改判断导出模版时参数变量名 action => template
    
    * [Update] 修复导入用户数据时,用户组不生效的bug
    
    * [Update] 修改前端导入信息展示
    
    * [Update] 抽象资源导入模版
    
    * [Update] 优化资源导入模版
    
    * [Update] 修改js设置url的params逻辑
    
    * [Update] 修改users序列类控制read_only字段方式
    
    * [Update] 资产列表采用新的导入/导出csv文件逻辑
    
    * [Update] 修改导入资产时设置资产所在节点逻辑
    
    * [Update] 添加用户组导入/导出功能
    
    * [Update] 修改前端变量名
    
    * [Update] 修改下载导入模版,不包含org字段
    
    * [Update] 增加管理用户导入/导出功能
    
    * [Update] 导入模版提供id字段(为了资源备份后导入直接使用); 修复资源导入时联合唯一字段不校验导致创建时报错的bug
    
    * [Update] 增加系统用户导入/导出功能
    
    * [Update] 排序资源导入/导出字段
    
    * [Update] 翻译导入/导出的字段和模版
    
    * [Update] 更改csv导出和导出模版数据的控制在render实现
    
    * [Update] 资产添加 更新导入 功能
    
    * [Update] 用户/用户组/管理用户/系统用户/ 添加导入更新
    
    * [Update] 翻译
    
    * [Update] 优化资源序列化中的label
    
    * [Update] 去掉资源IDInFilterMixin过滤
    
    * [Update] 翻译
    22f362aa
admin_user.py 2.64 KB
# -*- coding: utf-8 -*-
#
from django.core.cache import cache
from django.utils.translation import ugettext_lazy as _
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):
    """
    管理用户
    """
    password = serializers.CharField(
        required=False, write_only=True, label=_('Password')
    )
    unreachable_amount = serializers.SerializerMethodField(label=_('Unreachable'))
    assets_amount = serializers.SerializerMethodField(label=_('Asset'))
    reachable_amount = serializers.SerializerMethodField(label=_('Reachable'))

    class Meta:
        list_serializer_class = AdaptedBulkListSerializer
        model = AdminUser
        fields = [
            'id', 'org_id', 'name', 'username', 'assets_amount',
            'reachable_amount', 'unreachable_amount', 'password', 'comment',
            'date_created', 'date_updated', 'become', 'become_method',
            'become_user', 'created_by',
        ]

        extra_kwargs = {
            'date_created': {'label': _('Date created')},
            'date_updated': {'label': _('Date updated')},
            'become': {'read_only': True}, 'become_method': {'read_only': True},
            'become_user': {'read_only': True}, 'created_by': {'read_only': True}
        }

    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)