You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
385 lines
16 KiB
385 lines
16 KiB
from django.db import models
|
|
from django.utils.translation import gettext_lazy as _
|
|
from django.conf import settings
|
|
import datetime
|
|
from filer.fields.image import FilerImageField
|
|
from utils import generate_slug_for_model
|
|
from utils.mixins import LowercaseSlugMixin
|
|
from django.core.validators import FileExtensionValidator
|
|
from django.core.exceptions import ValidationError
|
|
|
|
|
|
class VideoCategory(LowercaseSlugMixin, models.Model):
|
|
title = models.CharField(max_length=255, verbose_name=_('title'))
|
|
slug = models.SlugField(allow_unicode=True, unique=True, verbose_name=_('slug'))
|
|
|
|
status = models.BooleanField(default=True, verbose_name=_('status'))
|
|
order = models.PositiveIntegerField(default=0, verbose_name=_('order'))
|
|
created_at = models.DateTimeField(auto_now_add=True, verbose_name=_('created at'))
|
|
updated_at = models.DateTimeField(auto_now=True, verbose_name=_('updated at'))
|
|
|
|
def __str__(self):
|
|
return self.title
|
|
|
|
class Meta:
|
|
verbose_name = _('Video Category')
|
|
verbose_name_plural = _('Video Categories')
|
|
ordering = ['order']
|
|
|
|
|
|
class VideoCollection(LowercaseSlugMixin, models.Model):
|
|
class DisplayPosition(models.TextChoices):
|
|
PINNED = 'pinned', _('Pinned')
|
|
MIDDLE = 'middle', _('Middle Section')
|
|
|
|
title = models.CharField(max_length=255)
|
|
slug = models.SlugField(max_length=255, unique=True, allow_unicode=True)
|
|
summary = models.CharField(max_length=512, null=True, blank=True, help_text=_('could be null'))
|
|
thumbnail = models.ImageField(upload_to='video/collection/', null=True, blank=True, help_text=_('image allowed'))
|
|
order = models.IntegerField(default=0, verbose_name=_('order'))
|
|
|
|
status = models.BooleanField(default=True, verbose_name=_('status'))
|
|
display_position = models.CharField(
|
|
max_length=20,
|
|
choices=DisplayPosition.choices,
|
|
default=DisplayPosition.PINNED,
|
|
verbose_name=_('Display Position')
|
|
)
|
|
created_at = models.DateTimeField(auto_now_add=True, verbose_name=_('created at'))
|
|
updated_at = models.DateTimeField(auto_now=True, verbose_name=_('updated at'))
|
|
|
|
def __str__(self):
|
|
return f'Collection #{self.id}/{self.title}'
|
|
|
|
class Meta:
|
|
verbose_name = _('Video Collection')
|
|
verbose_name_plural = _('Video Collections')
|
|
|
|
class PinnedVideoCollection(VideoCollection):
|
|
class Meta:
|
|
proxy = True
|
|
verbose_name = _('Pinned Video Collection')
|
|
verbose_name_plural = _('Pinned Video Collections')
|
|
|
|
|
|
class MiddleVideoCollection(VideoCollection):
|
|
class Meta:
|
|
proxy = True
|
|
verbose_name = _('Middle Section Video Collection')
|
|
verbose_name_plural = _('Middle Section Video Collections')
|
|
|
|
|
|
|
|
|
|
class Video(LowercaseSlugMixin, models.Model):
|
|
slug_source_field = 'title'
|
|
class VedioTypeChoices(models.TextChoices):
|
|
YOUTUBE_LINK = 'youtube_link', 'Youtube Link'
|
|
VIDEO_FILE = 'video_file', 'Video File'
|
|
|
|
title = models.CharField(max_length=255, null=True)
|
|
slug = models.SlugField(max_length=255, allow_unicode=True, unique=True, blank=True, null=True)
|
|
thumbnail = models.ImageField(upload_to='video/thumbnails/', null=True, blank=True, help_text=_('image allowed'))
|
|
description = models.TextField(null=True)
|
|
|
|
video_type = models.CharField(max_length=255, choices=VedioTypeChoices.choices)
|
|
video_file = models.FileField(upload_to='video/videos/', null=True, blank=True ,validators=[
|
|
FileExtensionValidator(
|
|
allowed_extensions=['mp4', 'mov', 'avi', 'mkv', 'webm']
|
|
)
|
|
])
|
|
video_url = models.CharField(max_length=655, null=True, blank=True)
|
|
video_time = models.TimeField()
|
|
|
|
view_count = models.PositiveBigIntegerField(default=0, verbose_name=_('view count'))
|
|
|
|
status = models.BooleanField(default=True, verbose_name=_('status'))
|
|
|
|
created_at = models.DateTimeField(auto_now_add=True, verbose_name=_('created at'))
|
|
updated_at = models.DateTimeField(auto_now=True, verbose_name=_('updated at'))
|
|
|
|
def __str__(self):
|
|
return self.title
|
|
|
|
@property
|
|
def share_link(self):
|
|
if self.slug:
|
|
return f"{settings.DOVODI_DOMAIN}/videos/{self.slug}"
|
|
return None
|
|
|
|
@property
|
|
def stream_url(self):
|
|
"""
|
|
Get direct playable stream URL for video.
|
|
For YouTube link types, extracts direct media URL via yt-dlp (cached).
|
|
For Video file types, returns video file URL.
|
|
"""
|
|
if self.video_type == self.VedioTypeChoices.YOUTUBE_LINK:
|
|
if self.video_url:
|
|
from utils.youtube import get_youtube_stream_url
|
|
extracted = get_youtube_stream_url(self.video_url)
|
|
return extracted or self.video_url
|
|
return None
|
|
elif self.video_type == self.VedioTypeChoices.VIDEO_FILE:
|
|
if self.video_file:
|
|
return self.video_file.url
|
|
return None
|
|
return self.video_url
|
|
|
|
|
|
def increment_view_count(self):
|
|
"""Increment the view count for this video"""
|
|
self.view_count += 1
|
|
self.save(update_fields=['view_count'])
|
|
return self.view_count
|
|
|
|
def clean(self):
|
|
super().clean()
|
|
if self.video_type == self.VedioTypeChoices.YOUTUBE_LINK and not self.video_url:
|
|
raise ValidationError({
|
|
'video_url': _('This field is required when video type is Link.')
|
|
})
|
|
if self.video_type == self.VedioTypeChoices.VIDEO_FILE and not self.video_file:
|
|
raise ValidationError({
|
|
'video_file': _('This field is required when video type is File.')
|
|
})
|
|
|
|
def save(self, *args, **kwargs):
|
|
is_new = self.pk is None
|
|
file_changed = False
|
|
url_changed = False
|
|
|
|
if not is_new:
|
|
try:
|
|
old_instance = Video.objects.get(pk=self.pk)
|
|
if old_instance.video_file != self.video_file:
|
|
file_changed = True
|
|
if old_instance.video_url != self.video_url:
|
|
url_changed = True
|
|
except Video.DoesNotExist:
|
|
pass
|
|
else:
|
|
file_changed = True
|
|
url_changed = True
|
|
|
|
# Handle YouTube link automatic metadata extraction (duration & thumbnail)
|
|
if self.video_type == self.VedioTypeChoices.YOUTUBE_LINK and self.video_url:
|
|
needs_thumbnail = not bool(self.thumbnail)
|
|
needs_time = not self.video_time or self.video_time == datetime.time(0, 0, 0)
|
|
|
|
if needs_thumbnail or needs_time or url_changed:
|
|
try:
|
|
from utils.youtube import get_youtube_video_metadata, download_youtube_thumbnail
|
|
from django.core.files.base import ContentFile
|
|
|
|
meta = get_youtube_video_metadata(self.video_url)
|
|
if meta:
|
|
# Auto set duration
|
|
if (needs_time or url_changed) and meta.get('duration_time'):
|
|
self.video_time = meta['duration_time']
|
|
|
|
# Auto set thumbnail if missing
|
|
if (needs_thumbnail or (url_changed and not self.thumbnail)) and meta.get('thumbnail_url'):
|
|
thumb_bytes = download_youtube_thumbnail(meta['thumbnail_url'])
|
|
if thumb_bytes:
|
|
vid_id = meta.get('video_id') or 'yt_thumb'
|
|
self.thumbnail.save(f"yt_{vid_id}.jpg", ContentFile(thumb_bytes), save=False)
|
|
except Exception as e:
|
|
import logging
|
|
logging.getLogger(__name__).error(f"Failed to auto-populate YouTube video info: {e}")
|
|
|
|
if not self.video_time:
|
|
self.video_time = datetime.time(0, 0, 0)
|
|
|
|
super().save(*args, **kwargs)
|
|
|
|
if self.video_type == self.VedioTypeChoices.VIDEO_FILE and self.video_file:
|
|
from utils.media import get_media_duration_time, extract_video_frame_bytes
|
|
from django.core.files.base import ContentFile
|
|
import os
|
|
|
|
fields_to_update = []
|
|
|
|
if (file_changed or self.video_time == datetime.time(0, 0, 0)):
|
|
try:
|
|
duration_time = get_media_duration_time(self.video_file.path)
|
|
if duration_time:
|
|
self.video_time = duration_time
|
|
fields_to_update.append('video_time')
|
|
except Exception as e:
|
|
import logging
|
|
logging.getLogger(__name__).error(f"Failed to calculate video duration: {e}")
|
|
|
|
if not self.thumbnail:
|
|
try:
|
|
thumb_bytes = extract_video_frame_bytes(self.video_file.path, timestamp_sec=1.0)
|
|
if thumb_bytes:
|
|
base_name = os.path.splitext(os.path.basename(self.video_file.name))[0]
|
|
filename = f"thumb_{base_name}_{self.pk}.jpg"
|
|
self.thumbnail.save(filename, ContentFile(thumb_bytes), save=False)
|
|
fields_to_update.append('thumbnail')
|
|
except Exception as e:
|
|
import logging
|
|
logging.getLogger(__name__).error(f"Failed to extract video thumbnail: {e}")
|
|
|
|
if fields_to_update:
|
|
update_dict = {field: getattr(self, field) for field in fields_to_update}
|
|
Video.objects.filter(pk=self.pk).update(**update_dict)
|
|
|
|
if 'video_time' in fields_to_update:
|
|
try:
|
|
# Update playlists that contain this video
|
|
for appearance in self.playlist_appearances.all():
|
|
playlist = appearance.playlist
|
|
playlist.total_time = playlist.calculate_total_time()
|
|
playlist.save(update_fields=['total_time'])
|
|
except Exception as e:
|
|
pass
|
|
|
|
if self.video_type == self.VedioTypeChoices.YOUTUBE_LINK:
|
|
try:
|
|
for appearance in self.playlist_appearances.all():
|
|
playlist = appearance.playlist
|
|
playlist.total_time = playlist.calculate_total_time()
|
|
playlist.save(update_fields=['total_time'])
|
|
except Exception:
|
|
pass
|
|
|
|
|
|
|
|
class VideoPlaylist(LowercaseSlugMixin, models.Model):
|
|
title = models.CharField(max_length=255, verbose_name=_('title'))
|
|
slug = models.SlugField(allow_unicode=True, unique=True, null=True, blank=True, verbose_name=_('slug'))
|
|
slogan = models.CharField(max_length=512, null=True, blank=True, verbose_name=_('slogan'))
|
|
description = models.TextField(null=True, blank=True, verbose_name=_('description'))
|
|
thumbnail = models.ImageField(upload_to='video/playlist/thumbnails/', null=True, blank=True, verbose_name=_('thumbnail'))
|
|
|
|
categories = models.ManyToManyField(
|
|
VideoCategory,
|
|
related_name='playlists',
|
|
verbose_name=_('categories'),
|
|
blank=True,
|
|
)
|
|
collections = models.ManyToManyField(
|
|
VideoCollection,
|
|
through='VideoPlaylistInCollection',
|
|
related_name='related_playlists',
|
|
verbose_name=_('collections'),
|
|
blank=True
|
|
)
|
|
|
|
order = models.PositiveIntegerField(default=0, verbose_name=_('order'))
|
|
status = models.BooleanField(default=True, verbose_name=_('status'))
|
|
view_count = models.PositiveBigIntegerField(default=0, verbose_name=_('view count'))
|
|
total_time = models.DurationField(null=True, blank=True, verbose_name=_('total time'))
|
|
|
|
created_at = models.DateTimeField(auto_now_add=True, verbose_name=_('created at'))
|
|
updated_at = models.DateTimeField(auto_now=True, verbose_name=_('updated at'))
|
|
|
|
def __str__(self):
|
|
return self.title
|
|
|
|
def increment_view_count(self):
|
|
"""Increment the view count for this playlist"""
|
|
self.view_count += 1
|
|
self.save(update_fields=['view_count'])
|
|
return self.view_count
|
|
|
|
def calculate_total_time(self):
|
|
"""Calculate total duration of all videos in this playlist"""
|
|
from datetime import timedelta
|
|
total_seconds = 0
|
|
|
|
for item in self.playlist_items.select_related('video'):
|
|
video_time = item.video.video_time
|
|
total_seconds += video_time.hour * 3600 + video_time.minute * 60 + video_time.second
|
|
|
|
return timedelta(seconds=total_seconds)
|
|
|
|
@property
|
|
def effective_thumbnail(self):
|
|
"""
|
|
Get the thumbnail for this playlist.
|
|
If playlist has no explicit thumbnail, fall back to the first video with a thumbnail.
|
|
"""
|
|
if self.thumbnail:
|
|
return self.thumbnail
|
|
|
|
if hasattr(self, '_prefetched_objects_cache') and 'playlist_items' in self._prefetched_objects_cache:
|
|
items = sorted(
|
|
[item for item in self.playlist_items.all() if getattr(item, 'video', None) and item.video.status and item.video.thumbnail],
|
|
key=lambda x: getattr(x, 'priority', 0)
|
|
)
|
|
if items and items[0].video and items[0].video.thumbnail:
|
|
return items[0].video.thumbnail
|
|
return None
|
|
|
|
first_item = self.playlist_items.filter(
|
|
video__thumbnail__isnull=False,
|
|
video__status=True
|
|
).exclude(video__thumbnail='').select_related('video').order_by('priority').first()
|
|
if first_item and first_item.video and first_item.video.thumbnail:
|
|
return first_item.video.thumbnail
|
|
return None
|
|
|
|
class Meta:
|
|
verbose_name = _('Video Playlist')
|
|
verbose_name_plural = _('Video Playlists')
|
|
ordering = ['order', '-created_at']
|
|
|
|
|
|
class VideoPlaylistInCollection(models.Model):
|
|
collection = models.ForeignKey(
|
|
VideoCollection,
|
|
on_delete=models.CASCADE,
|
|
related_name='collection_playlists',
|
|
verbose_name=_('collection')
|
|
)
|
|
playlist = models.ForeignKey(
|
|
VideoPlaylist,
|
|
on_delete=models.CASCADE,
|
|
related_name='playlist_collections',
|
|
verbose_name=_('playlist')
|
|
)
|
|
order = models.PositiveIntegerField(default=0, verbose_name=_('order'))
|
|
|
|
created_at = models.DateTimeField(auto_now_add=True, verbose_name=_('created at'))
|
|
updated_at = models.DateTimeField(auto_now=True, verbose_name=_('updated at'))
|
|
|
|
def __str__(self):
|
|
return f"{self.collection.title} - {self.playlist.title}"
|
|
|
|
class Meta:
|
|
verbose_name = _('Video Playlist in Collection')
|
|
verbose_name_plural = _('Video Playlists in Collections')
|
|
ordering = ['order']
|
|
unique_together = ['collection', 'playlist']
|
|
|
|
|
|
class PlaylistItem(models.Model):
|
|
playlist = models.ForeignKey(
|
|
VideoPlaylist,
|
|
on_delete=models.CASCADE,
|
|
related_name='playlist_items',
|
|
verbose_name=_('playlist')
|
|
)
|
|
video = models.ForeignKey(
|
|
Video,
|
|
on_delete=models.CASCADE,
|
|
related_name='playlist_appearances',
|
|
verbose_name=_('video')
|
|
)
|
|
priority = models.PositiveIntegerField(default=0, verbose_name=_('priority'))
|
|
|
|
created_at = models.DateTimeField(auto_now_add=True, verbose_name=_('created at'))
|
|
updated_at = models.DateTimeField(auto_now=True, verbose_name=_('updated at'))
|
|
|
|
def __str__(self):
|
|
return f"{self.playlist.title} - {self.video.title} (Priority: {self.priority})"
|
|
|
|
class Meta:
|
|
verbose_name = _('Playlist Item')
|
|
verbose_name_plural = _('Playlist Items')
|
|
ordering = ['priority']
|
|
unique_together = ['playlist', 'video']
|
|
|