X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/dac9a35e0d86c0ec50f33d69d73be5d7ae7ae7ef..f9ebfca6f51df543810066816ce90b79932cd458:/apps/funding/models.py

diff --git a/apps/funding/models.py b/apps/funding/models.py
index d193b41c4..162d33739 100644
--- a/apps/funding/models.py
+++ b/apps/funding/models.py
@@ -8,6 +8,7 @@ from django.db import models
 from django.utils.translation import ugettext_lazy as _, ugettext as __
 import getpaid
 from catalogue.models import Book
+from polls.models import Poll
 
 
 class Offer(models.Model):
@@ -17,14 +18,21 @@ class Offer(models.Model):
     slug = models.SlugField(_('slug'))
     description = models.TextField(_('description'), blank=True)
     target = models.DecimalField(_('target'), decimal_places=2, max_digits=10)
-    start = models.DateField(_('start'))
-    end = models.DateField(_('end'))
+    start = models.DateField(_('start'), db_index=True)
+    end = models.DateField(_('end'), db_index=True)
     due = models.DateField(_('due'),
         help_text=_('When will it be published if the money is raised.'))
     redakcja_url = models.URLField(_('redakcja URL'), blank=True)
     book = models.ForeignKey(Book, null=True, blank=True,
         help_text=_('Published book.'))
-
+    cover = models.ImageField(_('Cover'), upload_to = 'funding/covers')
+    poll = models.ForeignKey(Poll, help_text = _('Poll'),  null = True, on_delete = models.SET_NULL)
+        
+    def cover_img_tag(self):
+        return u'<img src="%s" />' % self.cover.url
+    cover_img_tag.short_description = _('Cover preview')
+    cover_img_tag.allow_tags = True
+        
     class Meta:
         verbose_name = _('offer')
         verbose_name_plural = _('offers')
@@ -125,7 +133,7 @@ class Funding(models.Model):
     name = models.CharField(_('name'), max_length=127, blank=True)
     email = models.EmailField(_('email'), blank=True)
     amount = models.DecimalField(_('amount'), decimal_places=2, max_digits=10)
-    payed_at = models.DateTimeField(_('payed at'), null=True, blank=True)
+    payed_at = models.DateTimeField(_('payed at'), null=True, blank=True, db_index=True)
     perks = models.ManyToManyField(Perk, verbose_name=_('perks'), blank=True)
 
     # Any additional info needed for perks?