mirror of
https://gitlab2.federez.net/re2o/re2o
synced 2024-11-25 12:53:11 +00:00
114 lines
4.3 KiB
Python
114 lines
4.3 KiB
Python
# -*- mode: python; coding: utf-8 -*-
|
|
# Re2o est un logiciel d'administration développé initiallement au rezometz. Il
|
|
# se veut agnostique au réseau considéré, de manière à être installable en
|
|
# quelques clics.
|
|
#
|
|
# 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.
|
|
from django import forms
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
from . import PAYMENT_METHODS
|
|
from cotisations.utils import find_payment_method
|
|
|
|
def payment_method_factory(payment, *args, creation=True, **kwargs):
|
|
"""This function finds the right payment method form for a given payment.
|
|
|
|
If the payment has a payment method, returns a ModelForm of it. Else if
|
|
it is the creation of the payment, a `PaymentMethodForm`.
|
|
Else `None`.
|
|
|
|
Args:
|
|
payment: The payment
|
|
*args: arguments passed to the form
|
|
creation: Should be True if you are creating the payment
|
|
**kwargs: passed to the form
|
|
|
|
Returns:
|
|
A form or None
|
|
"""
|
|
payment_method = kwargs.pop('instance', find_payment_method(payment))
|
|
if payment_method is not None:
|
|
return forms.modelform_factory(type(payment_method), fields='__all__')(
|
|
*args,
|
|
instance=payment_method,
|
|
**kwargs
|
|
)
|
|
elif creation:
|
|
return PaymentMethodForm(*args, **kwargs)
|
|
|
|
|
|
class PaymentMethodForm(forms.Form):
|
|
"""A special form which allows you to add a payment method to a `Payment`
|
|
object.
|
|
"""
|
|
|
|
payment_method = forms.ChoiceField(
|
|
label=_("Special payment method"),
|
|
help_text=_("Warning: you will not be able to change the payment "
|
|
"method later. But you will be allowed to edit the other "
|
|
"options."
|
|
),
|
|
required=False
|
|
)
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(PaymentMethodForm, self).__init__(*args, **kwargs)
|
|
prefix = kwargs.get('prefix', None)
|
|
self.fields['payment_method'].choices = [(i,p.NAME) for (i,p) in enumerate(PAYMENT_METHODS)]
|
|
self.fields['payment_method'].choices.insert(0, ('', _('no')))
|
|
self.fields['payment_method'].widget.attrs = {
|
|
'id': 'paymentMethodSelect'
|
|
}
|
|
self.templates = [
|
|
forms.modelform_factory(p.PaymentMethod, fields='__all__')(prefix=prefix)
|
|
for p in PAYMENT_METHODS
|
|
]
|
|
|
|
def clean(self):
|
|
"""A classic `clean` method, except that it replaces
|
|
`self.payment_method` by the payment method object if one has been
|
|
found. Tries to call `payment_method.valid_form` if it exists.
|
|
"""
|
|
super(PaymentMethodForm, self).clean()
|
|
choice = self.cleaned_data['payment_method']
|
|
if choice=='':
|
|
return
|
|
choice = int(choice)
|
|
model = PAYMENT_METHODS[choice].PaymentMethod
|
|
form = forms.modelform_factory(model, fields='__all__')(self.data, prefix=self.prefix)
|
|
self.payment_method = form.save(commit=False)
|
|
if hasattr(self.payment_method, 'valid_form'):
|
|
self.payment_method.valid_form(self)
|
|
return self.cleaned_data
|
|
|
|
|
|
|
|
def save(self, payment, *args, **kwargs):
|
|
"""Saves the payment method.
|
|
|
|
Tries to call `payment_method.alter_payment` if it exists.
|
|
"""
|
|
commit = kwargs.pop('commit', True)
|
|
if not hasattr(self, 'payment_method'):
|
|
return None
|
|
self.payment_method.payment = payment
|
|
if hasattr(self.payment_method, 'alter_payment'):
|
|
self.payment_method.alter_payment(payment)
|
|
if commit:
|
|
payment.save()
|
|
self.payment_method.save()
|
|
return self.payment_method
|