X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/55efada7f210cda246203bdef15bdd781ad26cfe..e70d66e668627a4d9c65900856ca7024d8d87fe1:/src/bookmarks/api/views.py

diff --git a/src/bookmarks/api/views.py b/src/bookmarks/api/views.py
index 82502e9e2..315624fab 100644
--- a/src/bookmarks/api/views.py
+++ b/src/bookmarks/api/views.py
@@ -48,6 +48,17 @@ class BookmarkSerializer(serializers.ModelSerializer):
         fields = ['book_slug', 'book', 'anchor', 'audio_timestamp', 'mode', 'note', 'href', 'uuid', 'location', 'timestamp', 'deleted']
         read_only_fields = ['uuid', 'mode']
 
+    def create(self, validated_data):
+        book = validated_data.pop('book_slug', None)
+        if book is not None:
+            validated_data['book'] = book
+        return super().create(validated_data)
+
+    def update(self, instance, validated_data):
+        book = validated_data.pop('book_slug', None)
+        if book is not None:
+            validated_data['book'] = book
+        return super().update(instance, validated_data)
 
 @never_cache
 class BookmarksView(ListCreateAPIView):