mirror of
https://github.com/mediacms-io/mediacms.git
synced 2025-08-17 08:21:05 +02:00
98 lines
2.9 KiB
Python
98 lines
2.9 KiB
Python
import uuid
|
|
|
|
from django.db import models
|
|
from django.urls import reverse
|
|
from django.utils.html import strip_tags
|
|
|
|
from .. import helpers
|
|
|
|
|
|
class Playlist(models.Model):
|
|
"""Playlists model"""
|
|
|
|
add_date = models.DateTimeField(auto_now_add=True, db_index=True)
|
|
|
|
description = models.TextField(blank=True, help_text="description")
|
|
|
|
friendly_token = models.CharField(blank=True, max_length=12, db_index=True)
|
|
|
|
media = models.ManyToManyField("Media", through="playlistmedia", blank=True)
|
|
|
|
title = models.CharField(max_length=100, db_index=True)
|
|
|
|
uid = models.UUIDField(unique=True, default=uuid.uuid4)
|
|
|
|
user = models.ForeignKey("users.User", on_delete=models.CASCADE, db_index=True, related_name="playlists")
|
|
|
|
def __str__(self):
|
|
return self.title
|
|
|
|
@property
|
|
def media_count(self):
|
|
return self.media.filter(listable=True).count()
|
|
|
|
def get_absolute_url(self, api=False):
|
|
if api:
|
|
return reverse("api_get_playlist", kwargs={"friendly_token": self.friendly_token})
|
|
else:
|
|
return reverse("get_playlist", kwargs={"friendly_token": self.friendly_token})
|
|
|
|
@property
|
|
def url(self):
|
|
return self.get_absolute_url()
|
|
|
|
@property
|
|
def api_url(self):
|
|
return self.get_absolute_url(api=True)
|
|
|
|
def user_thumbnail_url(self):
|
|
if self.user.logo:
|
|
return helpers.url_from_path(self.user.logo.path)
|
|
return None
|
|
|
|
def set_ordering(self, media, ordering):
|
|
if media not in self.media.all():
|
|
return False
|
|
pm = PlaylistMedia.objects.filter(playlist=self, media=media).first()
|
|
if pm and isinstance(ordering, int) and 0 < ordering:
|
|
pm.ordering = ordering
|
|
pm.save()
|
|
return True
|
|
return False
|
|
|
|
def save(self, *args, **kwargs):
|
|
strip_text_items = ["title", "description"]
|
|
for item in strip_text_items:
|
|
setattr(self, item, strip_tags(getattr(self, item, None)))
|
|
self.title = self.title[:99]
|
|
|
|
if not self.friendly_token:
|
|
while True:
|
|
friendly_token = helpers.produce_friendly_token()
|
|
if not Playlist.objects.filter(friendly_token=friendly_token):
|
|
self.friendly_token = friendly_token
|
|
break
|
|
super(Playlist, self).save(*args, **kwargs)
|
|
|
|
@property
|
|
def thumbnail_url(self):
|
|
pm = self.playlistmedia_set.filter(media__listable=True).first()
|
|
if pm and pm.media.thumbnail:
|
|
return helpers.url_from_path(pm.media.thumbnail.path)
|
|
return None
|
|
|
|
|
|
class PlaylistMedia(models.Model):
|
|
"""Helper model to store playlist specific media"""
|
|
|
|
action_date = models.DateTimeField(auto_now=True)
|
|
|
|
media = models.ForeignKey("Media", on_delete=models.CASCADE)
|
|
|
|
playlist = models.ForeignKey(Playlist, on_delete=models.CASCADE)
|
|
|
|
ordering = models.IntegerField(default=1)
|
|
|
|
class Meta:
|
|
ordering = ["ordering", "-action_date"]
|