typo fix
[wolnelektury.git] / apps / funding / models.py
index 1024313..8c3312e 100644 (file)
@@ -4,10 +4,14 @@
 #
 from datetime import date, datetime
 from django.core.urlresolvers import reverse
+from django.core.mail import send_mail
+from django.conf import settings
+from django.template.loader import render_to_string
 from django.db import models
-from django.utils.translation import ugettext_lazy as _, ugettext as __
+from django.utils.translation import ugettext_lazy as _, ugettext as __, override
 import getpaid
 from catalogue.models import Book
+from polls.models import Poll
 
 
 class Offer(models.Model):
@@ -15,22 +19,30 @@ class Offer(models.Model):
     author = models.CharField(_('author'), max_length=255)
     title = models.CharField(_('title'), max_length=255)
     slug = models.SlugField(_('slug'))
-    book = models.ForeignKey(Book, null=True, blank=True,
-        help_text=_('Published book.'))
-    redakcja_url = models.URLField(_('redakcja URL'), blank=True)
+    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, blank = 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')
         ordering = ['-end']
 
     def __unicode__(self):
-        return u"%s  %s" % (self.author, self.title)
+        return u"%s - %s" % (self.author, self.title)
 
     def get_absolute_url(self):
         return reverse('funding_offer', args=[self.slug])
@@ -38,6 +50,17 @@ class Offer(models.Model):
     def is_current(self):
         return self.start <= date.today() <= self.end
 
+    def is_win(self):
+        return self.sum() >= self.target
+
+    def remaining(self):
+        if self.is_current():
+            return None
+        if self.is_win():
+            return self.sum() - self.target
+        else:
+            return self.sum()
+
     @classmethod
     def current(cls):
         """ Returns current fundraiser or None. """
@@ -48,11 +71,17 @@ class Offer(models.Model):
         except IndexError:
             return None
 
+    @classmethod
+    def past(cls):
+        """ QuerySet for all current and past fundraisers. """
+        today = date.today()
+        return cls.objects.filter(end__lt=today)
+
     @classmethod
     def public(cls):
         """ QuerySet for all current and past fundraisers. """
         today = date.today()
-        return cls.objects.filter(start__lte=today)        
+        return cls.objects.filter(start__lte=today)
 
     def get_perks(self, amount=None):
         """ Finds all the perks for the offer.
@@ -62,7 +91,7 @@ class Offer(models.Model):
         """
         perks = Perk.objects.filter(
                 models.Q(offer=self) | models.Q(offer=None)
-            )
+            ).exclude(end_date__lt=date.today())
         if amount is not None:
             perks = perks.filter(price__lte=amount)
         return perks
@@ -75,14 +104,6 @@ class Offer(models.Model):
         """ The money gathered. """
         return self.funding_payed().aggregate(s=models.Sum('amount'))['s'] or 0
 
-    def state(self):
-        if self.sum() >= self.target:
-            return 'win'
-        elif self.start <= date.today() <= self.end:
-            return 'running'
-        else:
-            return 'lose'
-
 
 class Perk(models.Model):
     """ A perk offer.
@@ -93,7 +114,8 @@ class Perk(models.Model):
     offer = models.ForeignKey(Offer, verbose_name=_('offer'), null=True, blank=True)
     price = models.DecimalField(_('price'), decimal_places=2, max_digits=10)
     name = models.CharField(_('name'), max_length=255)
-    description = models.TextField(_('description'), blank=True)
+    long_name = models.CharField(_('long name'), max_length=255)
+    end_date = models.DateField(_('end date'), null=True, blank=True)
 
     class Meta:
         verbose_name = _('perk')
@@ -114,9 +136,11 @@ 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)
-    anonymous = models.BooleanField(_('anonymous'))
+    language_code = models.CharField(max_length = 2, null = True, blank = True)
+
+    # Any additional info needed for perks?
 
     @classmethod
     def payed(cls):
@@ -129,7 +153,7 @@ class Funding(models.Model):
         ordering = ['-payed_at']
 
     def __unicode__(self):
-        return "%s payed %s for %s" % (self.name, self.amount, self.offer)
+        return unicode(self.offer)
 
     def get_absolute_url(self):
         return reverse('funding_funding', args=[self.pk])
@@ -140,9 +164,9 @@ getpaid.register_to_payment(Funding, unique=False, related_name='payment')
 
 class Spent(models.Model):
     """ Some of the remaining money spent on a book. """
+    book = models.ForeignKey(Book)
     amount = models.DecimalField(_('amount'), decimal_places=2, max_digits=10)
     timestamp = models.DateField(_('when'))
-    book = models.ForeignKey(Book)
 
     class Meta:
         verbose_name = _('money spent on a book')
@@ -160,9 +184,26 @@ def new_payment_query_listener(sender, order=None, payment=None, **kwargs):
 getpaid.signals.new_payment_query.connect(new_payment_query_listener)
 
 
+def user_data_query_listener(sender, order, user_data, **kwargs):
+    """ Set user data for payment. """
+    user_data['email'] = order.email
+getpaid.signals.user_data_query.connect(user_data_query_listener)
+
 def payment_status_changed_listener(sender, instance, old_status, new_status, **kwargs):
     """ React to status changes from getpaid. """
     if old_status != 'paid' and new_status == 'paid':
         instance.order.payed_at = datetime.now()
         instance.order.save()
+        if instance.order.email:
+            send_thank_you_email(instance.order.name, instance.order.email, instance.order.language_code)
 getpaid.signals.payment_status_changed.connect(payment_status_changed_listener)
+
+def send_thank_you_email(name, address, language_code):
+    with override(language_code or 'pl'):
+        send_mail(_('Thank you for your support!'), 
+                render_to_string('funding/email.txt', dict(name = name)),
+                getattr(settings, 'CONTACT_EMAIL', 'wolnelektury@nowoczesnapolska.org.pl'),
+                [address],
+                fail_silently=False
+                )