090500

system/media/models.py

@@ -0,0 +1,115 @@
+import os
+from django.utils import timezone
+from django.db import models
+from django.contrib.auth import get_user_model
+from django.core.exceptions import ValidationError
+from simple_history.models import HistoricalRecords
+from django.utils.translation import gettext as _
+#
+from .validators import validate_max_file_size
+#
+User = get_user_model()
+#
+
+
+""" Абстрактная базовая модель для добавления автора полей времени создания и изменения."""
+
+
+class TimeStampedModel(models.Model):
+    created_at = models.DateTimeField(auto_now_add=True, verbose_name='Дата создания')
+    updated_at = models.DateTimeField(auto_now=True, verbose_name='Дата обновления')
+    author = models.ForeignKey(User, on_delete=models.SET_NULL, null=True, verbose_name='Автор',
+                               help_text='Выберите автора объекта')
+    file_name = models.CharField(max_length=255, verbose_name='Имя файла', help_text='Введите имя файла')
+
+    class Meta:
+        abstract = True
+
+    def save(self, *args, **kwargs):
+        if not self.pk:  # Проверяем, что это новый объект (не обновление)
+            self.created_at = timezone.now()  # Автоматически устанавливаем дату создания
+        self.updated_at = timezone.now()  # Всегда обновляем дату обновления
+        super().save(*args, **kwargs)  # Вызываем оригинальный метод сохранения
+
+    def __str__(self):
+        return f'{self.__class__.__name__}: {self.file_name}'
+
+    def generate_file_url(self):
+        if self.file_name:
+            return os.path.join('media', str(self.file_name))
+
+
+""" Модель файлов изображения """
+
+
+class Images(TimeStampedModel):
+    file = models.ImageField(upload_to='images/%Y/%m/%d/', verbose_name='Изображение',
+                             help_text='Загрузите изображение', validators=[validate_max_file_size])
+
+    history = HistoricalRecords()
+
+    def save(self, *args, **kwargs):
+        self.name = f'Image_{self.file_name}'  # Префикс "Image_" для изображений
+        super().save(*args, **kwargs)
+
+    class Meta:
+        verbose_name = 'Изображение'
+        verbose_name_plural = 'Изображения'
+        ordering = ['-created_at']
+
+    def clean(self):
+        super().clean()
+        if self.file:
+            # Проверяем расширение файла
+            ext = self.file.name.split('.')[-1]
+            if ext.lower() not in ['jpg', 'jpeg', 'png', 'gif']:
+                raise ValidationError(_(
+                    'Недопустимый формат изображения. Поддерживаемые форматы: JPG, JPEG, PNG, GIF.'
+                ))
+
+    def __str__(self):
+        return f'{self.__class__.__name__}: {self.file_name}'
+
+
+""" Модель видео файлов """
+
+
+class Videos(TimeStampedModel):
+    file = models.FileField(upload_to='videos/%Y/%m/%d/', verbose_name='Видео',
+                            help_text='Загрузите видео', validators=[validate_max_file_size])
+
+    history = HistoricalRecords()
+
+    def save(self, *args, **kwargs):
+        self.file_name = f'Video_{self.file_name}'  # Префикс "Video_" для видео
+        super().save(*args, **kwargs)
+
+    class Meta:
+        verbose_name = 'Видео'
+        verbose_name_plural = 'Видео'
+        ordering = ['-created_at']
+
+    def __str__(self):
+        return f'{self.__class__.__name__}: {self.file_name}'
+
+
+""" Модель файлов всех типов документов """
+
+
+class OtherFiles(TimeStampedModel):
+    file = models.FileField(upload_to='documents/%Y/%m/%d/', verbose_name='Документ',
+                            help_text='Загрузите документ', validators=[validate_max_file_size])
+
+    history = HistoricalRecords()
+
+    def save(self, *args, **kwargs):
+        self.file_name = f'Document_{self.file_name}'  # Префикс "Document_" для документов
+        super().save(*args, **kwargs)
+
+    class Meta:
+        verbose_name = 'Документ'
+        verbose_name_plural = 'Документы'
+        ordering = ['-created_at']
+
+    def __str__(self):
+        return f'{self.__class__.__name__}: {self.file_name}'