src/annoy/models.py

   1 import hashlib
   2 import json
   3 from django.apps import apps
   4 from django.conf import settings
   5 from django.db import models
   6 from django.template import Context, Template
   7 from django.utils.timezone import now
   8 from lxml import etree, html
   9 from .places import PLACES, PLACE_CHOICES, STYLES
  10
  11
  12 class Campaign(models.Model):
  13     name = models.CharField(max_length=255, help_text='Wewnętrzna nazwa kampanii')
  14     image = models.FileField('obraz', upload_to='annoy/banners/', blank=True)
  15     start = models.DateTimeField(null=True, blank=True)
  16     end = models.DateTimeField(null=True, blank=True)
  17     landing = models.CharField(blank=True, max_length=1024)
  18     target = models.IntegerField('cel', null=True, blank=True)
  19     progress = models.IntegerField('postęp', null=True, blank=True)
  20     priority = models.PositiveSmallIntegerField(
  21         'priorytet', default=0,
  22         help_text='Kampanie z wyższym priorytetem mają pierwszeństwo.')
  23
  24     def __str__(self):
  25         return self.name
  26
  27     @classmethod
  28     def get_active(cls):
  29         n = now()
  30         return cls.objects.exclude(start__gt=n).exclude(end__lt=n)
  31
  32     def has_progress(self):
  33         return self.target is not None and self.start is not None and self.end is not None
  34
  35     @property
  36     def progress_percent(self):
  37         if not self.target:
  38             return 0
  39         return (self.progress or 0) / self.target * 100
  40
  41     @property
  42     def progress_percent_pretty(self):
  43         return int(self.progress_percent)
  44
  45     def update_progress(self):
  46         if not self.start or not self.end or not self.target:
  47             return
  48         Schedule = apps.get_model('club', 'Schedule')
  49         PayUOrder = apps.get_model('club', 'PayUOrder')
  50         progress = PayUOrder.objects.filter(
  51             completed_at__gte=self.start,
  52             completed_at__lte=self.end,
  53         ).aggregate(c=models.Sum('schedule__amount'))['c'] or 0
  54
  55         for schedule in Schedule.objects.filter(
  56                 method='paypal',
  57                 expires_at__gt=self.start
  58         ):
  59             progress += schedule.n_paypal_payments(self.start, self.end) * schedule.amount
  60         self.progress = progress
  61         self.save(update_fields=['progress'])
  62
  63     @classmethod
  64     def update_all_progress(cls):
  65         n = now()
  66         for obj in cls.get_active().exclude(target=None):
  67             obj.update_progress()
  68
  69
  70 class Banner(models.Model):
  71     place = models.SlugField('miejsce', choices=PLACE_CHOICES)
  72     campaign = models.ForeignKey(Campaign, models.PROTECT, null=True, blank=True)
  73
  74     style = models.CharField(
  75         'styl', max_length=255, blank=True,
  76         choices=STYLES,
  77     )
  78     smallfont = models.BooleanField('mały font', default=False)
  79     text_color = models.CharField(max_length=10, blank=True)
  80     background_color = models.CharField(max_length=10, blank=True)
  81     action_label = models.CharField(
  82         'etykieta akcji',
  83         max_length=255, blank=True,
  84         help_text='Jeśli pusta, cały banner będzie służył jako link.'
  85     )
  86     open_label = models.CharField('etykieta otwierania', max_length=255, blank=True)
  87     close_label = models.CharField('etykieta zamykania', max_length=255, blank=True)
  88     text = models.TextField('tekst')
  89     image = models.FileField('obraz', upload_to='annoy/banners/', blank=True)
  90     url = models.CharField('URL', max_length=1024)
  91     priority = models.PositiveSmallIntegerField(
  92         'priorytet', default=0,
  93         help_text='Bannery z wyższym priorytetem mają pierwszeństwo.')
  94     since = models.DateTimeField('od', null=True, blank=True)
  95     until = models.DateTimeField('do', null=True, blank=True)
  96     books = models.ManyToManyField('catalogue.Book', blank=True)
  97
  98     show_members = models.BooleanField('widoczny dla członków klubu', default=False)
  99     staff_preview = models.BooleanField('podgląd tylko dla zespołu', default=False)
 100     only_authenticated = models.BooleanField('tylko dla zalogowanych', default=False)
 101
 102     class Meta:
 103         verbose_name = 'banner'
 104         verbose_name_plural = 'bannery'
 105         ordering = ('place', '-priority',)
 106
 107     def __str__(self):
 108         return self.text
 109
 110     def get_text(self):
 111         return Template(self.text).render(Context())
 112
 113     def get_image(self):
 114         if self.campaign and self.campaign.image:
 115             return self.campaign.image
 116         else:
 117             return self.image
 118
 119     def get_url(self):
 120         if self.url:
 121             return self.url
 122         if self.campaign:
 123             return self.campaign.landing
 124
 125     def is_external(self):
 126         return (self.url and
 127                 not self.url.startswith('/') and
 128                 not self.url.startswith('https://wolnelektury.pl/')
 129                 )
 130
 131     @classmethod
 132     def choice(cls, place, request, exemptions=True, book=None):
 133         Membership = apps.get_model('club', 'Membership')
 134
 135         if exemptions and hasattr(request, 'annoy_banner_exempt'):
 136             return cls.objects.none()
 137
 138         if settings.DEBUG:
 139             assert place in PLACES, f"Banner place `{place}` must be defined in annoy.places."
 140
 141         n = now()
 142         banners = cls.objects.filter(
 143             place=place
 144         ).exclude(
 145             campaign__start__gt=n
 146         ).exclude(
 147             campaign__end__lt=n
 148         ).exclude(
 149             since__gt=n
 150         ).exclude(
 151             until__lt=n
 152         ).order_by('-campaign__priority', '-priority', '?')
 153
 154         if book is None:
 155             banners = banners.filter(books=None)
 156         else:
 157             banners = banners.filter(models.Q(books=None) | models.Q(books=book))
 158
 159
 160         if not request.user.is_authenticated:
 161             banners = banners.filter(only_authenticated=False)
 162
 163         if not request.user.is_staff:
 164             banners = banners.filter(staff_preview=False)
 165
 166         if Membership.is_active_for(request.user):
 167             banners = banners.filter(show_members=True)
 168
 169         return banners
 170
 171
 172
 173 class DynamicTextInsert(models.Model):
 174     paragraphs = models.IntegerField('akapity')
 175     url = models.CharField(max_length=1024)
 176
 177     class Meta:
 178         verbose_name = 'dynamiczna wstawka'
 179         verbose_name_plural = 'dynamiczne wstawki'
 180         ordering = ('paragraphs', )
 181
 182     def __str__(self):
 183         return str(self.paragraphs)
 184
 185     @classmethod
 186     def get_all(cls, request):
 187         Membership = apps.get_model('club', 'Membership')
 188         if Membership.is_active_for(request.user) and not request.user.is_staff:
 189             return cls.objects.none()
 190         return cls.objects.all()
 191
 192
 193     def choose(self):
 194         return self.dynamictextinserttext_set.order_by('?').first()
 195
 196
 197 class DynamicTextInsertText(models.Model):
 198     insert = models.ForeignKey(DynamicTextInsert, models.CASCADE)
 199     own_colors = models.BooleanField(default=False)
 200     background_color = models.CharField(max_length=10, blank=True)
 201     text_color = models.CharField(max_length=10, blank=True)
 202     text = models.TextField('tekst')
 203     image = models.FileField(blank=True, upload_to='annoy/inserts/')
 204
 205
 206 class MediaInsertSet(models.Model):
 207     file_format = models.CharField(max_length=8, choices=[
 208         ('epub', 'epub'),
 209         ('mobi', 'mobi'),
 210         ('pdf', 'pdf'),
 211         ])
 212     etag = models.CharField(max_length=64, blank=True)
 213
 214     def update_etag(self):
 215         self.etag = hashlib.sha1(json.dumps(self.get_texts()).encode('utf-8')).hexdigest()
 216         self.save(update_fields=['etag'])
 217
 218     def get_texts(self):
 219         return [
 220             etree.tostring(html.fromstring(t.text), encoding='unicode')
 221             for t in self.mediainserttext_set.all()
 222         ]
 223
 224     @classmethod
 225     def get_for_format(cls, file_format):
 226         return cls.objects.filter(file_format=file_format).first()
 227
 228     @classmethod
 229     def get_texts_for(cls, file_format):
 230         self = cls.get_for_format(file_format)
 231         if self is None:
 232             return []
 233         return self.get_texts()
 234
 235
 236 class MediaInsertText(models.Model):
 237     media_insert_set = models.ForeignKey(MediaInsertSet, models.CASCADE)
 238     ordering = models.IntegerField()
 239     text = models.TextField()
 240
 241     class Meta:
 242         ordering = ('ordering',)