26b0136c7962e75f9604094df79d8e62058797b6
[wolnelektury.git] / src / annoy / models.py
1 from django.apps import apps
2 from django.conf import settings
3 from django.db import models
4 from django.utils.translation import ugettext_lazy as _
5 from django.utils.timezone import now
6 from .places import PLACES, PLACE_CHOICES
7
8
9 class Banner(models.Model):
10     place = models.SlugField(choices=PLACE_CHOICES)
11     action_label = models.CharField(
12         max_length=255, blank=True,
13         help_text=_('')
14     )
15     open_label = models.CharField(max_length=255, blank=True)
16     close_label = models.CharField(max_length=255, blank=True)
17     text = models.TextField()
18     url = models.CharField(max_length=1024)
19     priority = models.PositiveSmallIntegerField(default=0)
20     since = models.DateTimeField(null=True, blank=True)
21     until = models.DateTimeField(null=True, blank=True)
22     show_members = models.BooleanField(default=False)
23     staff_preview = models.BooleanField(default=False)
24
25     class Meta:
26         verbose_name = _('banner')
27         verbose_name_plural = _('banners')
28         ordering = ('place', '-priority',)
29
30     def __str__(self):
31         return self.text
32
33     @classmethod
34     def choice(cls, place, request):
35         Membership = apps.get_model('club', 'Membership')
36
37         if hasattr(request, 'annoy_banner_exempt'):
38             return cls.objects.none()
39         
40         if settings.DEBUG:
41             assert place in PLACES, "Banner place `{}` must be defined in annoy.places.".format(place)
42
43         n = now()
44         banners = cls.objects.filter(
45             place=place
46         ).exclude(
47             since__gt=n
48         ).exclude(
49             until__lt=n
50         ).order_by('-priority', '?')
51
52         if not request.user.is_staff:
53             banners = banners.filter(staff_preview=False)
54
55         if request:
56             if Membership.is_active_for(request.user):
57                 banners = banners.filter(show_members=True)
58         return banners
59         
60         
61 class DynamicTextInsert(models.Model):
62     paragraphs = models.IntegerField(_('pararaphs'))
63     url = models.CharField(max_length=1024)
64
65     class Meta:
66         verbose_name = _('dynamic insert')
67         verbose_name_plural = _('dynamic inserts')
68         ordering = ('paragraphs', )
69
70     def __str__(self):
71         return str(self.paragraphs)
72
73     def choose(self):
74         return self.dynamictextinserttext_set.order_by('?').first()
75
76
77 class DynamicTextInsertText(models.Model):
78     insert = models.ForeignKey(DynamicTextInsert, models.CASCADE)
79     background_color = models.CharField(max_length=10, blank=True)
80     text_color = models.CharField(max_length=10, blank=True)
81     text = models.TextField(_('text'))
82     image = models.FileField(blank=True)