Upgrades.
[wolnelektury.git] / src / pz / models.py
1 from django.db import models
2 from django.utils.timezone import now
3 from django.utils.translation import ugettext_lazy as _
4 from .bank import parse_export_feedback
5
6
7 class Campaign(models.Model):
8     name = models.CharField(_('name'), max_length=255, unique=True)
9     description = models.TextField(_('description'), blank=True)
10
11     class Meta:
12         verbose_name = _('campaign')
13         verbose_name_plural = _('campaigns')
14
15     def __str__(self):
16         return self.name
17
18
19 class Fundraiser(models.Model):
20     name = models.CharField(_('name'), max_length=255, unique=True)
21
22     class Meta:
23         verbose_name = _('fundraiser')
24         verbose_name_plural = _('fundraisers')
25
26     def __str__(self):
27         return self.name
28
29
30 class DirectDebit(models.Model):
31     first_name = models.CharField(_('first name'), max_length=255, blank=True)
32     last_name = models.CharField(_('last name'), max_length=255, blank=True)
33     sex = models.CharField(_('sex'), max_length=1, blank=True, choices=[
34         ('M', _('M')),
35         ('F', _('F')),
36     ])
37     date_of_birth = models.DateField(_('date of birth'), null=True, blank=True)
38     street = models.CharField(_('street'), max_length=255, blank=True)
39     building = models.CharField(_('building'), max_length=255, blank=True)
40     flat = models.CharField(_('flat'), max_length=255, blank=True)
41     town = models.CharField(_('town'), max_length=255, blank=True)
42     postal_code = models.CharField(_('postal code'),  max_length=255, blank=True)
43     phone = models.CharField(_('phone'), max_length=255, blank=True)
44     email = models.CharField(_('e-mail'), max_length=255, blank=True)
45     iban = models.CharField(_('IBAN'), max_length=255, blank=True)
46     iban_valid = models.NullBooleanField(_('IBAN valid'), default=False)
47     is_consumer = models.BooleanField(_('is a consumer'), default=True)
48     payment_id = models.CharField(_('payment identifier'), max_length=255, blank=True, unique=True)
49     agree_fundraising = models.BooleanField(_('agree fundraising'), default=False)
50     agree_newsletter = models.BooleanField(_('agree newsletter'), default=False)
51
52     acquisition_date = models.DateField(_('acquisition date'), help_text=_('Date from the form'), null=True, blank=True)
53     submission_date = models.DateField(_('submission date'), null=True, blank=True, help_text=_('Date the fundaiser submitted the form'))
54     bank_submission_date = models.DateField(_('bank submission date'), null=True, blank=True, help_text=_('Date when the form data is submitted to the bank'))
55     bank_acceptance_date = models.DateField(_('bank accepted date'), null=True, blank=True, help_text=_('Date when bank accepted the form'))
56
57     fundraiser = models.ForeignKey(Fundraiser, models.PROTECT, blank=True, null=True, verbose_name=_('fundraiser'))
58     fundraiser_commission = models.IntegerField(_('fundraiser commission'), null=True, blank=True)
59     fundraiser_bill = models.CharField(_('fundaiser bill number'), max_length=255, blank=True)
60
61     amount = models.IntegerField(_('amount'), null=True, blank=True)
62
63     notes = models.TextField(_('notes'), blank=True)
64
65     needs_redo = models.BooleanField(_('needs redo'), default=False)
66     cancelled_at = models.DateTimeField(_('cancelled at'), null=True, blank=True)
67     optout = models.BooleanField(_('optout'), default=False)
68
69     campaign = models.ForeignKey(Campaign, models.PROTECT, null=True, blank=True, verbose_name=_('campaign'))
70
71     class Meta:
72         verbose_name = _('direct debit')
73         verbose_name_plural = _('direct debits')
74
75     def __str__(self):
76         return self.payment_id
77
78     def save(self, **kwargs):
79         self.iban_valid = not self.iban_warning() if self.iban else None
80         super().save(**kwargs)
81
82     @classmethod
83     def get_next_payment_id(cls):
84         # Find the last object added.
85         last = cls.objects.order_by('-id').first()
86         if last is None:
87             return ''
88         match = re.match(r'^(.*?)(\d+)$', last.payment_id)
89         if match is None:
90             return ''
91         prefix = match.group(1)
92         number = int(match.group(2))
93         number_length = len(match.group(2))
94         while True:
95             number += 1
96             payment_id = f'{prefix}{number:0{number_length}}'
97             if not cls.objects.filter(payment_id=payment_id).exists():
98                 break
99         return payment_id
100
101     @property
102     def full_name(self):
103         return ' '.join((self.first_name, self.last_name)).strip()
104
105     @property
106     def street_address(self):
107         street_addr = self.street
108         if self.building:
109             street_addr += ' ' + self.building
110         if self.flat:
111             street_addr += ' m. ' + self.flat
112         street_addr = street_addr.strip()
113         return street_addr
114
115     def iban_warning(self):
116         if not self.iban:
117             return 'No IBAN'
118         if len(self.iban) != 26:
119             return 'Bad IBAN length'
120         if int(self.iban[2:] + '2521' + self.iban[:2]) % 97 != 1:
121             return 'This IBAN number looks invalid'
122         return ''
123     iban_warning.short_description = ''
124
125
126 class BankExportFeedback(models.Model):
127     created_at = models.DateTimeField(auto_now_add=True)
128     csv = models.FileField(upload_to='pz/feedback/')
129
130     def save(self, **kwargs):
131         super().save(**kwargs)
132         for payment_id, status, comment in parse_export_feedback(self.csv):
133             debit = DirectDebit.objects.get(payment_id = payment_id)
134             b, created = self.bankexportfeedbackline_set.get_or_create(
135                 debit=debit,
136                 defaults={
137                     "status": status,
138                     "comment": comment,
139                 }
140             )
141             if not created:
142                 b.status = status
143                 b.comment = comment
144                 b.save()
145             if status == 1 and not debit.bank_acceptance_date:
146                 debit.bank_acceptance_date = now().date()
147                 debit.save()
148
149
150 class BankExportFeedbackLine(models.Model):
151     feedback = models.ForeignKey(BankExportFeedback, models.CASCADE)
152     debit = models.ForeignKey(DirectDebit, models.CASCADE)
153     status = models.SmallIntegerField()
154     comment = models.CharField(max_length=255)
155
156
157
158 class BankOrder(models.Model):
159     payment_date = models.DateField(null=True, blank=True)
160     sent = models.DateTimeField(null=True, blank=True)
161     debits = models.ManyToManyField(DirectDebit, blank=True)