moviewyrm/bookwyrm/models/shelf.py

93 lines
2.7 KiB
Python
Raw Normal View History

2021-03-08 16:49:10 +00:00
""" puttin' books on shelves """
2020-11-10 22:52:04 +00:00
import re
2020-03-07 06:56:44 +00:00
from django.db import models
from bookwyrm import activitypub
from .activitypub_mixin import CollectionItemMixin, OrderedCollectionMixin
from .base_model import BookWyrmModel
from . import fields
2020-03-07 06:56:44 +00:00
2020-09-21 15:16:34 +00:00
class Shelf(OrderedCollectionMixin, BookWyrmModel):
2021-03-08 16:49:10 +00:00
""" a list of books owned by a user """
TO_READ = "to-read"
READING = "reading"
READ_FINISHED = "read"
READ_STATUS_IDENTIFIERS = (TO_READ, READING, READ_FINISHED)
name = fields.CharField(max_length=100)
2020-03-07 06:56:44 +00:00
identifier = models.CharField(max_length=100)
user = fields.ForeignKey(
2021-03-08 16:49:10 +00:00
"User", on_delete=models.PROTECT, activitypub_field="owner"
)
2020-03-07 06:56:44 +00:00
editable = models.BooleanField(default=True)
2021-02-02 17:37:46 +00:00
privacy = fields.PrivacyField()
2020-03-07 06:56:44 +00:00
books = models.ManyToManyField(
2021-03-08 16:49:10 +00:00
"Edition",
2020-03-07 06:56:44 +00:00
symmetrical=False,
2021-03-08 16:49:10 +00:00
through="ShelfBook",
through_fields=("shelf", "book"),
2020-03-07 06:56:44 +00:00
)
activity_serializer = activitypub.Shelf
2020-11-10 22:52:04 +00:00
def save(self, *args, **kwargs):
2021-03-08 16:49:10 +00:00
""" set the identifier """
2021-02-07 05:00:08 +00:00
super().save(*args, **kwargs)
2020-11-10 22:52:04 +00:00
if not self.identifier:
2021-03-31 18:04:20 +00:00
self.identifier = self.get_identifier()
super().save(*args, **kwargs, broadcast=False)
def get_identifier(self):
""" custom-shelf-123 for the url """
slug = re.sub(r"[^\w]", "", self.name).lower()
return "{:s}-{:d}".format(slug, self.id)
2020-11-10 22:52:04 +00:00
@property
def collection_queryset(self):
2021-03-08 16:49:10 +00:00
""" list of books for this shelf, overrides OrderedCollectionMixin """
return self.books.all().order_by("shelfbook")
def get_remote_id(self):
2021-03-08 16:49:10 +00:00
""" shelf identifier instead of id """
base_path = self.user.remote_id
2021-03-31 18:04:20 +00:00
identifier = self.identifier or self.get_identifier()
return "%s/books/%s" % (base_path, identifier)
2020-03-07 06:56:44 +00:00
class Meta:
2021-03-08 16:49:10 +00:00
""" user/shelf unqiueness """
unique_together = ("user", "identifier")
2020-03-07 06:56:44 +00:00
class ShelfBook(CollectionItemMixin, BookWyrmModel):
2021-03-08 16:49:10 +00:00
""" many to many join table for books and shelves """
book = fields.ForeignKey(
2021-03-08 16:49:10 +00:00
"Edition", on_delete=models.PROTECT, activitypub_field="object"
)
shelf = fields.ForeignKey(
2021-03-08 16:49:10 +00:00
"Shelf", on_delete=models.PROTECT, activitypub_field="target"
)
user = fields.ForeignKey(
2021-03-08 16:49:10 +00:00
"User", on_delete=models.PROTECT, activitypub_field="actor"
)
2020-03-07 06:56:44 +00:00
2021-02-19 19:16:01 +00:00
activity_serializer = activitypub.Add
2021-03-08 16:49:10 +00:00
object_field = "book"
collection_field = "shelf"
2021-03-16 20:37:22 +00:00
def save(self, *args, **kwargs):
if not self.user:
self.user = self.shelf.user
super().save(*args, **kwargs)
2020-03-07 06:56:44 +00:00
class Meta:
2021-03-08 16:49:10 +00:00
"""an opinionated constraint!
you can't put a book on shelf twice"""
unique_together = ("book", "shelf")
ordering = ("-created_date",)