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