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     video = models.FileField('video', upload_to='annoy/banners/', blank=True)
  91     url = models.CharField('URL', max_length=1024)
  92     priority = models.PositiveSmallIntegerField(
  93         'priorytet', default=0,
  94         help_text='Bannery z wyższym priorytetem mają pierwszeństwo.')
  95     since = models.DateTimeField('od', null=True, blank=True)
  96     until = models.DateTimeField('do', null=True, blank=True)
  97     books = models.ManyToManyField('catalogue.Book', blank=True)
  98
  99     show_members = models.BooleanField('widoczny dla członków klubu', default=False)
 100     staff_preview = models.BooleanField('podgląd tylko dla zespołu', default=False)
 101     only_authenticated = models.BooleanField('tylko dla zalogowanych', default=False)
 102
 103     class Meta:
 104         verbose_name = 'banner'
 105         verbose_name_plural = 'bannery'
 106         ordering = ('place', '-priority',)
 107
 108     def __str__(self):
 109         return self.text
 110
 111     def get_text(self):
 112         return Template(self.text).render(Context())
 113
 114     def get_image(self):
 115         if self.campaign and self.campaign.image:
 116             return self.campaign.image
 117         else:
 118             return self.image
 119
 120     def get_url(self):
 121         if self.url:
 122             return self.url
 123         if self.campaign:
 124             return self.campaign.landing
 125
 126     def is_external(self):
 127         return (self.url and
 128                 not self.url.startswith('/') and
 129                 not self.url.startswith('https://wolnelektury.pl/')
 130                 )
 131
 132     @classmethod
 133     def choice(cls, place, request, exemptions=True, book=None):
 134         Membership = apps.get_model('club', 'Membership')
 135
 136         if exemptions and hasattr(request, 'annoy_banner_exempt'):
 137             return cls.objects.none()
 138
 139         if settings.DEBUG:
 140             assert place in PLACES, f"Banner place `{place}` must be defined in annoy.places."
 141
 142         n = now()
 143         banners = cls.objects.filter(
 144             place=place
 145         ).exclude(
 146             campaign__start__gt=n
 147         ).exclude(
 148             campaign__end__lt=n
 149         ).exclude(
 150             since__gt=n
 151         ).exclude(
 152             until__lt=n
 153         ).order_by('-campaign__priority', '-priority', '?')
 154
 155         if book is None:
 156             banners = banners.filter(books=None)
 157         else:
 158             banners = banners.filter(models.Q(books=None) | models.Q(books=book))
 159
 160
 161         if not request.user.is_authenticated:
 162             banners = banners.filter(only_authenticated=False)
 163
 164         if not request.user.is_staff:
 165             banners = banners.filter(staff_preview=False)
 166
 167         if Membership.is_active_for(request.user):
 168             banners = banners.filter(show_members=True)
 169
 170         return banners
 171
 172
 173
 174 class DynamicTextInsert(models.Model):
 175     paragraphs = models.IntegerField('akapity')
 176     url = models.CharField(max_length=1024)
 177
 178     class Meta:
 179         verbose_name = 'dynamiczna wstawka'
 180         verbose_name_plural = 'dynamiczne wstawki'
 181         ordering = ('paragraphs', )
 182
 183     def __str__(self):
 184         return str(self.paragraphs)
 185
 186     @classmethod
 187     def get_all(cls, request):
 188         Membership = apps.get_model('club', 'Membership')
 189         if Membership.is_active_for(request.user) and not request.user.is_staff:
 190             return cls.objects.none()
 191         return cls.objects.all()
 192
 193
 194     def choose(self):
 195         return self.dynamictextinserttext_set.order_by('?').first()
 196
 197
 198 class DynamicTextInsertText(models.Model):
 199     insert = models.ForeignKey(DynamicTextInsert, models.CASCADE)
 200     own_colors = models.BooleanField(default=False)
 201     background_color = models.CharField(max_length=10, blank=True)
 202     text_color = models.CharField(max_length=10, blank=True)
 203     text = models.TextField('tekst')
 204     image = models.FileField(blank=True, upload_to='annoy/inserts/')
 205
 206
 207 class MediaInsertSet(models.Model):
 208     file_format = models.CharField(max_length=8, choices=[
 209         ('epub', 'epub'),
 210         ('mobi', 'mobi'),
 211         ('pdf', 'pdf'),
 212         ])
 213     etag = models.CharField(max_length=64, blank=True)
 214
 215     def update_etag(self):
 216         self.etag = hashlib.sha1(json.dumps(self.get_texts()).encode('utf-8')).hexdigest()
 217         self.save(update_fields=['etag'])
 218
 219     def get_texts(self):
 220         return [
 221             etree.tostring(html.fromstring(t.text), encoding='unicode')
 222             for t in self.mediainserttext_set.all()
 223         ]
 224
 225     @classmethod
 226     def get_for_format(cls, file_format):
 227         return cls.objects.filter(file_format=file_format).first()
 228
 229     @classmethod
 230     def get_texts_for(cls, file_format):
 231         self = cls.get_for_format(file_format)
 232         if self is None:
 233             return []
 234         return self.get_texts()
 235
 236
 237 class MediaInsertText(models.Model):
 238     media_insert_set = models.ForeignKey(MediaInsertSet, models.CASCADE)
 239     ordering = models.IntegerField()
 240     text = models.TextField()
 241
 242     class Meta:
 243         ordering = ('ordering',)