Merge branch 'master' of https://github.com/fnp/wolnelektury

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

diff --git a/apps/social/models.py b/apps/social/models.py
index 225477e..7539805 100644 (file)
--- a/apps/social/models.py
+++ b/apps/social/models.py
@@ -4,6 +4,7 @@
 #
 from django.db import models
 from django.utils.translation import ugettext_lazy as _
+from django.core.urlresolvers import reverse
 
 from catalogue.models import Book
 
@@ -11,8 +12,16 @@ from catalogue.models import Book
 class Cite(models.Model):
     book = models.ForeignKey(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'))
 
+    class Meta:
+        ordering = ('vip', 'text')
+
+    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)