re2o/cotisations/models.py

# -*- mode: python; coding: utf-8 -*-
# Re2o est un logiciel d'administration développé initiallement au Rézo Metz. Il
# se veut agnostique au réseau considéré, de manière à être installable en
# quelques clics.
#
# Copyright © 2017  Gabriel Détraz
# Copyright © 2017  Lara Kermarec
# Copyright © 2017  Augustin Lemesle
# Copyright © 2018  Hugo Levy-Falk
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along
# with this program; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
"""
The database models for the 'cotisation' app of re2o.
The goal is to keep the main actions here, i.e. the 'clean' and 'save'
function are higly reposnsible for the changes, checking the coherence of the
data and the good behaviour in general for not breaking the database.

For further details on each of those models, see the documentation details for
each.
"""

from __future__ import unicode_literals

from dateutil.relativedelta import relativedelta
from django.contrib import messages
from django.core.validators import MinValueValidator
from django.db import models
from django.db.models import Max, Q
from django.db.models.signals import post_delete, post_save
from django.dispatch import receiver
from django.forms import ValidationError
from django.shortcuts import redirect
from django.urls import reverse
from django.utils import timezone
from django.utils.translation import ugettext_lazy as _

import users.models
import users.signals
from cotisations.utils import (find_payment_method, send_mail_invoice,
                               send_mail_voucher)
from cotisations.validators import check_no_balance
from machines.models import regen
from preferences.models import CotisationsOption
from re2o.field_permissions import FieldPermissionModelMixin
from re2o.mixins import AclMixin, RevMixin


class BaseInvoice(RevMixin, AclMixin, FieldPermissionModelMixin, models.Model):
    date = models.DateTimeField(auto_now_add=True, verbose_name=_("date"))

    class Meta:
        abstract = False

    # TODO : change prix to price
    def prix(self):
        """
        Returns: the raw price without the quantities.
        Deprecated, use :total_price instead.
        """
        price = Vente.objects.filter(facture=self).aggregate(models.Sum("prix"))[
            "prix__sum"
        ]
        return price

    # TODO : change prix to price
    def prix_total(self):
        """
        Returns: the total price for an invoice. Sum all the articles' prices
        and take the quantities into account.
        """
        # TODO : change Vente to somethingelse
        return (
            Vente.objects.filter(facture=self).aggregate(
                total=models.Sum(
                    models.F("prix") * models.F("number"),
                    output_field=models.DecimalField(),
                )
            )["total"]
            or 0
        )

    def name(self):
        """
        Returns : a string with the name of all the articles in the invoice.
        Used for reprensenting the invoice with a string.
        """
        name = " - ".join(
            Vente.objects.filter(facture=self).values_list("name", flat=True)
        )
        return name

    def name_detailed(self):
        """
        Return:
         - a list of strings with the name of all article in the invoice
        and their quantity.
        """
        ventes = self.vente_set.all()
        strings = ["{} x {}".format(v.number, v.name) for v in ventes]
        return strings


# TODO : change facture to invoice
class Facture(BaseInvoice):
    """
    The model for an invoice. It reprensents the fact that a user paid for
    something (it can be multiple article paid at once).

    An invoice is linked to :
        * one or more purchases (one for each article sold that time)
        * a user (the one who bought those articles)
        * a payment method (the one used by the user)
        * (if applicable) a bank
        * (if applicable) a cheque number.
    Every invoice is dated throught the 'date' value.
    An invoice has a 'controlled' value (default : False) which means that
    someone with high enough rights has controlled that invoice and taken it
    into account. It also has a 'valid' value (default : True) which means
    that someone with high enough rights has decided that this invoice was not
    valid (thus it's like the user never paid for his articles). It may be
    necessary in case of non-payment.
    """

    user = models.ForeignKey("users.User", on_delete=models.PROTECT)
    # TODO : change paiement to payment
    paiement = models.ForeignKey("Paiement", on_delete=models.PROTECT)
    # TODO : change banque to bank
    banque = models.ForeignKey(
        "Banque", on_delete=models.PROTECT, blank=True, null=True
    )
    # TODO : maybe change to cheque nummber because not evident
    cheque = models.CharField(
        max_length=255, blank=True, verbose_name=_("cheque number")
    )
    # TODO : change name to validity for clarity
    valid = models.BooleanField(default=False, verbose_name=_("validated"))
    # TODO : changed name to controlled for clarity
    control = models.BooleanField(default=False, verbose_name=_("controlled"))

    class Meta:
        abstract = False
        permissions = (
            # TODO : change facture to invoice
            ("change_facture_control", _('Can edit the "controlled" state')),
            ("change_all_facture", _("Can edit all the previous invoices")),
        )
        verbose_name = _("invoice")
        verbose_name_plural = _("invoices")

    def linked_objects(self):
        """Return linked objects : machine and domain.
        Usefull in history display"""
        return self.vente_set.all()

    def can_edit(self, user_request, *args, **kwargs):
        user_can, _message, permissions = self.user.can_edit(
            user_request, *args, **kwargs
        )
        if not user_request.has_perm("cotisations.change_facture"):
            return (
                False,
                _("You don't have the right to edit an invoice."),
                ("cotisations.change_facture",),
            )
        elif (
            not user_request.has_perm("cotisations.change_all_facture") and not user_can
        ):
            return (
                False,
                _("You don't have the right to edit this user's invoices."),
                ("cotisations.change_all_facture",) + (permissions or ()),
            )
        elif not user_request.has_perm("cotisations.change_all_facture") and (
            self.control or not self.valid
        ):
            return (
                False,
                _(
                    "You don't have the right to edit an invoice"
                    " already controlled or invalidated."
                ),
                ("cotisations.change_all_facture",),
            )
        else:
            return True, None, None

    def can_delete(self, user_request, *args, **kwargs):
        user_can, _message, permissions = self.user.can_edit(
            user_request, *args, **kwargs
        )
        if not user_request.has_perm("cotisations.delete_facture"):
            return (
                False,
                _("You don't have the right to delete an invoice."),
                ("cotisations.delete_facture",),
            )
        elif (
            not user_request.has_perm("cotisations.change_all_facture") and not user_can
        ):
            return (
                False,
                _("You don't have the right to delete this user's invoices."),
                ("cotisations.change_all_facture",) + (permissions or ()),
            )
        elif not user_request.has_perm("cotisations.change_all_facture") and (
            self.control or not self.valid
        ):
            return (
                False,
                _(
                    "You don't have the right to delete an invoice"
                    " already controlled or invalidated."
                ),
                ("cotisations.change_all_facture",),
            )
        else:
            return True, None, None

    def can_view(self, user_request, *_args, **_kwargs):
        if not user_request.has_perm("cotisations.view_facture"):
            if self.user != user_request:
                return (
                    False,
                    _(
                        "You don't have the right to view someone else's"
                        " invoices history."
                    ),
                    ("cotisations.view_facture",),
                )
            elif not self.valid:
                return (
                    False,
                    _("The invoice has been invalidated."),
                    ("cotisations.view_facture",),
                )
            else:
                return True, None, None
        else:
            return True, None, None

    @staticmethod
    def can_change_control(user_request, *_args, **_kwargs):
        """Returns True if the user can change the 'controlled' status of
        this invoice"""
        can = user_request.has_perm("cotisations.change_facture_control")
        return (
            can,
            _('You don\'t have the right to edit the "controlled" state.')
            if not can
            else None,
            ("cotisations.change_facture_control",),
        )

    @staticmethod
    def can_create(user_request, *_args, **_kwargs):
        """Check if a user can create an invoice.

        :param user_request: The user who wants to create an invoice.
        :return: a message and a boolean which is True if the user can create
            an invoice or if the `options.allow_self_subscription` is set.
        """
        if user_request.has_perm("cotisations.add_facture"):
            return True, None, None
        if len(Paiement.find_allowed_payments(user_request)) <= 0:
            return (
                False,
                _("There are no payment methods that you can use."),
                ("cotisations.add_facture",),
            )
        if len(Article.find_allowed_articles(user_request, user_request)) <= 0:
            return (
                False,
                _("There are no articles that you can buy."),
                ("cotisations.add_facture",),
            )
        return True, None, None

    def __init__(self, *args, **kwargs):
        super(Facture, self).__init__(*args, **kwargs)
        self.field_permissions = {"control": self.can_change_control}
        self.__original_valid = self.valid
        self.__original_control = self.control

    def get_subscription(self):
        """Returns every subscription associated with this invoice."""
        return Cotisation.objects.filter(
            vente__in=self.vente_set.filter(
                ~(Q(duration_membership=0)) | ~(Q(duration_days_membership=0))
            )
        )

    def is_subscription(self):
        """Returns True if this invoice contains at least one subscribtion."""
        return bool(self.get_subscription())

    def reorder_purchases(self):
        date_adh = max(self.date, self.user.end_adhesion())
        date_con = max(self.date, self.user.end_connexion())
        for purchase in self.vente_set.all():
            if hasattr(purchase, "cotisation"):
                cotisation = purchase.cotisation
                cotisation.date_start_con = date_con
                date_con += relativedelta(
                    months=(purchase.duration_connection or 0) * purchase.number,
                    days=(purchase.duration_days_connection or 0) * purchase.number,
                )
                cotisation.date_end_con = date_con
                cotisation.date_start_memb = date_adh
                date_adh += relativedelta(
                    months=(purchase.duration_membership or 0) * purchase.number,
                    days=(purchase.duration_days_membership or 0) * purchase.number,
                )
                cotisation.date_end_memb = date_adh
                cotisation.save()
                purchase.facture = self
                purchase.save()

    def save(self, *args, **kwargs):
        try:
            request = kwargs.pop("request")
        except:
            request = None

        super(Facture, self).save(*args, **kwargs)

        if not self.__original_valid and self.valid:
            self.reorder_purchases()
            send_mail_invoice(self, request)
        if (
            self.is_subscription()
            and not self.__original_control
            and self.control
            and CotisationsOption.get_cached_value("send_voucher_mail")
            and self.user.is_adherent()
        ):
            send_mail_voucher(self, request)

    def __str__(self):
        return str(self.user) + " " + str(self.date)


@receiver(post_save, sender=Facture)
def facture_post_save(**kwargs):
    """
    Synchronise the LDAP user after an invoice has been saved.
    """
    facture = kwargs["instance"]
    if facture.valid:
        user = facture.user
        user.set_active()
        users.signals.synchronise.send(
            sender=users.models.User,
            instance=user,
            base=False,
            access_refresh=True,
            mac_refresh=False,
        )


@receiver(post_delete, sender=Facture)
def facture_post_delete(**kwargs):
    """
    Synchronise the LDAP user after an invoice has been deleted.
    """
    user = kwargs["instance"].user
    users.signals.synchronise.send(
        sender=users.models.User,
        instance=user,
        base=False,
        access_refresh=True,
        mac_refresh=False,
    )


class CustomInvoice(BaseInvoice):

    recipient = models.CharField(max_length=255, verbose_name=_("recipient"))
    payment = models.CharField(max_length=255, verbose_name=_("payment type"))
    address = models.CharField(max_length=255, verbose_name=_("address"))
    paid = models.BooleanField(verbose_name=_("paid"), default=False)
    remark = models.TextField(verbose_name=_("remark"), blank=True, null=True)


class CostEstimate(CustomInvoice):

    validity = models.DurationField(
        verbose_name=_("period of validity"), help_text="DD HH:MM:SS"
    )
    final_invoice = models.ForeignKey(
        CustomInvoice,
        on_delete=models.SET_NULL,
        null=True,
        blank=True,
        related_name="origin_cost_estimate",
        primary_key=False,
    )

    def create_invoice(self):
        """Create a CustomInvoice from the CostEstimate."""
        if self.final_invoice is not None:
            return self.final_invoice
        invoice = CustomInvoice()
        invoice.recipient = self.recipient
        invoice.payment = self.payment
        invoice.address = self.address
        invoice.paid = False
        invoice.remark = self.remark
        invoice.date = timezone.now()
        invoice.save()
        self.final_invoice = invoice
        self.save()
        for sale in self.vente_set.all():
            Vente.objects.create(
                facture=invoice, name=sale.name, prix=sale.prix, number=sale.number
            )
        return invoice

    def can_delete(self, user_request, *args, **kwargs):
        if not user_request.has_perm("cotisations.delete_costestimate"):
            return (
                False,
                _("You don't have the right to delete a cost estimate."),
                ("cotisations.delete_costestimate",),
            )
        if self.final_invoice is not None:
            return (
                False,
                _("The cost estimate has an invoice and can't be deleted."),
                None,
            )
        return True, None, None


# TODO : change Vente to Purchase
class Vente(RevMixin, AclMixin, models.Model):
    """
    The model defining a purchase. It consist of one type of article being
    sold. In particular there may be multiple purchases in a single invoice.

    It's reprensentated by:
        * an amount (the number of items sold)
        * an invoice (whose the purchase is part of)
        * an article
        * (if applicable) a cotisation (which holds some informations about
            the effect of the purchase on the time agreed for this user)
    """

    # TODO : change facture to invoice
    facture = models.ForeignKey(
        "BaseInvoice", on_delete=models.CASCADE, verbose_name=_("invoice")
    )
    # TODO : change number to amount for clarity
    number = models.IntegerField(
        validators=[MinValueValidator(1)], verbose_name=_("amount")
    )
    # TODO : change this field for a ForeinKey to Article
    # Note: With a foreign key, modifing an Article modifis the Purchase, wich is bad.
    # To use a foreign key, you need to make Article read only
    name = models.CharField(max_length=255, verbose_name=_("article"))
    # TODO : change prix to price
    # TODO : this field is not needed if you use Article ForeignKey
    prix = models.DecimalField(max_digits=5, decimal_places=2, verbose_name=_("price"))
    # TODO : this field is not needed if you use Article ForeignKey
    duration_connection = models.PositiveIntegerField(
        default=0, verbose_name=_("duration of the connection (in months)")
    )
    duration_days_connection = models.PositiveIntegerField(
        default=0,
        verbose_name=_(
            "duration of the connection (in days, will be added to duration in months)"
        ),
    )
    duration_membership = models.PositiveIntegerField(
        default=0, verbose_name=_("duration of the membership (in months)")
    )
    duration_days_membership = models.PositiveIntegerField(
        default=0,
        verbose_name=_(
            "duration of the membership (in days, will be added to duration in months)"
        ),
    )

    class Meta:
        permissions = (("change_all_vente", _("Can edit all the previous purchases")),)
        verbose_name = _("purchase")
        verbose_name_plural = _("purchases")

    # TODO : change prix_total to total_price
    def prix_total(self):
        """
        Returns: the total of price for this amount of items.
        """
        return self.prix * self.number

    def update_cotisation(self):
        """
        Update the related object 'cotisation' if there is one. Based on the
        duration of the purchase.
        """
        if hasattr(self, "cotisation"):
            cotisation = self.cotisation
            cotisation.date_end_memb = cotisation.date_start_memb + relativedelta(
                months=(self.duration_membership or 0) * self.number,
                days=(self.duration_days_membership or 0) * self.number,
            )
            cotisation.date_end_con = cotisation.date_start_con + relativedelta(
                months=(self.duration_connection or 0) * self.number,
                days=(self.duration_days_connection or 0) * self.number,
            )
        return

    def create_cotis(self, date_start_con=False, date_start_memb=False):
        """
        Creates a cotisation without initializing the dates (start and end ar set to self.facture.facture.date)
        and without saving it. You should use Facture.reorder_purchases to set the right dates.
        """
        try:
            invoice = self.facture.facture
        except Facture.DoesNotExist:
            return
        if not hasattr(self, "cotisation") and self.test_membership_or_connection():
            cotisation = Cotisation(vente=self)
            if date_start_con:
                cotisation.date_start_con = date_start_con
                cotisation.date_end_con = cotisation.date_start_con + relativedelta(
                    months=(self.duration_connection or 0) * self.number,
                    days=(self.duration_days_connection or 0) * self.number,
                )
                self.save()
                cotisation.save()
            if date_start_memb:
                cotisation.date_start_memb = date_start_memb
                cotisation.date_end_memb = cotisation.date_start_memb + relativedelta(
                    months=(self.duration_membership or 0) * self.number,
                    days=(self.duration_days_membership or 0) * self.number,
                )
                self.save()
                cotisation.save()
            else:
                cotisation.date_start_con = invoice.date
                cotisation.date_start_memb = invoice.date
                cotisation.date_end_con = invoice.date
                cotisation.date_end_memb = invoice.date

    def save(self, *args, **kwargs):
        """
        Save a purchase object and check if all the fields are coherents
        It also update the associated cotisation in the changes have some
        effect on the user's cotisation
        """
        self.update_cotisation()
        super(Vente, self).save(*args, **kwargs)

    def can_edit(self, user_request, *args, **kwargs):
        user_can, _message, permissions = self.facture.user.can_edit(
            user_request, *args, **kwargs
        )
        if not user_request.has_perm("cotisations.change_vente"):
            return (
                False,
                _("You don't have the right to edit a purchase."),
                ("cotisations.change_vente",),
            )
        elif not (user_request.has_perm("cotisations.change_all_facture") or user_can):
            return (
                False,
                _("You don't have the right to edit this user's purchases."),
                ("cotisations.change_all_facture",) + (permissions or ()),
            )
        elif not user_request.has_perm("cotisations.change_all_vente") and (
            self.facture.control or not self.facture.valid
        ):
            return (
                False,
                _(
                    "You don't have the right to edit a purchase"
                    " already controlled or invalidated."
                ),
                ("cotisations.change_all_vente",),
            )
        else:
            return True, None, None

    def can_delete(self, user_request, *args, **kwargs):
        user_can, _message, permissions = self.facture.user.can_edit(
            user_request, *args, **kwargs
        )
        if not user_request.has_perm("cotisations.delete_vente"):
            return (
                False,
                _("You don't have the right to delete a purchase."),
                ("cotisations.delete_vente",),
            )
        if not user_can:
            return (
                False,
                _("You don't have the right to delete this user's purchases."),
                permissions,
            )
        if self.facture.control or not self.facture.valid:
            return (
                False,
                _(
                    "You don't have the right to delete a purchase"
                    " already controlled or invalidated."
                ),
                None,
            )
        else:
            return True, None, None

    def can_view(self, user_request, *_args, **_kwargs):
        if (
            not user_request.has_perm("cotisations.view_vente")
            and self.facture.user != user_request
        ):
            return (
                False,
                _(
                    "You don't have the right to view someone"
                    " else's purchase history."
                ),
                ("cotisations.view_vente",),
            )
        else:
            return True, None, None

    def __str__(self):
        return str(self.name) + " " + str(self.facture)

    def test_membership_or_connection(self):
        """Test if the purchase include membership or connecton"""
        return (
            self.duration_membership
            or self.duration_days_membership
            or self.duration_connection
            or self.duration_days_connection
        )


# TODO : change vente to purchase
@receiver(post_save, sender=Vente)
def vente_post_save(**kwargs):
    """
    Creates a 'cotisation' related object if needed and synchronise the
    LDAP user when a purchase has been saved.
    """
    purchase = kwargs["instance"]
    try:
        purchase.facture.facture
    except Facture.DoesNotExist:
        return
    if hasattr(purchase, "cotisation"):
        purchase.cotisation.vente = purchase
        purchase.cotisation.save()
    if purchase.test_membership_or_connection():
        purchase.create_cotis()
        purchase.cotisation.save()
        user = purchase.facture.facture.user
        user.set_active()
        users.signals.synchronise.send(
            sender=users.models.User,
            instance=user,
            base=True,
            access_refresh=True,
            mac_refresh=False,
        )


# TODO : change vente to purchase
@receiver(post_delete, sender=Vente)
def vente_post_delete(**kwargs):
    """
    Synchronise the LDAP user after a purchase has been deleted.
    """
    purchase = kwargs["instance"]
    try:
        invoice = purchase.facture.facture
    except Facture.DoesNotExist:
        return
    if purchase.type_cotisation:
        user = invoice.user
        users.signals.synchronise.send(
            sender=users.models.User,
            instance=user,
            base=True,
            access_refresh=True,
            mac_refresh=False,
        )


class Article(RevMixin, AclMixin, models.Model):
    """
    The definition of an article model. It represents a type of object
    that can be sold to the user.

    It's represented by:
        * a name
        * a price
        * a duration for the membership
        * a duration for the connection
        * if the article can be purchased without membership
        * a type of user (indicating what kind of user can buy this article)
    """

    USER_TYPES = (
        ("Adherent", _("Member")),
        ("Club", _("Club")),
        ("All", _("Both of them")),
    )

    name = models.CharField(max_length=255, verbose_name=_("designation"))
    # TODO : change prix to price
    prix = models.DecimalField(
        max_digits=5, decimal_places=2, verbose_name=_("unit price")
    )

    duration_membership = models.PositiveIntegerField(
        verbose_name=_("duration of the membership (in months)")
    )
    duration_days_membership = models.PositiveIntegerField(
        verbose_name=_(
            "duration of the membership (in days, will be added to duration in months)"
        ),
    )
    duration_connection = models.PositiveIntegerField(
        verbose_name=_("duration of the connection (in months)")
    )
    duration_days_connection = models.PositiveIntegerField(
        verbose_name=_(
            "duration of the connection (in days, will be added to duration in months)"
        ),
    )

    need_membership = models.BooleanField(
        default=True,
        verbose_name=_("need membership to be purchased"),
    )

    type_user = models.CharField(
        choices=USER_TYPES,
        default="All",
        max_length=255,
        verbose_name=_("type of users concerned"),
    )
    available_for_everyone = models.BooleanField(
        default=False, verbose_name=_("is available for every user")
    )

    unique_together = ("name", "type_user")

    class Meta:
        permissions = (("buy_every_article", _("Can buy every article")),)
        verbose_name = "article"
        verbose_name_plural = "articles"

    def clean(self):
        if self.name.lower() == "solde":
            raise ValidationError(_("Solde is a reserved article name."))

    def __str__(self):
        return self.name

    def can_buy_article(self, user, *_args, **_kwargs):
        """Check if a user can buy this article.

        Args:
            self: The article
            user: The user requesting buying

        Returns:
            A boolean stating if usage is granted and an explanation
            message if the boolean is `False`.
        """
        can = (
            self.available_for_everyone
            or user.has_perm("cotisations.buy_every_article")
            or user.has_perm("cotisations.add_facture")
        )
        return (
            can,
            _("You can't buy this article.") if not can else None,
            ("cotisations.buy_every_article", "cotisations.add_facture"),
        )

    @classmethod
    def find_allowed_articles(cls, user, target_user):
        """Finds every allowed articles for an user, on a target user.

        Args:
            user: The user requesting articles.
            target_user: The user to sell articles to
        """
        if target_user is None:
            objects_pool = cls.objects.all()
        elif target_user.is_class_club:
            objects_pool = cls.objects.filter(Q(type_user="All") | Q(type_user="Club"))
        else:
            objects_pool = cls.objects.filter(
                Q(type_user="All") | Q(type_user="Adherent")
            )
        if target_user is not None and not target_user.is_adherent():
            objects_pool = objects_pool.filter(
                Q(duration_membership__gt=0)
                | Q(duration_days_membership__gt=0)
                | Q(need_membership=False)
            )
        if user.has_perm("cotisations.buy_every_article"):
            return objects_pool
        return objects_pool.filter(available_for_everyone=True)


class Banque(RevMixin, AclMixin, models.Model):
    """
    The model defining a bank. It represents a user's bank. It's mainly used
    for statistics by regrouping the user under their bank's name and avoid
    the use of a simple name which leads (by experience) to duplicates that
    only differs by a capital letter, a space, a misspelling, ... That's why
    it's easier to use simple object for the banks.
    """

    name = models.CharField(max_length=255)

    class Meta:
        verbose_name = _("bank")
        verbose_name_plural = _("banks")

    def __str__(self):
        return self.name


# TODO : change Paiement to Payment
class Paiement(RevMixin, AclMixin, models.Model):
    """
    The model defining a payment method. It is how the user is paying for the
    invoice. It's easier to know this information when doing the accouts.
    It is represented by:
        * a name
    """

    # TODO : change moyen to method
    moyen = models.CharField(max_length=255, verbose_name=_("method"))
    available_for_everyone = models.BooleanField(
        default=False, verbose_name=_("is available for every user")
    )
    is_balance = models.BooleanField(
        default=False,
        editable=False,
        verbose_name=_("is user balance"),
        help_text=_("There should be only one balance payment method."),
        validators=[check_no_balance],
    )

    class Meta:
        permissions = (("use_every_payment", _("Can use every payment method")),)
        verbose_name = _("payment method")
        verbose_name_plural = _("payment methods")

    def __str__(self):
        return self.moyen

    def clean(self):
        """l
        Override of the herited clean function to get a correct name
        """
        self.moyen = self.moyen.title()

    def end_payment(self, invoice, request, use_payment_method=True):
        """
        The general way of ending a payment.

        Args:
            invoice: The invoice being created.
            request: Request sent by the user.
            use_payment_method: If this flag is set to True and`self` has
                an attribute `payment_method`, returns the result of
                `self.payment_method.end_payment(invoice, request)`

        Returns:
            An `HttpResponse`-like object.
        """
        payment_method = find_payment_method(self)
        if payment_method is not None and use_payment_method:
            return payment_method.end_payment(invoice, request)

        # So make this invoice valid, trigger send mail
        invoice.valid = True
        invoice.save(request=request)

        # In case a cotisation was bought, inform the user, the
        # cotisation time has been extended too
        if any(
            sell.test_membership_or_connection() for sell in invoice.vente_set.all()
        ):
            messages.success(
                request,
                _(
                    "The subscription of %(member_name)s was extended to"
                    " %(end_date)s."
                )
                % {
                    "member_name": invoice.user.pseudo,
                    "end_date": invoice.user.end_adhesion(),
                },
            )
        # Else, only tell the invoice was created
        else:
            messages.success(request, _("The invoice was created."))
        return redirect(reverse("users:profil", kwargs={"userid": invoice.user.pk}))

    def can_use_payment(self, user, *_args, **_kwargs):
        """Check if a user can use this payment.

        Args:
            self: The payment
            user: The user requesting usage
        Returns:
            A boolean stating if usage is granted and an explanation
            message if the boolean is `False`.
        """
        can = (
            self.available_for_everyone
            or user.has_perm("cotisations.use_every_payment")
            or user.has_perm("cotisations.add_facture")
        )
        return (
            can,
            _("You can't use this payment method.") if not can else None,
            ("cotisations.use_every_payment", "cotisations.add_facture"),
        )

    @classmethod
    def find_allowed_payments(cls, user):
        """Finds every allowed payments for an user.

        Args:
            user: The user requesting payment methods.
        """
        if user.has_perm("cotisations.use_every_payment"):
            return cls.objects.all()
        return cls.objects.filter(available_for_everyone=True)

    def get_payment_method_name(self):
        p = find_payment_method(self)
        if p is not None:
            return p._meta.verbose_name
        return _("No custom payment methods.")


class Cotisation(RevMixin, AclMixin, models.Model):
    """
    The model defining a cotisation. It holds information about the time a user
    is allowed when he has paid something.
    It characterised by :
        * a date_start_memb (the date when the membership begins/began
        * a date_end_memb (the date when the membership ends/ended
        * a date_start_con (the date when the connection begins/began)
        * a date_end_con (the date when the connection ends/ended)
        * a purchase (the related objects this cotisation is linked to)
    """

    # TODO : change vente to purchase
    vente = models.OneToOneField(
        "Vente", on_delete=models.CASCADE, null=True, verbose_name=_("purchase")
    )
    date_start_con = models.DateTimeField(
        verbose_name=_("start date for the connection")
    )
    date_end_con = models.DateTimeField(verbose_name=_("end date for the connection"))
    date_start_memb = models.DateTimeField(
        verbose_name=_("start date for the membership")
    )
    date_end_memb = models.DateTimeField(verbose_name=_("end date for the membership"))

    class Meta:
        permissions = (
            ("change_all_cotisation", _("Can edit the previous subscriptions")),
        )
        verbose_name = _("subscription")
        verbose_name_plural = _("subscriptions")

    def can_edit(self, user_request, *_args, **_kwargs):
        if not user_request.has_perm("cotisations.change_cotisation"):
            return (
                False,
                _("You don't have the right to edit a subscription."),
                ("cotisations.change_cotisation",),
            )
        elif not user_request.has_perm("cotisations.change_all_cotisation") and (
            self.vente.facture.control or not self.vente.facture.valid
        ):
            return (
                False,
                _(
                    "You don't have the right to edit a subscription"
                    " already controlled or invalidated."
                ),
                ("cotisations.change_all_cotisation",),
            )
        else:
            return True, None, None

    def can_delete(self, user_request, *_args, **_kwargs):
        if not user_request.has_perm("cotisations.delete_cotisation"):
            return (
                False,
                _("You don't have the right to delete a subscription."),
                ("cotisations.delete_cotisation",),
            )
        if self.vente.facture.control or not self.vente.facture.valid:
            return (
                False,
                _(
                    "You don't have the right to delete a subscription"
                    " already controlled or invalidated."
                ),
                None,
            )
        else:
            return True, None, None

    def can_view(self, user_request, *_args, **_kwargs):
        if (
            not user_request.has_perm("cotisations.view_cotisation")
            and self.vente.facture.user != user_request
        ):
            return (
                False,
                _(
                    "You don't have the right to view someone else's"
                    " subscription history."
                ),
                ("cotisations.view_cotisation",),
            )
        else:
            return True, None, None

    def __str__(self):
        return (
            str(self.vente)
            + "from "
            + str(self.date_start_memb)
            + " to "
            + str(self.date_end_memb)
            + " for membership, "
            + str(self.date_start_con)
            + " to "
            + str(self.date_end_con)
            + " for the connection."
        )


@receiver(post_save, sender=Cotisation)
def cotisation_post_save(**_kwargs):
    """
    Mark some services as needing a regeneration after the edition of a
    cotisation. Indeed the membership status may have changed.
    """
    regen("dns")
    regen("dhcp")
    regen("mac_ip_list")
    regen("mailing")


@receiver(post_delete, sender=Cotisation)
def cotisation_post_delete(**_kwargs):
    """
    Mark some services as needing a regeneration after the deletion of a
    cotisation. Indeed the membership status may have changed.
    """
    regen("mac_ip_list")
    regen("mailing")