Aarni Halinen
132 lines
3.7 KiB
Python
132 lines
3.7 KiB
Python
"""File containing member forms."""
|
|
|
|
from django import forms
|
|
from django.utils import timezone
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from members.models import Member, Payment, Request
|
|
|
|
import csv
|
|
import logging
|
|
from dal import autocomplete
|
|
|
|
|
|
class CSVValidationError(Exception):
|
|
def __init__(self, form_errors):
|
|
self.form_errors = form_errors
|
|
|
|
|
|
class MemberForm(forms.ModelForm):
|
|
"""Member model form."""
|
|
|
|
class Meta:
|
|
"""Meta for Member model form."""
|
|
|
|
model = Member
|
|
fields = ["first_name", "last_name", "email", "POR", "AYY", "jas"]
|
|
|
|
class ImportResult:
|
|
def __init__(self, members, payments):
|
|
self.members = members
|
|
self.payments = payments
|
|
|
|
def _clean_boolean_field(self, key):
|
|
value = self.data.get(key, None)
|
|
if value in ["1", "0"]:
|
|
return bool(int(value))
|
|
else:
|
|
return value == "on"
|
|
|
|
def clean_jas(self):
|
|
return self._clean_boolean_field("jas")
|
|
|
|
def clean_AYY(self):
|
|
return self._clean_boolean_field("AYY")
|
|
|
|
@staticmethod
|
|
def csv_to_models(data, payment_source="AYY", delimiter=","):
|
|
clean_data = data.strip().split("\n")
|
|
clean_data = [row.rstrip(",").rstrip("\r").strip() for row in clean_data]
|
|
csv_reader = csv.DictReader(
|
|
clean_data,
|
|
fieldnames=MemberForm.Meta.fields,
|
|
delimiter=delimiter,
|
|
quoting=csv.QUOTE_NONE,
|
|
)
|
|
|
|
members = []
|
|
payments = []
|
|
for line in csv_reader:
|
|
try:
|
|
for key, value in line.items():
|
|
line[key] = value.strip()
|
|
except AttributeError as ex:
|
|
logging.error('Invalid line in CSV: "{}"'.format(line))
|
|
logging.error("Delimiter: {}".format(delimiter))
|
|
raise
|
|
|
|
email = line["email"]
|
|
member_exists = False
|
|
if Member.objects.filter(email=email).exists():
|
|
member_exists = True
|
|
|
|
if not member_exists:
|
|
form = MemberForm(line)
|
|
if not form.is_valid():
|
|
raise CSVValidationError(form.errors)
|
|
|
|
model = form.save(commit=False)
|
|
members.append(model)
|
|
|
|
else:
|
|
member = Member.objects.get(email=email)
|
|
payment_data = {
|
|
"source": payment_source,
|
|
"member": member.id,
|
|
"date": timezone.now(),
|
|
}
|
|
form = PaymentForm(payment_data)
|
|
if not form.is_valid():
|
|
raise CSVValidationError(form.errors)
|
|
|
|
model = form.save(commit=False)
|
|
payments.append(model)
|
|
|
|
return MemberForm.ImportResult(members, payments)
|
|
|
|
|
|
class PaymentForm(forms.ModelForm):
|
|
"""Payment model form."""
|
|
|
|
member = forms.ModelChoiceField(
|
|
queryset=Member.objects.all(),
|
|
widget=autocomplete.ModelSelect2(url="member-autocomplete"),
|
|
)
|
|
|
|
class Meta:
|
|
"""Meta for Payment model form."""
|
|
|
|
model = Payment
|
|
fields = ["date", "source", "member"]
|
|
labels = {"member": _("Member")}
|
|
|
|
|
|
class ApplicationForm(forms.ModelForm):
|
|
"""Member application model form."""
|
|
|
|
class Meta:
|
|
"""Meta for application model form."""
|
|
|
|
model = Request
|
|
fields = ["first_name", "last_name", "email", "AYY", "jas", "POR"]
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(ApplicationForm, self).__init__(*args, **kwargs)
|
|
|
|
self.fields["AYY"].label = _("I'm a member of AYY")
|
|
self.fields["jas"].label = _("I want to receive a weekly newsletter")
|
|
|
|
|
|
class UploadFileForm(forms.Form):
|
|
file = forms.FileField()
|