X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/07cd561683d722537e51ca728c584abb7caec92f..14e08cfc833a63ad2c51aa3ac0ce60960d4ace0d:/src/annoy/models.py

diff --git a/src/annoy/models.py b/src/annoy/models.py
index 6af1cc045..7ef119ec2 100644
--- a/src/annoy/models.py
+++ b/src/annoy/models.py
@@ -5,16 +5,67 @@ 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='Dla zespołu')
+    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)
@@ -44,8 +95,6 @@ class Banner(models.Model):
     until = models.DateTimeField('do', null=True, blank=True)
     books = models.ManyToManyField('catalogue.Book', blank=True)
 
-    target = models.IntegerField('cel', null=True, blank=True)
-    progress = models.IntegerField('postęp', 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)
@@ -67,6 +116,12 @@ class Banner(models.Model):
         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
@@ -86,11 +141,15 @@ 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)
@@ -109,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):
@@ -183,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):