Merge pull request #20 from arbisoft/feature/update-custom-admin-model

📹 Override VideoAsset save method to include fize size and duration

Dockerfile

@@ -3,6 +3,7 @@ FROM python:3.12-slim AS base
 RUN apt-get update && apt-get install -y \
     build-essential \
     libpq-dev \
+    ffmpeg \
     && rm -rf /var/lib/apt/lists/*
 
 ENV PYTHONUNBUFFERED 1

docker-compose.yml

@@ -28,7 +28,7 @@ services:
     command: bash -c "python manage.py runserver 0.0.0.0:${DJANGO_PORT}"
     volumes:
       - .:/app
-      - ./media:/app/media
+      - ./media:/app/arbisoft_sessions_portal/media
     ports:
       - "${DJANGO_PORT}:${DJANGO_PORT}"
     depends_on:

events/forms.py

@@ -12,7 +12,7 @@ class VideoAssetForm(forms.ModelForm):
 
     class Meta:
         model = VideoAsset
-        fields = ('title', 'video_file', 'duration', 'thumbnail', 'file_size')
+        fields = ('title', 'video_file', 'thumbnail')
 
     def clean(self):
         """ Infer, save and clean the data related to videoasset """

events/migrations/0003_alter_videoasset_duration_alter_videoasset_file_size.py

@@ -0,0 +1,23 @@
+# Generated by Django 4.2.16 on 2025-02-05 15:21
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('events', '0002_remove_videoasset_cdn_url_and_more'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='videoasset',
+            name='duration',
+            field=models.IntegerField(default=0),
+        ),
+        migrations.AlterField(
+            model_name='videoasset',
+            name='file_size',
+            field=models.BigIntegerField(default=0),
+        ),
+    ]

events/models.py

@@ -1,3 +1,5 @@
+import ffmpeg
+
 from django.conf import settings
 from django.contrib.auth import get_user_model
 from django.core.files.storage import FileSystemStorage
@@ -65,12 +67,32 @@ class VideoStatus(models.TextChoices):
     event = models.ForeignKey(Event, on_delete=models.DO_NOTHING, related_name='videos')
     title = models.CharField(max_length=255)
     video_file = models.FileField(storage=video_storage, null=True, blank=True)
-    duration = models.IntegerField()  # in seconds
+    duration = models.IntegerField(default=0)  # in seconds
     thumbnail = models.ImageField(storage=thumbnail_storage, null=True, blank=True)
     status = models.CharField(max_length=20, choices=VideoStatus.choices)
-    file_size = models.BigIntegerField()  # in bytes
+    file_size = models.BigIntegerField(default=0)  # in bytes
     created = models.DateTimeField(auto_now_add=True)
     modified = models.DateTimeField(auto_now=True)
 
+    def save(self, *args, **kwargs):
+        """ Override save method to extract file size and duration """
+        if self.video_file:
+            super().save(*args, **kwargs)
+
+            self.file_size = self.video_file.size
+            video_path = self.video_file.path
+            try:
+                metadata = ffmpeg.probe(video_path)
+                duration = float(metadata['format']['duration'])
+                self.duration = int(duration)  # Convert to seconds
+            except ffmpeg.Error as e:
+                print(f"FFmpeg processing error: {e.stderr.decode() if hasattr(e, 'stderr') else e}")
+            except KeyError:
+                print("Metadata does not contain 'duration'. Invalid file format.")
+            except ValueError:
+                print("Invalid duration value, unable to convert to float.")
+
+        super().save(*args, **kwargs)
+
     def __str__(self):
         return self.title

requirements/base.txt

@@ -7,6 +7,7 @@ django-filter==24.3
 djangorestframework==3.15.2
 djangorestframework-simplejwt==5.3.1
 drf-spectacular==0.28.0
+ffmpeg-python==0.2.0
 idna==3.10
 Markdown==3.7
 pillow==11.1.0