jumpserver/apps/terminal/serializers/storage.py

# -*- coding: utf-8 -*-
#
import copy
from rest_framework import serializers

from common.fields.serializer import CustomMetaDictField
from ..models import ReplayStorage, CommandStorage
from .. import const


class ReplayStorageMetaDictField(CustomMetaDictField):
    type_map_fields = const.REPLAY_STORAGE_TYPE_MAP_FIELDS
    default_type = const.REPLAY_STORAGE_TYPE_SERVER
    need_convert_key = True


class BaseStorageSerializerMixin:
    type_map_fields = None

    def process_meta(self, instance, validated_data):
        new_meta = copy.deepcopy(validated_data.get('meta', {}))
        tp = validated_data.get('type', '')

        if tp != instance.type:
            return new_meta

        old_meta = instance.meta
        fields = self.type_map_fields.get(instance.type, [])
        for field in fields:
            if not field.get('write_only', False):
                continue
            field_name = field['name']
            new_value = new_meta.get(field_name, '')
            old_value = old_meta.get(field_name, '')
            field_value = new_value if new_value else old_value
            new_meta[field_name] = field_value

        return new_meta

    def update(self, instance, validated_data):
        meta = self.process_meta(instance, validated_data)
        validated_data['meta'] = meta
        return super().update(instance, validated_data)


class ReplayStorageSerializer(BaseStorageSerializerMixin,
                              serializers.ModelSerializer):

    meta = ReplayStorageMetaDictField()

    type_map_fields = const.REPLAY_STORAGE_TYPE_MAP_FIELDS

    class Meta:
        model = ReplayStorage
        fields = ['id', 'name', 'type', 'meta', 'comment']


class CommandStorageMetaDictField(CustomMetaDictField):
    type_map_fields = const.COMMAND_STORAGE_TYPE_MAP_FIELDS
    default_type = const.COMMAND_STORAGE_TYPE_SERVER
    need_convert_key = True


class CommandStorageSerializer(BaseStorageSerializerMixin,
                               serializers.ModelSerializer):

    meta = CommandStorageMetaDictField()

    type_map_fields = const.COMMAND_STORAGE_TYPE_MAP_FIELDS

    class Meta:
        model = CommandStorage
        fields = ['id', 'name', 'type', 'meta', 'comment']