Introduce banner styles.

[wolnelektury.git] / src / annoy / models.py

diff --git a/src/annoy/models.py b/src/annoy/models.py
index 9ed724c..4df3a5e 100644 (file)
--- a/src/annoy/models.py
+++ b/src/annoy/models.py
@@ -1,13 +1,19 @@
 from django.apps import apps
 from django.conf import settings
 from django.db import models
+from django.template import Context, Template
 from django.utils.translation import ugettext_lazy as _
 from django.utils.timezone import now
-from .places import PLACES, PLACE_CHOICES
+from .places import PLACES, PLACE_CHOICES, STYLES
 
 
 class Banner(models.Model):
     place = models.SlugField(_('place'), choices=PLACE_CHOICES)
+    style = models.CharField(
+        _('style'), max_length=255, blank=True,
+        choices=STYLES,
+        help_text=_('Affects blackout.')
+    )
     action_label = models.CharField(
         _('action label'),
         max_length=255, blank=True,
@@ -34,6 +40,9 @@ class Banner(models.Model):
     def __str__(self):
         return self.text
 
+    def get_text(self):
+        return Template(self.text).render(Context())
+
     @classmethod
     def choice(cls, place, request):
         Membership = apps.get_model('club', 'Membership')