Cop notes.

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

diff --git a/src/club/forms.py b/src/club/forms.py
index adf8959..2664fa3 100644 (file)
--- a/src/club/forms.py
+++ b/src/club/forms.py
@@ -1,29 +1,46 @@
-# -*- coding: utf-8
+# This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+#
+from decimal import Decimal
 from django import forms
+from django.utils.translation import ugettext as _
+from newsletter.forms import NewsletterForm
 from . import models
-from . import widgets
-from .payment_methods import method_by_slug 
+from .payu.forms import CardTokenForm
 
 
-class ScheduleForm(forms.ModelForm):
+class ScheduleForm(forms.ModelForm, NewsletterForm):
+    data_processing = '''Administratorem danych osobowych jest Fundacja Nowoczesna Polska (ul. Marszałkowska 84/92 lok. 125, 00-514 Warszawa). Podanie danych osobowych jest dobrowolne, ale konieczne do przeprowadzenia wpłaty. Dane są przetwarzane w zakresie niezbędnym do zaksięgowania darowizny i przekazywania Tobie powiadomień dotyczących wpłaty, a także wysyłania Tobie wiadomości mailowych promujących zbiórki i inne formy wsparcia Fundacji. W przypadku wyrażenia dodatkowej zgody adres e-mail zostanie wykorzystany także w zakresie niezbędnym do wysyłania newslettera odbiorcom. Osobom, których dane są zbierane, przysługuje prawo dostępu do treści swoich danych oraz ich poprawiania.'''
+
     class Meta:
         model = models.Schedule
-        fields = ['plan', 'method', 'amount', 'email']
+        fields = ['monthly', 'amount', 'email']
         widgets = {
-            'plan': forms.RadioSelect,
-            'method': forms.RadioSelect,
+            'amount': forms.HiddenInput,
+            'monthly': forms.HiddenInput,
         }
 
-    def __init__(self, *args, **kwargs):
-        super(ScheduleForm, self).__init__(*args, **kwargs)
-        self.fields['plan'].empty_label = None
+    def __init__(self, referer=None, **kwargs):
+        self.referer = referer
+        super().__init__(**kwargs)
+
+    def clean_amount(self):
+        value = self.cleaned_data['amount']
+        club = models.Club.objects.first()
+        if club and value < club.min_amount:
+            raise forms.ValidationError(
+                _('Minimal amount is %(amount)d PLN.') % {
+                    'amount': club.min_amount
+                }
+            )
+        return value
+
+    def save(self, *args, **kwargs):
+        NewsletterForm.save(self, *args, **kwargs)
+        self.instance.source = self.referer or ''
+        return super().save(*args, **kwargs)
 
-    def clean(self):
-        cleaned_data = super(ScheduleForm, self).clean()
-        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', 'Metoda płatności niedostępna dla tego planu.')
-        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)
 
+class PayUCardTokenForm(CardTokenForm):
+    def get_queryset(self, view):
+        return view.get_schedule().payucardtoken_set