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 Banner(models.Model):
  12     place = models.SlugField('miejsce', choices=PLACE_CHOICES)
  13     style = models.CharField(
  14         'styl', max_length=255, blank=True,
  15         choices=STYLES,
  16     )
  17     smallfont = models.BooleanField('mały font', default=False)
  18     text_color = models.CharField(max_length=10, blank=True)
  19     background_color = models.CharField(max_length=10, blank=True)
  20     action_label = models.CharField(
  21         'etykieta akcji',
  22         max_length=255, blank=True,
  23         help_text='Jeśli pusta, cały banner będzie służył jako link.'
  24     )
  25     open_label = models.CharField('etykieta otwierania', max_length=255, blank=True)
  26     close_label = models.CharField('etykieta zamykania', max_length=255, blank=True)
  27     text = models.TextField('tekst')
  28     image = models.FileField('obraz', upload_to='annoy/banners/', blank=True)
  29     url = models.CharField('URL', max_length=1024)
  30     priority = models.PositiveSmallIntegerField(
  31         'priorytet', default=0,
  32         help_text='Bannery z wyższym priorytetem mają pierwszeństwo.')
  33     since = models.DateTimeField('od', null=True, blank=True)
  34     until = models.DateTimeField('do', null=True, blank=True)
  35     target = models.IntegerField('cel', null=True, blank=True)
  36     progress = models.IntegerField('postęp', null=True, blank=True)
  37     show_members = models.BooleanField('widoczny dla członków klubu', default=False)
  38     staff_preview = models.BooleanField('podgląd tylko dla zespołu', default=False)
  39     only_authenticated = models.BooleanField('tylko dla zalogowanych', default=False)
  40
  41     class Meta:
  42         verbose_name = 'banner'
  43         verbose_name_plural = 'bannery'
  44         ordering = ('place', '-priority',)
  45
  46     def __str__(self):
  47         return self.text
  48
  49     def get_text(self):
  50         return Template(self.text).render(Context())
  51
  52     @classmethod
  53     def choice(cls, place, request, exemptions=True):
  54         Membership = apps.get_model('club', 'Membership')
  55
  56         if exemptions and hasattr(request, 'annoy_banner_exempt'):
  57             return cls.objects.none()
  58
  59         if settings.DEBUG:
  60             assert place in PLACES, f"Banner place `{place}` must be defined in annoy.places."
  61
  62         n = now()
  63         banners = cls.objects.filter(
  64             place=place
  65         ).exclude(
  66             since__gt=n
  67         ).exclude(
  68             until__lt=n
  69         ).order_by('-priority', '?')
  70
  71         if not request.user.is_authenticated:
  72             banners = banners.filter(only_authenticated=False)
  73
  74         if not request.user.is_staff:
  75             banners = banners.filter(staff_preview=False)
  76
  77         if Membership.is_active_for(request.user):
  78             banners = banners.filter(show_members=True)
  79
  80         return banners
  81
  82     @property
  83     def progress_percent(self):
  84         if not self.target:
  85             return 0
  86         return (self.progress or 0) / self.target * 100
  87
  88     @property
  89     def progress_percent_pretty(self):
  90         return int(self.progress_percent)
  91
  92     def update_progress(self):
  93         # Total of new payments during the action.
  94         # This definition will need to change for longer timespans.
  95         if not self.since or not self.until or not self.target:
  96             return
  97         Schedule = apps.get_model('club', 'Schedule')
  98         self.progress = Schedule.objects.filter(
  99             payed_at__gte=self.since,
 100             payed_at__lte=self.until,
 101         ).aggregate(c=models.Sum('amount'))['c']
 102         self.save(update_fields=['progress'])
 103
 104     @classmethod
 105     def update_all_progress(cls):
 106         for obj in cls.objects.exclude(target=None):
 107             obj.update_progress()
 108
 109
 110 class DynamicTextInsert(models.Model):
 111     paragraphs = models.IntegerField('akapity')
 112     url = models.CharField(max_length=1024)
 113
 114     class Meta:
 115         verbose_name = 'dynamiczna wstawka'
 116         verbose_name_plural = 'dynamiczne wstawki'
 117         ordering = ('paragraphs', )
 118
 119     def __str__(self):
 120         return str(self.paragraphs)
 121
 122     @classmethod
 123     def get_all(cls, request):
 124         Membership = apps.get_model('club', 'Membership')
 125         if Membership.is_active_for(request.user) and not request.user.is_staff:
 126             return cls.objects.none()
 127         return cls.objects.all()
 128
 129
 130     def choose(self):
 131         return self.dynamictextinserttext_set.order_by('?').first()
 132
 133
 134 class DynamicTextInsertText(models.Model):
 135     insert = models.ForeignKey(DynamicTextInsert, models.CASCADE)
 136     own_colors = models.BooleanField(default=False)
 137     background_color = models.CharField(max_length=10, blank=True)
 138     text_color = models.CharField(max_length=10, blank=True)
 139     text = models.TextField('tekst')
 140     image = models.FileField(blank=True, upload_to='annoy/inserts/')
 141
 142
 143 class MediaInsertSet(models.Model):
 144     file_format = models.CharField(max_length=8, choices=[
 145         ('epub', 'epub'),
 146         ('mobi', 'mobi'),
 147         ('pdf', 'pdf'),
 148         ])
 149     etag = models.CharField(max_length=64, blank=True)
 150
 151     def update_etag(self):
 152         self.etag = hashlib.sha1(json.dumps(self.get_texts()).encode('utf-8')).hexdigest()
 153         self.save(update_fields=['etag'])
 154
 155     def get_texts(self):
 156         return [t.text for t in self.mediainserttext_set.all()]
 157
 158     @classmethod
 159     def get_for_format(cls, file_format):
 160         return cls.objects.filter(file_format=file_format).first()
 161
 162     @classmethod
 163     def get_texts_for(cls, file_format):
 164         self = cls.get_for_format(file_format)
 165         if self is None:
 166             return []
 167         return self.get_texts()
 168
 169
 170 class MediaInsertText(models.Model):
 171     media_insert_set = models.ForeignKey(MediaInsertSet, models.CASCADE)
 172     ordering = models.IntegerField()
 173     text = models.TextField()
 174
 175     class Meta:
 176         ordering = ('ordering',)