pangea
/
django-orchestra
7
0
Fork 0
Code Issues 1 Pull Requests 2 Packages Projects Releases Wiki Activity

Filter related addresses by account

This commit is contained in:
Santiago L 2021-10-14 13:03:08 +02:00
parent e88e27a56e
commit 03666d8ed0
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
orchestra/contrib/mailboxes/serializers.py
View File

@ -36,8 +36,15 @@ class MailboxSerializer(AccountSerializerMixin, SetPasswordHyperlinkedSerializer
postonly_fields = ('name', 'password') postonly_fields = ('name', 'password')
class AddressRelatedField(serializers.HyperlinkedRelatedField):
# Filter addresses by account (user)
def get_queryset(self):
qs = super().get_queryset()
return qs.filter(account=self.context['account'])
class MailboxWritableSerializer(AccountSerializerMixin, SetPasswordHyperlinkedSerializer): class MailboxWritableSerializer(AccountSerializerMixin, SetPasswordHyperlinkedSerializer):
addresses = serializers.HyperlinkedRelatedField(many=True, view_name='address-detail', queryset=Address.objects.all()) addresses = AddressRelatedField(many=True, view_name='address-detail', queryset=Address.objects.all())
class Meta: class Meta:
model = Mailbox model = Mailbox
@ -46,6 +53,10 @@ class MailboxWritableSerializer(AccountSerializerMixin, SetPasswordHyperlinkedSe
) )
postonly_fields = ('name', 'password') postonly_fields = ('name', 'password')
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.fields['addresses'].context['account'] = self.account
@transaction.atomic @transaction.atomic
def create(self, validated_data): def create(self, validated_data):
addresses = validated_data.pop('addresses', []) addresses = validated_data.pop('addresses', [])