78 lines
2.8 KiB
Python
78 lines
2.8 KiB
Python
from django.db import models
|
|
from common.models import UserOwnedEntity
|
|
from movies.models import Movie
|
|
from books.models import Book
|
|
from music.models import Song, Album
|
|
from games.models import Game
|
|
from markdownx.models import MarkdownxField
|
|
from django.utils.translation import gettext_lazy as _
|
|
from django.conf import settings
|
|
from common.utils import ChoicesDictGenerator, GenerateDateUUIDMediaFilePath
|
|
|
|
|
|
def collection_cover_path(instance, filename):
|
|
return GenerateDateUUIDMediaFilePath(instance, filename, settings.COLLECTION_MEDIA_PATH_ROOT)
|
|
|
|
|
|
class Collection(UserOwnedEntity):
|
|
title = models.CharField(max_length=200)
|
|
description = MarkdownxField()
|
|
cover = models.ImageField(_("封面"), upload_to=collection_cover_path, default=settings.DEFAULT_COLLECTION_IMAGE, blank=True)
|
|
|
|
def __str__(self):
|
|
return f"Collection({self.id} {self.owner} {self.title})"
|
|
|
|
@property
|
|
def collectionitem_list(self):
|
|
return sorted(list(self.collectionitem_set.all()), key=lambda i: i.position)
|
|
|
|
@property
|
|
def item_list(self):
|
|
return map(lambda i: i.item, self.collectionitem_list)
|
|
|
|
@property
|
|
def plain_description(self):
|
|
html = markdown(self.description)
|
|
return RE_HTML_TAG.sub(' ', html)
|
|
|
|
def has_item(self, item):
|
|
return len(list(filter(lambda i: i.item == item, self.collectionitem_list))) > 0
|
|
|
|
def append_item(self, item, comment=""):
|
|
cl = self.collectionitem_list
|
|
if item is None or self.has_item(item):
|
|
return None
|
|
else:
|
|
i = CollectionItem(collection=self, position=cl[-1].position + 1 if len(cl) else 1, comment=comment)
|
|
i.set_item(item)
|
|
i.save()
|
|
return i
|
|
|
|
|
|
class CollectionItem(models.Model):
|
|
movie = models.ForeignKey(Movie, on_delete=models.CASCADE, null=True)
|
|
album = models.ForeignKey(Album, on_delete=models.CASCADE, null=True)
|
|
song = models.ForeignKey(Song, on_delete=models.CASCADE, null=True)
|
|
book = models.ForeignKey(Book, on_delete=models.CASCADE, null=True)
|
|
game = models.ForeignKey(Game, on_delete=models.CASCADE, null=True)
|
|
collection = models.ForeignKey(Collection, on_delete=models.CASCADE)
|
|
position = models.PositiveIntegerField()
|
|
comment = models.TextField(_("备注"), default='')
|
|
|
|
@property
|
|
def item(self):
|
|
items = list(filter(lambda i: i is not None, [self.movie, self.book, self.album, self.song, self.game]))
|
|
return items[0] if len(items) > 0 else None
|
|
|
|
# @item.setter
|
|
def set_item(self, new_item):
|
|
old_item = self.item
|
|
if old_item == new_item:
|
|
return
|
|
if old_item is not None:
|
|
self.movie = None
|
|
self.book = None
|
|
self.album = None
|
|
self.song = None
|
|
self.game = None
|
|
setattr(self, new_item.__class__.__name__.lower(), new_item)
|