zedfrigg/bookwyrm
1
0
Fork 0
Code Activity

Update ordered collection classes to use CollectionItem

Browse source
This commit is contained in:
Mouse Reeve 2021-04-08 14:17:29 -07:00
parent 98161b9041
commit 31077a9790
3 changed files with 13 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

5
bookwyrm/models/shelf.py
View file

@ -32,6 +32,7 @@ class Shelf(OrderedCollectionMixin, BookWyrmModel):
)
activity_serializer = activitypub.Shelf
collection_field = "shelf"
def save(self, *args, **kwargs):
""" set the identifier """
@ -75,9 +76,7 @@ class ShelfBook(CollectionItemMixin, BookWyrmModel):
"User", on_delete=models.PROTECT, activitypub_field="actor"
)
activity_serializer = activitypub.Add
object_field = "book"
collection_field = "shelf"
activity_serializer = activitypub.ShelfItem
def save(self, *args, **kwargs):
if not self.user: