nk20/apps/note/models/transactions.py

352 lines
10 KiB
Python
Raw Normal View History

2020-02-18 20:30:26 +00:00
# Copyright (C) 2018-2020 by BDE ENS Paris-Saclay
2019-07-16 10:43:23 +00:00
# SPDX-License-Identifier: GPL-3.0-or-later
2020-08-15 17:10:23 +00:00
2020-05-29 19:43:24 +00:00
from django.core.exceptions import ValidationError
from django.db import models, transaction
2020-03-07 21:28:59 +00:00
from django.urls import reverse
2019-07-16 10:43:23 +00:00
from django.utils import timezone
from django.utils.translation import gettext_lazy as _
2020-02-24 09:36:04 +00:00
from polymorphic.models import PolymorphicModel
2019-07-16 10:43:23 +00:00
2020-08-06 10:50:24 +00:00
from .notes import Note, NoteClub, NoteSpecial
from ..templatetags.pretty_money import pretty_money
2019-07-16 10:43:23 +00:00
"""
Defines transactions
"""
class TemplateCategory(models.Model):
2020-02-04 00:18:03 +00:00
"""
Defined a recurrent transaction category
Example: food, softs, ...
"""
name = models.CharField(
verbose_name=_("name"),
max_length=31,
unique=True,
)
class Meta:
verbose_name = _("transaction category")
verbose_name_plural = _("transaction categories")
def __str__(self):
return str(self.name)
2019-07-16 10:43:23 +00:00
2020-02-18 20:14:29 +00:00
2019-07-16 10:43:23 +00:00
class TransactionTemplate(models.Model):
"""
2020-02-04 00:18:03 +00:00
Defined a recurrent transaction
associated to selling something (a burger, a beer, ...)
"""
2019-07-16 10:43:23 +00:00
name = models.CharField(
verbose_name=_('name'),
max_length=255,
2019-08-13 16:23:15 +00:00
unique=True,
2020-03-07 21:28:59 +00:00
error_messages={'unique': _("A template with this name already exist")},
2019-07-16 10:43:23 +00:00
)
2020-04-01 18:56:24 +00:00
2019-07-16 10:43:23 +00:00
destination = models.ForeignKey(
2019-08-11 17:54:18 +00:00
NoteClub,
2019-07-16 10:43:23 +00:00
on_delete=models.PROTECT,
related_name='+', # no reverse
verbose_name=_('destination'),
)
2020-04-01 18:56:24 +00:00
2019-07-16 10:43:23 +00:00
amount = models.PositiveIntegerField(
verbose_name=_('amount'),
)
category = models.ForeignKey(
TemplateCategory,
2020-02-04 00:18:03 +00:00
on_delete=models.PROTECT,
2020-07-25 15:25:57 +00:00
related_name='templates',
2019-07-16 10:43:23 +00:00
verbose_name=_('type'),
2020-02-18 20:14:29 +00:00
max_length=31,
2019-07-16 10:43:23 +00:00
)
2020-04-01 18:56:24 +00:00
display = models.BooleanField(
2020-03-07 21:28:59 +00:00
default=True,
2020-04-01 18:56:24 +00:00
verbose_name=_("display"),
)
2020-04-01 18:56:24 +00:00
2020-07-25 15:25:57 +00:00
highlighted = models.BooleanField(
default=False,
verbose_name=_("highlighted"),
)
description = models.CharField(
verbose_name=_('description'),
max_length=255,
2020-03-14 18:00:20 +00:00
blank=True,
)
2019-07-16 11:50:05 +00:00
class Meta:
verbose_name = _("transaction template")
verbose_name_plural = _("transaction templates")
def get_absolute_url(self):
2020-03-07 21:28:59 +00:00
return reverse('note:template_update', args=(self.pk,))
2019-07-16 10:43:23 +00:00
2020-02-24 09:36:04 +00:00
class Transaction(PolymorphicModel):
"""
General transaction between two :model:`note.Note`
amount is store in centimes of currency, making it a positive integer
value. (from someone to someone else)
"""
2019-07-16 10:43:23 +00:00
source = models.ForeignKey(
Note,
on_delete=models.PROTECT,
related_name='+',
verbose_name=_('source'),
)
2020-03-26 13:45:48 +00:00
source_alias = models.CharField(
max_length=255,
default="", # Will be remplaced by the name of the note on save
verbose_name=_('used alias'),
)
2019-07-16 10:43:23 +00:00
destination = models.ForeignKey(
Note,
on_delete=models.PROTECT,
related_name='+',
verbose_name=_('destination'),
)
2020-03-26 13:45:48 +00:00
destination_alias = models.CharField(
max_length=255,
default="", # Will be remplaced by the name of the note on save
verbose_name=_('used alias'),
)
2019-07-17 10:48:03 +00:00
created_at = models.DateTimeField(
verbose_name=_('created at'),
2019-07-16 10:43:23 +00:00
default=timezone.now,
)
2019-07-19 08:12:46 +00:00
quantity = models.PositiveIntegerField(
2019-07-16 10:43:23 +00:00
verbose_name=_('quantity'),
2019-07-19 08:12:46 +00:00
default=1,
2019-07-16 10:43:23 +00:00
)
2020-03-16 11:11:16 +00:00
amount = models.PositiveIntegerField(
verbose_name=_('amount'),
)
2019-07-17 11:53:58 +00:00
reason = models.CharField(
verbose_name=_('reason'),
max_length=255,
2019-07-16 10:43:23 +00:00
)
2019-07-19 08:12:46 +00:00
valid = models.BooleanField(
2019-07-16 10:43:23 +00:00
verbose_name=_('valid'),
2019-07-19 08:12:46 +00:00
default=True,
2019-07-16 10:43:23 +00:00
)
invalidity_reason = models.CharField(
verbose_name=_('invalidity reason'),
max_length=255,
default=None,
null=True,
2020-04-01 01:42:19 +00:00
blank=True,
)
2019-07-16 11:50:05 +00:00
class Meta:
verbose_name = _("transaction")
verbose_name_plural = _("transactions")
indexes = [
models.Index(fields=['created_at']),
models.Index(fields=['source']),
models.Index(fields=['destination']),
]
2019-07-16 11:50:05 +00:00
2020-08-09 10:31:06 +00:00
def validate(self):
previous_source_balance = self.source.balance
previous_dest_balance = self.destination.balance
source_balance = self.source.balance
dest_balance = self.destination.balance
created = self.pk is None
to_transfer = self.amount * self.quantity
if not created:
# Revert old transaction
old_transaction = Transaction.objects.get(pk=self.pk)
# Check that nothing important changed
for field_name in ["source_id", "destination_id", "quantity", "amount"]:
if getattr(self, field_name) != getattr(old_transaction, field_name):
raise ValidationError(_("You can't update the {field} on a Transaction. "
"Please invalidate it and create one other.").format(field=field_name))
if old_transaction.valid == self.valid:
# Don't change anything
return 0, 0
if old_transaction.valid:
source_balance += to_transfer
dest_balance -= to_transfer
if self.valid:
source_balance -= to_transfer
dest_balance += to_transfer
# When a transaction is declared valid, we ensure that the invalidity reason is null, if it was
# previously invalid
self.invalidity_reason = None
2020-08-13 16:04:28 +00:00
if source_balance > 9223372036854775807 or source_balance < -9223372036854775808\
or dest_balance > 9223372036854775807 or dest_balance < -9223372036854775808:
raise ValidationError(_("The note balances must be between - 92 233 720 368 547 758.08 € "
"and 92 233 720 368 547 758.07 €."))
2020-08-09 10:31:06 +00:00
return source_balance - previous_source_balance, dest_balance - previous_dest_balance
@transaction.atomic
def save(self, *args, **kwargs):
2019-07-19 13:00:44 +00:00
"""
When saving, also transfer money between two notes
"""
if self.source.pk == self.destination.pk:
# When source == destination, no money is transferred and no transaction is created
return
# We refresh the notes with the "select for update" tag to avoid concurrency issues
self.source = Note.objects.filter(pk=self.source_id).select_for_update().get()
self.destination = Note.objects.filter(pk=self.destination_id).select_for_update().get()
# Check that the amounts stay between big integer bounds
diff_source, diff_dest = self.validate()
if not self.source.is_active or not self.destination.is_active:
2020-08-31 18:15:48 +00:00
raise ValidationError(_("The transaction can't be saved since the source note "
"or the destination note is not active."))
# If the aliases are not entered, we assume that the used alias is the name of the note
if not self.source_alias:
self.source_alias = str(self.source)
if not self.destination_alias:
self.destination_alias = str(self.destination)
# We save first the transaction, in case of the user has no right to transfer money
super().save(*args, **kwargs)
# Save notes
self.source.balance += diff_source
self.source._force_save = True
self.source.save()
self.destination.balance += diff_dest
self.destination._force_save = True
self.destination.save()
2019-07-19 13:00:44 +00:00
2019-08-15 21:08:15 +00:00
@property
def total(self):
2020-02-18 20:14:29 +00:00
return self.amount * self.quantity
2019-08-15 21:08:15 +00:00
2020-03-16 12:14:06 +00:00
@property
def type(self):
return _('Transfer')
def __str__(self):
return self.__class__.__name__ + " from " + str(self.source) + " to " + str(self.destination) + " of "\
+ pretty_money(self.quantity * self.amount) + ("" if self.valid else " invalid")
2019-07-16 10:43:23 +00:00
class RecurrentTransaction(Transaction):
"""
Special type of :model:`note.Transaction` associated to a :model:`note.TransactionTemplate`.
"""
template = models.ForeignKey(
TransactionTemplate,
on_delete=models.PROTECT,
)
def clean(self):
if self.template.destination != self.destination:
raise ValidationError(
_("The destination of this transaction must equal to the destination of the template."))
return super().clean()
def save(self, *args, **kwargs):
self.clean()
return super().save(*args, **kwargs)
2020-03-16 12:14:06 +00:00
@property
def type(self):
return _('Template')
class Meta:
verbose_name = _("recurrent transaction")
verbose_name_plural = _("recurrent transactions")
2020-03-07 21:28:59 +00:00
2020-03-14 14:13:58 +00:00
class SpecialTransaction(Transaction):
"""
Special type of :model:`note.Transaction` associated to transactions with special notes
"""
last_name = models.CharField(
max_length=255,
verbose_name=_("name"),
)
first_name = models.CharField(
max_length=255,
verbose_name=_("first_name"),
)
bank = models.CharField(
max_length=255,
2020-03-16 11:11:16 +00:00
verbose_name=_("bank"),
blank=True,
2020-03-14 14:13:58 +00:00
)
2020-03-16 12:14:06 +00:00
@property
def type(self):
return _('Credit') if isinstance(self.source, NoteSpecial) else _("Debit")
2020-04-18 13:59:06 +00:00
def is_credit(self):
return isinstance(self.source, NoteSpecial)
def is_debit(self):
return isinstance(self.destination, NoteSpecial)
def clean(self):
# SpecialTransaction are only possible with NoteSpecial object
if self.is_credit() == self.is_debit():
2020-05-29 19:43:24 +00:00
raise(ValidationError(_("A special transaction is only possible between a"
" Note associated to a payment method and a User or a Club")))
2020-04-18 13:59:06 +00:00
def save(self, *args, **kwargs):
self.clean()
super().save(*args, **kwargs)
class Meta:
verbose_name = _("Special transaction")
verbose_name_plural = _("Special transactions")
2020-03-14 14:13:58 +00:00
2019-07-16 10:43:23 +00:00
class MembershipTransaction(Transaction):
"""
Special type of :model:`note.Transaction` associated to a :model:`member.Membership`.
"""
2019-07-16 10:43:23 +00:00
membership = models.OneToOneField(
'member.Membership',
on_delete=models.PROTECT,
related_name='transaction',
)
2019-07-16 11:50:05 +00:00
class Meta:
verbose_name = _("membership transaction")
verbose_name_plural = _("membership transactions")
2020-03-16 12:14:06 +00:00
@property
def type(self):
return _('membership').capitalize()