Fixing widths

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

diff --git a/apps/funding/models.py b/apps/funding/models.py
index de90226..ae034f0 100644 (file)
--- a/apps/funding/models.py
+++ b/apps/funding/models.py
@@ -17,8 +17,8 @@ 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)
@@ -60,11 +60,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.
@@ -119,7 +125,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?
@@ -166,6 +172,11 @@ 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':