X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/4bf7a487910abd0487983e85722d134aa9bb95fe..14e08cfc833a63ad2c51aa3ac0ce60960d4ace0d:/src/annoy/models.py diff --git a/src/annoy/models.py b/src/annoy/models.py index 0d887920d..7ef119ec2 100644 --- a/src/annoy/models.py +++ b/src/annoy/models.py @@ -5,11 +5,72 @@ from django.conf import settings from django.db import models from django.template import Context, Template from django.utils.timezone import now +from lxml import etree, html from .places import PLACES, PLACE_CHOICES, STYLES +class Campaign(models.Model): + name = models.CharField(max_length=255, help_text='Wewnętrzna nazwa kampanii') + image = models.FileField('obraz', upload_to='annoy/banners/', blank=True) + start = models.DateTimeField(null=True, blank=True) + end = models.DateTimeField(null=True, blank=True) + landing = models.CharField(blank=True, max_length=1024) + target = models.IntegerField('cel', null=True, blank=True) + progress = models.IntegerField('postęp', null=True, blank=True) + priority = models.PositiveSmallIntegerField( + 'priorytet', default=0, + help_text='Kampanie z wyższym priorytetem mają pierwszeństwo.') + + def __str__(self): + return self.name + + @classmethod + def get_active(cls): + n = now() + return cls.objects.exclude(start__gt=n).exclude(end__lt=n) + + def has_progress(self): + return self.target is not None and self.start is not None and self.end is not None + + @property + def progress_percent(self): + if not self.target: + return 0 + return (self.progress or 0) / self.target * 100 + + @property + def progress_percent_pretty(self): + return int(self.progress_percent) + + def update_progress(self): + if not self.start or not self.end or not self.target: + return + Schedule = apps.get_model('club', 'Schedule') + PayUOrder = apps.get_model('club', 'PayUOrder') + progress = PayUOrder.objects.filter( + completed_at__gte=self.start, + completed_at__lte=self.end, + ).aggregate(c=models.Sum('schedule__amount'))['c'] or 0 + + for schedule in Schedule.objects.filter( + method='paypal', + expires_at__gt=self.start + ): + progress += schedule.n_paypal_payments(self.start, self.end) * schedule.amount + self.progress = progress + self.save(update_fields=['progress']) + + @classmethod + def update_all_progress(cls): + n = now() + for obj in cls.get_active().exclude(target=None): + obj.update_progress() + + class Banner(models.Model): place = models.SlugField('miejsce', choices=PLACE_CHOICES) + campaign = models.ForeignKey(Campaign, models.PROTECT, null=True, blank=True) + style = models.CharField( 'styl', max_length=255, blank=True, choices=STYLES, @@ -32,8 +93,8 @@ class Banner(models.Model): 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) - target = models.IntegerField('cel', null=True, blank=True) - progress = models.IntegerField('postęp', null=True, blank=True) + books = models.ManyToManyField('catalogue.Book', 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) @@ -49,8 +110,26 @@ class Banner(models.Model): def get_text(self): return Template(self.text).render(Context()) + def get_image(self): + if self.campaign and self.campaign.image: + return self.campaign.image + else: + return self.image + + def get_url(self): + if self.url: + return self.url + if self.campaign: + return self.campaign.landing + + def is_external(self): + return (self.url and + not self.url.startswith('/') and + not self.url.startswith('https://wolnelektury.pl/') + ) + @classmethod - def choice(cls, place, request, exemptions=True): + def choice(cls, place, request, exemptions=True, book=None): Membership = apps.get_model('club', 'Membership') if exemptions and hasattr(request, 'annoy_banner_exempt'): @@ -62,12 +141,22 @@ class Banner(models.Model): n = now() banners = cls.objects.filter( place=place + ).exclude( + campaign__start__gt=n + ).exclude( + campaign__end__lt=n ).exclude( since__gt=n ).exclude( until__lt=n - ).order_by('-priority', '?') - + ).order_by('-campaign__priority', '-priority', '?') + + if book is None: + banners = banners.filter(books=None) + else: + banners = banners.filter(models.Q(books=None) | models.Q(books=book)) + + if not request.user.is_authenticated: banners = banners.filter(only_authenticated=False) @@ -79,32 +168,6 @@ class Banner(models.Model): return banners - @property - def progress_percent(self): - if not self.target: - return 0 - return (self.progress or 0) / self.target * 100 - - @property - def progress_percent_pretty(self): - return int(self.progress_percent) - - def update_progress(self): - # Total of new payments during the action. - # This definition will need to change for longer timespans. - if not self.since or not self.until or not self.target: - return - Schedule = apps.get_model('club', 'Schedule') - self.progress = Schedule.objects.filter( - payed_at__gte=self.since, - payed_at__lte=self.until, - ).aggregate(c=models.Sum('amount'))['c'] - self.save(update_fields=['progress']) - - @classmethod - def update_all_progress(cls): - for obj in cls.objects.exclude(target=None): - obj.update_progress() class DynamicTextInsert(models.Model): @@ -153,7 +216,10 @@ class MediaInsertSet(models.Model): self.save(update_fields=['etag']) def get_texts(self): - return [t.text for t in self.mediainserttext_set.all()] + return [ + etree.tostring(html.fromstring(t.text), encoding='unicode') + for t in self.mediainserttext_set.all() + ] @classmethod def get_for_format(cls, file_format):