Cites: sticky and custom backgrounds.

[wolnelektury.git] / apps / social / models.py

diff --git a/apps/social/models.py b/apps/social/models.py
index 225477e..e1c7e37 100644 (file)
--- a/apps/social/models.py
+++ b/apps/social/models.py
@@ -4,15 +4,40 @@
 #
 from django.db import models
 from django.utils.translation import ugettext_lazy as _
+from django.core.urlresolvers import reverse
 
 from catalogue.models import Book
 
 
 class Cite(models.Model):
-    book = models.ForeignKey(Book)
+    book = models.ForeignKey(Book, verbose_name=_('book'))
     text = models.TextField(_('text'))
+    small = models.BooleanField(_('small'), default=False,
+        help_text=_('Make this cite display smaller.'))
     vip = models.CharField(_('VIP'), max_length=128, null=True, blank=True)
     link = models.URLField(_('link'))
+    sticky = models.BooleanField(_('sticky'), default=False, db_index=True,
+        help_text=_('Sticky cites will take precedense.'))
+
+    image = models.ImageField(_('image'), upload_to='social/cite',
+                null=True, blank=True)
+    image_title = models.CharField(_('title'), max_length=255,
+                null=True, blank=True)
+    image_author = models.CharField(_('author'),
+                max_length=255, blank=True, null=True)
+    image_link = models.URLField(_('link'), blank=True, null=True)
+    image_license = models.CharField(_('license name'),
+                max_length=255, blank=True, null=True)
+    image_license_link = models.URLField(_('license link'), blank=True, null=True)
+
+    class Meta:
+        ordering = ('vip', 'text')
+        verbose_name = _('cite')
+        verbose_name_plural = _('cites')
+
+    def __unicode__(self):
+        return u"%s: %s…" % (self.vip, self.text[:60])
 
     def get_absolute_url(self):
-        return self.link
+        """This is used for testing."""
+        return "%s?choose_cite=%d" % (reverse('main_page'), self.id)