61cb94cad87e2c6a67263e525472ab0d29a7d25a
[wolnelektury.git] / 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.translation import gettext_lazy as _
8 from django.utils.timezone import now
9 from .places import PLACES, PLACE_CHOICES, STYLES
10
11
12 class Banner(models.Model):
13     place = models.SlugField(_('place'), choices=PLACE_CHOICES)
14     style = models.CharField(
15         _('style'), max_length=255, blank=True,
16         choices=STYLES,
17         help_text=_('Affects blackout.')
18     )
19     smallfont = models.BooleanField(_('small font'), default=False)
20     text_color = models.CharField(max_length=10, blank=True)
21     background_color = models.CharField(max_length=10, blank=True)
22     action_label = models.CharField(
23         _('action label'),
24         max_length=255, blank=True,
25         help_text=_('If empty, whole banner will serve as a link')
26     )
27     open_label = models.CharField(_('open label'), max_length=255, blank=True)
28     close_label = models.CharField(_('close label'), max_length=255, blank=True)
29     text = models.TextField(_('text'))
30     image = models.FileField(_('image'), upload_to='annoy/banners/', blank=True)
31     url = models.CharField(_('url'), max_length=1024)
32     priority = models.PositiveSmallIntegerField(
33         _('priority'), default=0,
34         help_text=_('Banners with higher priority come first.'))
35     since = models.DateTimeField(_('since'), null=True, blank=True)
36     until = models.DateTimeField(_('until'), null=True, blank=True)
37     show_members = models.BooleanField(_('show members'), default=False)
38     staff_preview = models.BooleanField(_('staff preview'), default=False)
39     only_authenticated = models.BooleanField(_('only for authenticated users'), default=False)
40
41     class Meta:
42         verbose_name = _('banner')
43         verbose_name_plural = _('banners')
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):
54         Membership = apps.get_model('club', 'Membership')
55
56         if 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
83 class DynamicTextInsert(models.Model):
84     paragraphs = models.IntegerField(_('pararaphs'))
85     url = models.CharField(max_length=1024)
86
87     class Meta:
88         verbose_name = _('dynamic insert')
89         verbose_name_plural = _('dynamic inserts')
90         ordering = ('paragraphs', )
91
92     def __str__(self):
93         return str(self.paragraphs)
94
95     @classmethod
96     def get_all(cls, request):
97         Membership = apps.get_model('club', 'Membership')
98         if Membership.is_active_for(request.user) and not request.user.is_staff:
99             return cls.objects.none()
100         return cls.objects.all()
101
102
103     def choose(self):
104         return self.dynamictextinserttext_set.order_by('?').first()
105
106
107 class DynamicTextInsertText(models.Model):
108     insert = models.ForeignKey(DynamicTextInsert, models.CASCADE)
109     own_colors = models.BooleanField(default=False)
110     background_color = models.CharField(max_length=10, blank=True)
111     text_color = models.CharField(max_length=10, blank=True)
112     text = models.TextField(_('text'))
113     image = models.FileField(blank=True, upload_to='annoy/inserts/')
114
115
116 class MediaInsertSet(models.Model):
117     file_format = models.CharField(max_length=8, choices=[
118         ('epub', 'epub'),
119         ('mobi', 'mobi'),
120         ])
121     etag = models.CharField(max_length=64, blank=True)
122
123     def update_etag(self):
124         self.etag = hashlib.sha1(json.dumps(self.get_texts()).encode('utf-8')).hexdigest()
125         self.save(update_fields=['etag'])
126
127     def get_texts(self):
128         return [t.text for t in self.mediainserttext_set.all()]
129
130     @classmethod
131     def get_for_format(cls, file_format):
132         return cls.objects.filter(file_format=file_format).first()
133
134     @classmethod
135     def get_texts_for(cls, file_format):
136         self = cls.get_for_format(file_format)
137         if self is None:
138             return []
139         return self.get_texts()
140
141
142 class MediaInsertText(models.Model):
143     media_insert_set = models.ForeignKey(MediaInsertSet, models.CASCADE)
144     ordering = models.IntegerField()
145     text = models.TextField()
146
147     class Meta:
148         ordering = ('ordering',)
149
150
151 from django.db.models.signals import post_save, post_delete
152 from django.dispatch import receiver
153
154 @receiver(post_delete, sender=MediaInsertText)
155 @receiver(post_save, sender=MediaInsertText)
156 def update_etag(sender, instance, **kwargs):
157     instance.media_insert_set.update_etag()