LilSchool/apps/content/models.py

from urllib.parse import urlparse
from django.db import models
from django.contrib.auth import get_user_model
from django.urls import reverse_lazy
from imagekit.models import ImageSpecField
from imagekit.processors import ResizeToCover

from polymorphic.models import PolymorphicModel


User = get_user_model()


class ImageObject(models.Model):
    image = models.ImageField('Изображение', upload_to='content/imageobject')
    image_thumbnail = ImageSpecField(source='image',
                         processors=[ResizeToCover(300, 200, False)],
                         format='JPEG',
                         options={'quality': 85})

    created_at = models.DateTimeField(auto_now_add=True)
    update_at = models.DateTimeField(auto_now=True)

    class Meta:
        verbose_name = 'Объект изображения'
        verbose_name_plural = 'Объекты изображения'
        ordering = ('-created_at',)


class Content(PolymorphicModel):
    uuid = models.UUIDField(null=True, blank=True)
    course = models.ForeignKey(
        'course.Course', on_delete=models.CASCADE,
        null=True, blank=True,
        verbose_name='Курс',
        related_name='content',
    )
    lesson = models.ForeignKey(
        'course.Lesson', on_delete=models.CASCADE,
        null=True, blank=True,
        verbose_name='Урок',
        related_name='content',
    )
    live_lesson = models.ForeignKey(
        'school.LiveLesson', on_delete=models.CASCADE,
        null=True, blank=True,
        verbose_name='Урок онлайн школы',
        related_name='content',
    )
    contest = models.ForeignKey(
        'Contest', on_delete=models.CASCADE,
        null=True, blank=True,
        verbose_name='Конкурс',
        related_name='content',
    )
    title = models.CharField('Заголовок', max_length=100, default='')
    position = models.PositiveSmallIntegerField(
        'Положение на странице',
        default=1,
    )

    created_at = models.DateTimeField(auto_now_add=True)
    update_at = models.DateTimeField(auto_now=True)

    class Meta:
        verbose_name = 'Контент'
        verbose_name_plural = 'Контент'
        ordering = ('position', '-created_at',)

    def ctype(self):
        return self.__class__.__name__.lower()


class Image(Content):
    img = models.ForeignKey(
        ImageObject, related_name='content_images',
        verbose_name='Объект изображения', on_delete=models.CASCADE,
        null=True
    )


class Text(Content):
    txt = models.TextField('Текст', default='')


class ImageText(Content):
    img = models.ForeignKey(
        ImageObject, related_name='content_imagetexts',
        verbose_name='Объект изображения', on_delete=models.CASCADE,
        null=True
    )
    txt = models.TextField('Текст', default='')


class Video(Content):
    url = models.URLField('Ссылка')

    def video_index(self):
        if 'youtu.be' in self.url or 'youtube.com' in self.url and 'watch' in self.url:
            url = urlparse(self.url)
            query = url.query.split('&')
            for q in query:
                if 'v=' in q:
                    return q.split('=')[-1]
        return self.url.split('/')[-1]


class Gallery(Content):
    pass


class GalleryImage(models.Model):
    gallery = models.ForeignKey(
        Gallery, on_delete=models.CASCADE,
        verbose_name='Галерея', related_name='gallery_images'
    )
    img = models.ForeignKey(
        ImageObject, related_name='gallery_images',
        verbose_name='Объект изображения', on_delete=models.CASCADE,
        null=True, blank=True,
    )

    created_at = models.DateTimeField(auto_now_add=True)
    update_at = models.DateTimeField(auto_now=True)

    class Meta:
        verbose_name = 'Изображение в галерее'
        verbose_name_plural = 'Изображения в галерее'
        ordering = ('-created_at',)


class Baner(models.Model):
    text = models.TextField()
    button_text = models.CharField(max_length=50)
    url = models.URLField()
    image = models.ImageField()
    use = models.BooleanField(default=False)

    created_at = models.DateTimeField(auto_now_add=True)
    update_at = models.DateTimeField(auto_now=True)

    class Meta:
        verbose_name = 'Банер'
        verbose_name_plural = 'Банеры'
        ordering = ('-created_at',)

    def save(self, *args, **kwargs):
        if self.use:
            Baner.objects.filter(use=True).update(use=False)
        return super().save(*args, **kwargs)


class Contest(models.Model):
    title = models.CharField(max_length=255)
    description = models.TextField(max_length=1000, blank=True, default='')
    slug = models.SlugField(
        allow_unicode=True, null=True, blank=True,
        max_length=100, unique=True, db_index=True,
    )
    cover = models.ForeignKey(
        ImageObject, related_name='contest_covers',
        verbose_name='Фоновая картинка', on_delete=models.CASCADE,
        null=True, blank=True,
    )
    date_start = models.DateField('Дата начала', null=True, blank=True)
    date_end = models.DateField('Дата окончания', null=True, blank=True)
    active = models.BooleanField(default=True)
    # TODO? baner

    def save(self, *args, **kwargs):
        if self.active:
            Contest.objects.filter(active=True).update(active=False)
        return super().save(*args, **kwargs)


class ContestWork(models.Model):
    user = models.ForeignKey(
        User, on_delete=models.CASCADE
    )
    contest = models.ForeignKey(Contest, on_delete=models.CASCADE)
    image = models.ForeignKey(
        ImageObject, related_name='contest_work_images',
        verbose_name='Работа участника', on_delete=models.CASCADE,
    )
    child_full_name = models.CharField(max_length=255)
    age = models.SmallIntegerField()
    created_at = models.DateTimeField(auto_now_add=True)
    likes = models.ManyToManyField('course.Like', blank=True)

    class Meta:
        verbose_name = 'Конкурсная работа'
        verbose_name_plural = 'Конкурсные работы'
        ordering = ('-created_at',)

    @property
    def img_width(self):
        return self.image.image.width if self.image and self.image.image else None

    @property
    def img_height(self):
        return self.image.image.height if self.image and self.image.image else None

    def get_absolute_url(self):
        return reverse_lazy('contest_work', args=[self.id])