jumpserver/apps/applications/serializers/application.py

# coding: utf-8
#

from orgs.mixins.serializers import BulkOrgResourceModelSerializer

from .. import models

__all__ = [
    'ApplicationSerializer',
]


class ApplicationSerializer(BulkOrgResourceModelSerializer):

    class Meta:
        model = models.Application
        fields = [
            'id', 'name', 'category', 'type', 'get_type_display', 'attrs',
            'domain', 'created_by', 'date_created', 'date_updated', 'comment'
        ]
        read_only_fields = [
            'created_by', 'date_created', 'date_updated', 'get_type_display',
        ]

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        app_type = ''
        attrs_data = {}
        request = self.context.get('request')
        if request:
            app_type = request.query_params.get('type')
        if hasattr(self, 'initial_data'):
            app_type = self.initial_data.get('type')
            attrs_data = self.initial_data.get('attrs')
        if self.instance:
            app_type = self.instance.type
            attrs_data = self.instance.attrs
        if not app_type:
            return
        attrs_cls = models.Category.get_type_serializer_cls(app_type)
        if attrs_data:
            attrs_serializer = attrs_cls(data=attrs_data)
        else:
            attrs_serializer = attrs_cls()
        self.fields['attrs'] = attrs_serializer

    def create(self, validated_data):
        attrs = validated_data.pop('attrs', {})
        instance = super().create(validated_data)
        instance.attrs = attrs
        instance.save()
        return instance

    def update(self, instance, validated_data):
        attrs = validated_data.pop('attrs', {})
        instance = super().update(instance, validated_data)
        instance.attrs = attrs
        instance.save()
        return instance