django-orchestra-test/orchestra/contrib/webapps/serializers.py

70 lines
2.4 KiB
Python

from django.db import models
from rest_framework import serializers
from orchestra.api.serializers import HyperlinkedModelSerializer
from orchestra.contrib.accounts.serializers import AccountSerializerMixin
from .models import WebApp, WebAppOption
class OptionSerializer(serializers.ModelSerializer):
class Meta:
model = WebAppOption
fields = ('name', 'value')
def to_representation(self, instance):
return {prop.name: prop.value for prop in instance.all()}
def to_internal_value(self, data):
return data
class DataField(serializers.Field):
def to_representation(self, data):
return data
class WebAppSerializer(AccountSerializerMixin, HyperlinkedModelSerializer):
options = OptionSerializer(required=False)
data = DataField()
class Meta:
model = WebApp
fields = ('url', 'id', 'name', 'type', 'options', 'data')
postonly_fields = ('name', 'type')
def __init__(self, *args, **kwargs):
super(WebAppSerializer, self).__init__(*args, **kwargs)
if isinstance(self.instance, models.Model):
type_serializer = self.instance.type_instance.serializer
if type_serializer:
self.fields['data'] = type_serializer()
def create(self, validated_data):
options_data = validated_data.pop('options')
webapp = super(WebAppSerializer, self).create(validated_data)
for key, value in options_data.items():
WebAppOption.objects.create(webapp=webapp, name=key, value=value)
return webap
def update(self, instance, validated_data):
options_data = validated_data.pop('options')
instance = super(WebAppSerializer, self).update(instance, validated_data)
existing = {}
for obj in instance.options.all():
existing[obj.name] = obj
posted = set()
for key, value in options_data.items():
posted.add(key)
try:
option = existing[key]
except KeyError:
option = instance.options.create(name=key, value=value)
else:
if option.value != value:
option.value = value
option.save(update_fields=('value',))
for to_delete in set(existing.keys())-posted:
existing[to_delete].delete()
return instance