a088a07b7b737098e50c675a484bfb5d8fe724e4
[wolnelektury.git] / apps / funding / models.py
1 # -*- coding: utf-8 -*-
2 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
3 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
4 #
5 from django.core.urlresolvers import reverse
6 from django.db import models
7 from django.utils.translation import ugettext_lazy as _, ugettext as __
8 from datetime import date, datetime
9
10
11 class Offer(models.Model):
12     author = models.CharField(_('author'), max_length=255)
13     title = models.CharField(_('title'), max_length=255)
14     slug = models.SlugField(_('slug'))
15     book_url = models.URLField(_('book URL'), blank=True)
16     redakcja_url = models.URLField(_('redakcja URL'), blank=True)
17     target = models.DecimalField(_('target'), decimal_places=2, max_digits=10)
18     start = models.DateField(_('start'))
19     end = models.DateField(_('end'))
20
21     class Meta:
22         verbose_name = _('offer')
23         verbose_name_plural = _('offers')
24         ordering = ['-end']
25
26     def __unicode__(self):
27         return u"%s – %s" % (self.author, self.title)
28
29     def get_absolute_url(self):
30         return reverse('funding_offer', args=[self.slug])
31
32     @classmethod
33     def current(cls):
34         today = date.today()
35         objects = cls.objects.filter(start__lte=today, end__gte=today)
36         try:
37             return objects[0]
38         except IndexError:
39             return None
40
41     @classmethod
42     def public(cls):
43         today = date.today()
44         return cls.objects.filter(start__lte=today)        
45
46     def get_perks(self, amount=None):
47         perks = Perk.objects.filter(
48                 models.Q(offer=self) | models.Q(offer=None)
49             )
50         if amount is not None:
51             perks = perks.filter(price__lte=amount)
52         return perks
53
54     def fund(self, name, email, amount):
55         funding = self.funding_set.create(
56             name=name, email=email, amount=amount,
57             payed_at=datetime.now())
58         funding.perks = self.get_perks(amount)
59         return funding
60
61     def sum(self):
62         return self.funding_set.aggregate(s=models.Sum('amount'))['s'] or 0
63
64     def state(self):
65         if self.sum() >= self.target:
66             return 'win'
67         elif self.start <= date.today() <= self.end:
68             return 'running'
69         else:
70             return 'lose'
71
72
73 class Perk(models.Model):
74     offer = models.ForeignKey(Offer, verbose_name=_('offer'), null=True, blank=True)
75     price = models.DecimalField(_('price'), decimal_places=2, max_digits=10)
76     name = models.CharField(_('name'), max_length=255)
77     description = models.TextField(_('description'), blank=True)
78
79     class Meta:
80         verbose_name = _('perk')
81         verbose_name_plural = _('perks')
82         ordering = ['-price']
83
84     def __unicode__(self):
85         return "%s (%s%s)" % (self.name, self.price, u" for %s" % self.offer if self.offer else "")
86
87
88 class Funding(models.Model):
89     offer = models.ForeignKey(Offer, verbose_name=_('offer'))
90     name = models.CharField(_('name'), max_length=127)
91     email = models.EmailField(_('email'))
92     amount = models.DecimalField(_('amount'), decimal_places=2, max_digits=10)
93     payed_at = models.DateTimeField(_('payed_at'))
94     perks = models.ManyToManyField(Perk, verbose_name=_('perks'), blank=True)
95
96     class Meta:
97         verbose_name = _('funding')
98         verbose_name_plural = _('fundings')
99         ordering = ['-payed_at']
100
101     def __unicode__(self):
102         return "%s payed %s for %s" % (self.name, self.amount, self.offer)