mirror of https://gitlab.crans.org/bde/nk20
167 lines
4.0 KiB
Python
167 lines
4.0 KiB
Python
# -*- mode: python; coding: utf-8 -*-
|
|
# Copyright (C) 2018-2019 by BDE ENS Paris-Saclay
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
from django.conf import settings
|
|
from django.db import models
|
|
from django.utils.translation import gettext_lazy as _
|
|
from polymorphic.models import PolymorphicModel
|
|
|
|
"""
|
|
Defines each note types
|
|
"""
|
|
|
|
|
|
class Note(PolymorphicModel):
|
|
"""
|
|
An model, use to add transactions capabilities
|
|
"""
|
|
balance = models.IntegerField(
|
|
verbose_name=_('account balance'),
|
|
help_text=_('in centimes, money credited for this instance'),
|
|
default=0,
|
|
)
|
|
is_active = models.BooleanField(
|
|
_('active'),
|
|
default=True,
|
|
help_text=_(
|
|
'Designates whether this note should be treated as active. '
|
|
'Unselect this instead of deleting notes.'
|
|
),
|
|
)
|
|
display_image = models.ImageField(
|
|
verbose_name=_('display image'),
|
|
max_length=255,
|
|
blank=True,
|
|
)
|
|
created_at = models.DateTimeField(
|
|
verbose_name=_('created at'),
|
|
auto_now_add=True,
|
|
)
|
|
|
|
class Meta:
|
|
verbose_name = _("note")
|
|
verbose_name_plural = _("notes")
|
|
|
|
|
|
class NoteUser(Note):
|
|
"""
|
|
A Note associated to an User
|
|
"""
|
|
user = models.OneToOneField(
|
|
settings.AUTH_USER_MODEL,
|
|
on_delete=models.PROTECT,
|
|
related_name='note',
|
|
verbose_name=_('user'),
|
|
)
|
|
|
|
class Meta:
|
|
verbose_name = _("one's note")
|
|
verbose_name_plural = _("users note")
|
|
|
|
def __str__(self):
|
|
return _("%(user)s's note") % {'user': str(self.user)}
|
|
|
|
def save(self, *args, **kwargs):
|
|
"""
|
|
When saving, also create an alias
|
|
TODO: check availability of alias
|
|
TODO: remove old alias
|
|
"""
|
|
created = self.pk is None
|
|
if created:
|
|
alias = Alias.objects.create(name=str(self.user), note=self)
|
|
alias.save()
|
|
|
|
super().save(*args, **kwargs)
|
|
|
|
|
|
class NoteClub(Note):
|
|
"""
|
|
A Note associated to a Club
|
|
"""
|
|
club = models.OneToOneField(
|
|
'member.Club',
|
|
on_delete=models.PROTECT,
|
|
related_name='note',
|
|
verbose_name=_('club'),
|
|
)
|
|
|
|
class Meta:
|
|
verbose_name = _("club note")
|
|
verbose_name_plural = _("clubs notes")
|
|
|
|
def __str__(self):
|
|
return _("Note for %(club)s club") % {'club': str(self.club)}
|
|
|
|
def save(self, *args, **kwargs):
|
|
"""
|
|
When saving, also create an alias
|
|
TODO: check availability of alias
|
|
TODO: remove old alias
|
|
"""
|
|
created = self.pk is None
|
|
if created:
|
|
alias = Alias.objects.create(name=str(self.club), note=self)
|
|
alias.save()
|
|
|
|
super().save(*args, **kwargs)
|
|
|
|
|
|
class NoteSpecial(Note):
|
|
"""
|
|
A Note for special account, where real money enter or leave the system
|
|
- bank check
|
|
- credit card
|
|
- bank transfer
|
|
- cash
|
|
- refund
|
|
"""
|
|
special_type = models.CharField(
|
|
verbose_name=_('type'),
|
|
max_length=255,
|
|
unique=True,
|
|
)
|
|
|
|
class Meta:
|
|
verbose_name = _("special note")
|
|
verbose_name_plural = _("special notes")
|
|
|
|
def __str__(self):
|
|
return self.special_type
|
|
|
|
def save(self, *args, **kwargs):
|
|
"""
|
|
When saving, also create an alias
|
|
TODO: check availability of alias
|
|
TODO: remove old alias
|
|
"""
|
|
created = self.pk is None
|
|
if created:
|
|
alias = Alias.objects.create(name=str(self.club), note=self)
|
|
alias.save()
|
|
|
|
super().save(*args, **kwargs)
|
|
|
|
|
|
class Alias(models.Model):
|
|
"""
|
|
An alias labels a Note instance, only for user and clubs
|
|
"""
|
|
name = models.CharField(
|
|
verbose_name=_('name'),
|
|
max_length=255,
|
|
unique=True,
|
|
)
|
|
note = models.ForeignKey(
|
|
Note,
|
|
on_delete=models.PROTECT,
|
|
)
|
|
|
|
class Meta:
|
|
verbose_name = _("alias")
|
|
verbose_name_plural = _("aliases")
|
|
|
|
def __str__(self):
|
|
return self.name
|