Make /aoi/addresses/ endpoint writable

This commit is contained in:
Santiago L 2021-06-18 11:11:50 +02:00
parent e1d71fa620
commit 867d9afe65
1 changed files with 14 additions and 8 deletions

View File

@ -1,3 +1,4 @@
from django.db import transaction
from rest_framework import serializers from rest_framework import serializers
from orchestra.api.serializers import SetPasswordHyperlinkedSerializer, RelatedHyperlinkedModelSerializer from orchestra.api.serializers import SetPasswordHyperlinkedSerializer, RelatedHyperlinkedModelSerializer
@ -43,7 +44,7 @@ class RelatedMailboxSerializer(AccountSerializerMixin, RelatedHyperlinkedModelSe
class AddressSerializer(AccountSerializerMixin, serializers.HyperlinkedModelSerializer): class AddressSerializer(AccountSerializerMixin, serializers.HyperlinkedModelSerializer):
domain = RelatedDomainSerializer() domain = RelatedDomainSerializer()
mailboxes = RelatedMailboxSerializer(many=True, required=False) #allow_add_remove=True mailboxes = RelatedMailboxSerializer(many=True, required=False)
class Meta: class Meta:
model = Address model = Address
@ -51,16 +52,21 @@ class AddressSerializer(AccountSerializerMixin, serializers.HyperlinkedModelSeri
def validate(self, attrs): def validate(self, attrs):
attrs = super(AddressSerializer, self).validate(attrs) attrs = super(AddressSerializer, self).validate(attrs)
if not attrs['mailboxes'] and not attrs['forward']: mailboxes = attrs.get('mailboxes', [])
forward = attrs.get('forward', '')
if not mailboxes and not forward:
raise serializers.ValidationError("A mailbox or forward address should be provided.") raise serializers.ValidationError("A mailbox or forward address should be provided.")
return attrs return attrs
@transaction.atomic
def create(self, validated_data): def create(self, validated_data):
mailboxes = validated_data.pop('mailboxes') mailboxes = validated_data.pop('mailboxes', [])
obj = super().create(validated_data)
# assign address to same account than domain
account = validated_data['domain'].account
obj = self.Meta.model.objects.create(account=account, **validated_data)
obj.mailboxes.set(mailboxes) obj.mailboxes.set(mailboxes)
return obj return obj
@transaction.atomic
def update(self, instance, validated_data):
mailboxes = validated_data.pop('mailboxes', [])
instance.mailboxes.set(mailboxes)
return super().update(instance, validated_data)