fix fav

[wolnelektury.git] / src / club / forms.py

diff --git a/src/club/forms.py b/src/club/forms.py
index 1906bc9..0ee00dc 100644 (file)
--- a/src/club/forms.py
+++ b/src/club/forms.py
@@ -1,50 +1,97 @@
-# This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
-# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+# This file is part of Wolne Lektury, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Wolne Lektury. See NOTICE for more information.

 #
 from decimal import Decimal
 from django import forms
 #
 from decimal import Decimal
 from django import forms

-from . import models
-from .payment_methods import method_by_slug, methods
+from newsletter.forms import NewsletterForm
+from . import models, payment_methods

 from .payu.forms import CardTokenForm
 
 
 from .payu.forms import CardTokenForm
 
 

-class ScheduleForm(forms.ModelForm):
+class PayUCardTokenForm(CardTokenForm):
+    def get_queryset(self, view):
+        return view.get_schedule().payucardtoken_set
+
+
+
+class DonationStep1Form(forms.ModelForm):
+    switch = forms.CharField()
+    single_amount = forms.IntegerField(required=False)
+    monthly_amount = forms.IntegerField(required=False)
+    single_amount_selected = forms.IntegerField(required=False)
+    monthly_amount_selected = forms.IntegerField(required=False)
+    custom_amount = forms.IntegerField(required=False)
+
+    amount = forms.IntegerField(required=False) # hidden
+

     class Meta:
         model = models.Schedule
     class Meta:
         model = models.Schedule

-        fields = ['plan', 'method', 'amount', 'email']
-        widgets = {
-            'plan': forms.RadioSelect,
-            'method': forms.RadioSelect,
-        }
+        fields = [
+            'amount',
+            'monthly'
+            ]

 
 

-    def __init__(self, *args, request=None, **kwargs):
-        super(ScheduleForm, self).__init__(*args, **kwargs)
-        self.request = request
-        self.plans = models.Plan.objects.all()
-        self.payment_methods = methods
-        self.fields['amount'].required = False
+    def __init__(self, *args, referer=None, **kwargs):
+        self.referer = referer
+        super().__init__(*args, **kwargs)
+        club = models.Club.objects.first()
+        if self.instance.is_custom_amount():
+            self.fields['custom_amount'].initial = int(self.instance.amount)
+        if club is not None:
+            self.fields['custom_amount'].widget.attrs['min'] = club.min_amount

 
     def clean(self):
 
     def clean(self):

-        cleaned_data = super(ScheduleForm, self).clean()
+        state = {}
+        state['monthly'] = self.cleaned_data['switch'] == 'monthly'
+        which = 'monthly' if state['monthly'] else 'single'
+        state['amount'] = \
+            self.cleaned_data[f'{which}_amount'] or \
+            self.cleaned_data['custom_amount'] or \
+            self.cleaned_data[f'{which}_amount_selected']

 
 

-        if 'plan' in cleaned_data:
-            cleaned_data['amount'] = self.fields['amount'].clean(
-                self.request.POST['amount-{}'.format(cleaned_data['plan'].id)]
-            )
+        return state

 
 

-            if cleaned_data['amount'] < cleaned_data['plan'].min_amount:
-                self.add_error(
-                    'amount',
-                    'Minimalna kwota dla tego planu to %d zł.' % cleaned_data['plan'].min_amount
+    def save(self, *args, **kwargs):
+        if self.referer is not None:
+            self.instance.source = self.referer
+        return super().save(*args, **kwargs)
+
+
+class DonationStep2Form(forms.ModelForm, NewsletterForm):
+    class Meta:
+        model = models.Schedule
+        fields = [
+            'first_name', 'last_name',
+            'email',
+            ]
+        widgets = {
+            'amount': forms.HiddenInput,
+            'monthly': forms.HiddenInput,
+        }
+    
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+
+        self.consent = []
+        for c in models.Consent.objects.filter(active=True).order_by('order'):
+            key = f'consent{c.id}'
+            if not c.required:
+                self.fields[key] = forms.BooleanField(
+                    label=c.text,
+                    required=c.required

                 )
                 )

+            self.consent.append((
+                c, key, (lambda k: lambda: self[k])(key)
+            ))

 
 

-        if 'method' in cleaned_data:
-            method = method_by_slug[cleaned_data['method']]
-            if method not in cleaned_data['plan'].payment_methods():
-                self.add_error('method', 'Wybrana metoda płatności nie jest dostępna dla tego planu.')
+    def save(self, *args, **kwargs):
+        NewsletterForm.save(self, *args, **kwargs)
+        instance = super().save(*args, **kwargs)

 
 

+        consents = []
+        for consent, key, consent_field in self.consent:
+            if consent.required or self.cleaned_data[key]:
+                instance.consent.add(consent)

 
 

+        return instance

 
 

-class PayUCardTokenForm(CardTokenForm):
-    def get_queryset(self, view):
-        return view.get_schedule().payucardtoken_set