X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/967eed676fc83d15b26149047f353ac61faa8217..355e42e2b2244470ac6a5b9ad056e4a3f38f838c:/src/social/models.py

diff --git a/src/social/models.py b/src/social/models.py
index b9417bae5..588b2eb8e 100644
--- a/src/social/models.py
+++ b/src/social/models.py
@@ -1,21 +1,41 @@
-# -*- coding: utf-8 -*-
 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #
 from django.db import models
-from django.utils.translation import ugettext_lazy as _
 from django.conf import settings
+from django.core.exceptions import ValidationError
 from django.core.urlresolvers import reverse
+from django.utils.translation import ugettext_lazy as _
 from ssify import flush_ssi_includes
 from catalogue.models import Book
 
 
+class BannerGroup(models.Model):
+    name = models.CharField(_('name'), max_length=255, unique=True)
+    created_at = models.DateTimeField(_('created at'), auto_now_add=True)
+
+    class Meta:
+        ordering = ('name',)
+        verbose_name = _('banner group')
+        verbose_name_plural = _('banner groups')
+
+    def __str__(self):
+        return self.name
+
+    def get_absolute_url(self):
+        """This is used for testing."""
+        return "%s?banner_group=%d" % (reverse('main_page'), self.id)
+
+
 class Cite(models.Model):
     book = models.ForeignKey(Book, verbose_name=_('book'), null=True, blank=True)
     text = models.TextField(_('text'), blank=True)
     small = models.BooleanField(_('small'), default=False, help_text=_('Make this cite display smaller.'))
     vip = models.CharField(_('VIP'), max_length=128, null=True, blank=True)
     link = models.URLField(_('link'))
+    video = models.URLField(_('video'), blank=True)
+    picture = models.ImageField(_('picture'), blank=True)
+
     sticky = models.BooleanField(_('sticky'), default=False, db_index=True,
                                  help_text=_('Sticky cites will take precedense.'))
     banner = models.BooleanField(_('banner'), default=False, help_text=_('Adjust size to image, ignore the text'))
@@ -32,10 +52,13 @@ class Cite(models.Model):
     image_license = models.CharField(_('license name'), max_length=255, blank=True, null=True)
     image_license_link = models.URLField(_('license link'), blank=True, null=True)
 
+    created_at = models.DateTimeField(_('created at'), auto_now_add=True)
+    group = models.ForeignKey(BannerGroup, verbose_name=_('group'), null=True, blank=True, on_delete=models.SET_NULL)
+
     class Meta:
         ordering = ('vip', 'text')
-        verbose_name = _('cite')
-        verbose_name_plural = _('cites')
+        verbose_name = _('banner')
+        verbose_name_plural = _('banners')
 
     def __str__(self):
         return u"%s: %s…" % (self.vip, self.text[:60])
@@ -58,3 +81,36 @@ class Cite(models.Model):
             ]
             for lang in [lc for (lc, _ln) in settings.LANGUAGES]] +
             ['/ludzie/cite_info/%s.html' % self.pk])
+
+
+class Carousel(models.Model):
+    slug = models.SlugField(_('slug'), unique=True)
+
+    class Meta:
+        ordering = ('slug',)
+        verbose_name = _('carousel')
+        verbose_name_plural = _('carousels')
+
+    def __str__(self):
+        return self.slug
+
+class CarouselItem(models.Model):
+    order = models.PositiveSmallIntegerField(_('order'), unique=True)
+    carousel = models.ForeignKey(Carousel, models.CASCADE, verbose_name=_('carousel'))
+    banner = models.ForeignKey(Cite, models.CASCADE, null=True, blank=True, verbose_name=_('banner'))
+    banner_group = models.ForeignKey(BannerGroup, models.CASCADE, null=True, blank=True, verbose_name=_('banner group'))
+
+    class Meta:
+        ordering = ('order',)
+        unique_together = [('carousel', 'order')]
+        verbose_name = _('carousel item')
+        verbose_name_plural = _('carousel items')
+
+    def __str__(self):
+        return str(self.banner or self.banner_group)
+
+    def clean(self):
+        if not self.banner and not self.banner_group:
+            raise ValidationError(_('Either banner or banner group is required.'))
+        elif self.banner and self.banner_group:
+            raise ValidationError(_('Either banner or banner group is required.'))