X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/f5129e7c47fa5fcf0a20f04249b90afcbc54cb9c..HEAD:/src/annoy/models.py diff --git a/src/annoy/models.py b/src/annoy/models.py index 26b0136c7..e60e3c024 100644 --- a/src/annoy/models.py +++ b/src/annoy/models.py @@ -1,44 +1,62 @@ +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(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( + 'etykieta akcji', max_length=255, blank=True, - help_text=_('') + help_text='Jeśli pusta, cały banner będzie służył jako link.' ) - open_label = models.CharField(max_length=255, blank=True) - close_label = models.CharField(max_length=255, blank=True) - text = models.TextField() - url = models.CharField(max_length=1024) - priority = models.PositiveSmallIntegerField(default=0) - since = models.DateTimeField(null=True, blank=True) - until = models.DateTimeField(null=True, blank=True) - show_members = models.BooleanField(default=False) - staff_preview = models.BooleanField(default=False) + 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( + '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') if hasattr(request, 'annoy_banner_exempt'): return cls.objects.none() - + if settings.DEBUG: - assert place in PLACES, "Banner place `{}` must be defined in annoy.places.".format(place) + assert place in PLACES, f"Banner place `{place}` must be defined in annoy.places." n = now() banners = cls.objects.filter( @@ -49,34 +67,91 @@ 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): return str(self.paragraphs) + @classmethod + def get_all(cls, request): + Membership = apps.get_model('club', 'Membership') + if Membership.is_active_for(request.user) and not request.user.is_staff: + return cls.objects.none() + return cls.objects.all() + + def choose(self): return self.dynamictextinserttext_set.order_by('?').first() class DynamicTextInsertText(models.Model): insert = models.ForeignKey(DynamicTextInsert, models.CASCADE) + 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')) - image = models.FileField(blank=True) + 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()