X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/8952b9530d943655e552ea660c47e850123c5105..ca8f4e8fef80cb603117ed579da8554f503698e5:/src/annoy/models.py

diff --git a/src/annoy/models.py b/src/annoy/models.py
index 9ed724c0e..e60e3c024 100644
--- a/src/annoy/models.py
+++ b/src/annoy/models.py
@@ -1,39 +1,53 @@
+import hashlib
+import json
 from django.apps import apps
 from django.conf import settings
 from django.db import models
-from django.utils.translation import ugettext_lazy as _
+from django.template import Context, Template
 from django.utils.timezone import now
-from .places import PLACES, PLACE_CHOICES
+from .places import PLACES, PLACE_CHOICES, STYLES
 
 
 class Banner(models.Model):
-    place = models.SlugField(_('place'), choices=PLACE_CHOICES)
+    place = models.SlugField('miejsce', choices=PLACE_CHOICES)
+    style = models.CharField(
+        'styl', max_length=255, blank=True,
+        choices=STYLES,
+        help_text='Dotyczy blackoutu.'
+    )
+    smallfont = models.BooleanField('mały font', default=False)
+    text_color = models.CharField(max_length=10, blank=True)
+    background_color = models.CharField(max_length=10, blank=True)
     action_label = models.CharField(
-        _('action label'),
+        'etykieta akcji',
         max_length=255, blank=True,
-        help_text=_('If empty, whole banner will serve as a link')
+        help_text='Jeśli pusta, cały banner będzie służył jako link.'
     )
-    open_label = models.CharField(_('open label'), max_length=255, blank=True)
-    close_label = models.CharField(_('close label'), max_length=255, blank=True)
-    text = models.TextField(_('text'))
-    image = models.FileField(_('image'), upload_to='annoy/banners/', blank=True)
-    url = models.CharField(_('url'), max_length=1024)
+    open_label = models.CharField('etykieta otwierania', max_length=255, blank=True)
+    close_label = models.CharField('etykieta zamykania', max_length=255, blank=True)
+    text = models.TextField('tekst')
+    image = models.FileField('obraz', upload_to='annoy/banners/', blank=True)
+    url = models.CharField('URL', max_length=1024)
     priority = models.PositiveSmallIntegerField(
-        _('priority'), default=0,
-        help_text=_('Banners with higher priority come first.'))
-    since = models.DateTimeField(_('since'), null=True, blank=True)
-    until = models.DateTimeField(_('until'), null=True, blank=True)
-    show_members = models.BooleanField(_('show members'), default=False)
-    staff_preview = models.BooleanField(_('staff preview'), default=False)
+        'priorytet', default=0,
+        help_text='Bannery z wyższym priorytetem mają pierwszeństwo.')
+    since = models.DateTimeField('od', null=True, blank=True)
+    until = models.DateTimeField('do', null=True, blank=True)
+    show_members = models.BooleanField('widoczny dla członków klubu', default=False)
+    staff_preview = models.BooleanField('podgląd tylko dla zespołu', default=False)
+    only_authenticated = models.BooleanField('tylko dla zalogowanych', default=False)
 
     class Meta:
-        verbose_name = _('banner')
-        verbose_name_plural = _('banners')
+        verbose_name = 'banner'
+        verbose_name_plural = 'bannery'
         ordering = ('place', '-priority',)
 
     def __str__(self):
         return self.text
 
+    def get_text(self):
+        return Template(self.text).render(Context())
+
     @classmethod
     def choice(cls, place, request):
         Membership = apps.get_model('club', 'Membership')
@@ -53,22 +67,25 @@ class Banner(models.Model):
             until__lt=n
         ).order_by('-priority', '?')
 
+        if not request.user.is_authenticated:
+            banners = banners.filter(only_authenticated=False)
+
         if not request.user.is_staff:
             banners = banners.filter(staff_preview=False)
 
-        if request:
-            if Membership.is_active_for(request.user):
-                banners = banners.filter(show_members=True)
+        if Membership.is_active_for(request.user):
+            banners = banners.filter(show_members=True)
+
         return banners
 
 
 class DynamicTextInsert(models.Model):
-    paragraphs = models.IntegerField(_('pararaphs'))
+    paragraphs = models.IntegerField('akapity')
     url = models.CharField(max_length=1024)
 
     class Meta:
-        verbose_name = _('dynamic insert')
-        verbose_name_plural = _('dynamic inserts')
+        verbose_name = 'dynamiczna wstawka'
+        verbose_name_plural = 'dynamiczne wstawki'
         ordering = ('paragraphs', )
 
     def __str__(self):
@@ -91,5 +108,50 @@ class DynamicTextInsertText(models.Model):
     own_colors = models.BooleanField(default=False)
     background_color = models.CharField(max_length=10, blank=True)
     text_color = models.CharField(max_length=10, blank=True)
-    text = models.TextField(_('text'))
+    text = models.TextField('tekst')
     image = models.FileField(blank=True, upload_to='annoy/inserts/')
+
+
+class MediaInsertSet(models.Model):
+    file_format = models.CharField(max_length=8, choices=[
+        ('epub', 'epub'),
+        ('mobi', 'mobi'),
+        ('pdf', 'pdf'),
+        ])
+    etag = models.CharField(max_length=64, blank=True)
+
+    def update_etag(self):
+        self.etag = hashlib.sha1(json.dumps(self.get_texts()).encode('utf-8')).hexdigest()
+        self.save(update_fields=['etag'])
+
+    def get_texts(self):
+        return [t.text for t in self.mediainserttext_set.all()]
+
+    @classmethod
+    def get_for_format(cls, file_format):
+        return cls.objects.filter(file_format=file_format).first()
+
+    @classmethod
+    def get_texts_for(cls, file_format):
+        self = cls.get_for_format(file_format)
+        if self is None:
+            return []
+        return self.get_texts()
+
+
+class MediaInsertText(models.Model):
+    media_insert_set = models.ForeignKey(MediaInsertSet, models.CASCADE)
+    ordering = models.IntegerField()
+    text = models.TextField()
+
+    class Meta:
+        ordering = ('ordering',)
+
+
+from django.db.models.signals import post_save, post_delete
+from django.dispatch import receiver
+
+@receiver(post_delete, sender=MediaInsertText)
+@receiver(post_save, sender=MediaInsertText)
+def update_etag(sender, instance, **kwargs):
+    instance.media_insert_set.update_etag()