Fix creation of collection membership when creating collections.

This commit is contained in:
Tom Hacohen
2020-02-26 21:11:29 +02:00
parent 7a38e26872
commit 771d2d013d
2 changed files with 19 additions and 9 deletions

View File

@@ -56,6 +56,22 @@ class CollectionSerializer(serializers.ModelSerializer):
def get_ctag(self, obj):
return 'FIXME'
def create(self, validated_data):
"""Function that's called when this serializer creates an item"""
encryption_key = validated_data.pop('encryptionKey')
instance = self.__class__.Meta.model(**validated_data)
print(validated_data)
with transaction.atomic():
instance.save()
models.CollectionMember(collection=instance,
user=validated_data.get('owner'),
accessLevel=models.CollectionMember.AccessLevels.ADMIN,
encryptionKey=encryption_key,
).save()
return instance
class CollectionItemChunkSerializer(serializers.ModelSerializer):
class Meta: