X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/ea656434007c3ec64f0e583c65e177b08403c865..f9ebfca6f51df543810066816ce90b79932cd458:/apps/funding/models.py?ds=inline diff --git a/apps/funding/models.py b/apps/funding/models.py index 10243132c..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): @@ -15,22 +16,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, on_delete = models.SET_NULL) + + def cover_img_tag(self): + return u'' % 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 +47,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 +68,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 +88,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 +101,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. @@ -94,6 +112,7 @@ class Perk(models.Model): price = models.DecimalField(_('price'), decimal_places=2, max_digits=10) name = models.CharField(_('name'), max_length=255) description = models.TextField(_('description'), blank=True) + end_date = models.DateField(_('end date'), null=True, blank=True) class Meta: verbose_name = _('perk') @@ -114,9 +133,10 @@ 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')) + + # Any additional info needed for perks? @classmethod def payed(cls): @@ -129,7 +149,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 +160,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,6 +180,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':