cqtu/contest/forms.py

from django import forms
from django.contrib.auth.forms import UserCreationForm
from django.utils import timezone

from crispy_forms.helper import FormHelper
from crispy_forms.layout import Submit, Layout
from django.urls import reverse

from .models import User, Reference, QSO, ShadowCall, EntryCategory, Contest
from .validators import CallUsernameValidator, CallLogValidator

class CustomUserCreationForm(UserCreationForm):
    class Meta:
        model = User
        fields = ("username", "email", "dncall", "qrv2m", "qrv70cm", "extra2m70cm")

    username = forms.CharField(max_length=50, validators=[CallUsernameValidator()])
    email = forms.EmailField(required=True)

class UpdateRefForm(forms.Form):
    existingRef = forms.ModelChoiceField(label="Existing Exchange", queryset=Reference.objects.all(), help_text="If exchange already exists, select it here.", required=False)
    newRefName = forms.CharField(max_length=50, label="New Exchange", help_text="Enter name of new exchange, if we should create a new", required=False)

    location = forms.CharField(max_length=128, label='Exact Location', help_text="E.g. MAR bei den Fahrstuehlen, TEL 15. OG", required=False)
    opName = forms.CharField(max_length=128, label='Operators', help_text="Name of operator(s)", required=False)
    regTime = forms.DateTimeField(label="Registration time", help_text="Time of Registration")

    def clean_newRefName(self):
        data = self.cleaned_data["newRefName"].strip().upper()

        return data

    def clean(self):
        cleaned_data = super(UpdateRefForm, self).clean()

        existingRef = cleaned_data.get("existingRef")
        newRefName = cleaned_data.get("newRefName")

        if newRefName:
            try:
                ref = Reference.objects.get(name=newRefName)
                self.cleaned_data['newRefName'] = None
                self.cleaned_data['existingRef'] = ref
            except Reference.DoesNotExist:
                pass

        if existingRef and newRefName:
            raise forms.ValidationError("Select an existing exchange or create a new one, not both!")
        if not existingRef and not newRefName:
            raise forms.ValidationError("Select either an existing exchange or create a new one!")

class UpdateCategoryForm(forms.Form):
    entry = forms.ModelChoiceField(label="Entry category", queryset=EntryCategory.objects.all())
    def __init__(self, *args, **kwargs):
        super(UpdateCategoryForm, self).__init__(*args, **kwargs)

        self.helper = FormHelper()
        self.helper.form_class = "form-inline "
        self.helper.form_style = 'inline'
        self.helper.field_template = "bootstrap3/layout/inline_field.html"
        self.helper.action = reverse("profile")
        self.helper.add_input(Submit('submit', 'Set category'))
        self.helper.layout = Layout('entry')

    def clean(self):
        contest = Contest.objects.get(id=1)

        if contest.deadline < timezone.now():
            raise forms.ValidationError("The deadline for setting your contest category has passed")

class QSOForm(forms.ModelForm):
    class Meta:
        model = QSO
        #fields = ["ownNo", "band", "call", "reportTX", "reportRX", "refStr", "otherNo", "remarks"]
        fields = ["ownNo", "band", "call", "reportTX", "reportRX", "refStr", "remarks"]

    def __init__(self, user, *args, **kwargs):
        super(QSOForm, self).__init__(*args, **kwargs)
        self.user = user

        self.helper = FormHelper()
        self.helper.form_id = "qso-log-form"
        #self.helper.form_class = "form-inline "
        #self.helper.form_class = "form-horizontal"
        #self.helper.form_style = 'inline'
        #self.helper.field_template = "bootstrap3/layout/inline_field.html"
        self.helper.action = reverse("contest:log")
        self.helper.add_input(Submit('submit', 'Log'))
        #self.helper.layout = Layout(
        #    #*(QSOForm.Meta.fields +  [ButtonHolder(Submit('submit', 'Submit', css_class='button white'))]))
        #    *(QSOForm.Meta.fields +  [FormActions(Submit('submit', 'Log!'))]))


    def clean_call(self):
        data = self.cleaned_data["call"].upper().strip()

        try:
            CallLogValidator()(data)
        except forms.ValidationError:
            raise forms.ValidationError("Enter a valid callsign (1-2 chars, a number, 1-4 chars, maybe a /[A-Z])")

        if data == self.user.username:
            raise forms.ValidationError("You cannot log QSOs with yourself")

        return data

    def clean_ownNo(self):
        data = self.cleaned_data["ownNo"]

        if data < 1 or data > 100000:
            raise forms.ValidationError("Number has to be in range of [1, 1000000]")

        try:
            o = QSO.objects.get(owner=self.user.id, ownNo=data)
            if not (self.instance and self.instance.id and self.instance.id == o.id):
                raise forms.ValidationError("You already logged a QSO with the number %s" % data)
        except QSO.DoesNotExist:
            pass

        return data

    def clean_otherNo(self):
        data = self.cleaned_data["otherNo"]

        if not data:
            # empty value
            return None

        if data < 1 or data > 100000:
            raise forms.ValidationError("Number has to be in range of [1, 1000000]")

        return data

    def clean_refStr(self):
        return self.cleaned_data["refStr"].upper()

    def clean(self):
        cleaned_data = super(QSOForm, self).clean()
        band = cleaned_data.get("band")

        if band.contest.deadline < timezone.now():
            raise forms.ValidationError("The deadline for logging and editing QSOs has passed")

class QSOFormWithTime(QSOForm):
    class Meta:
        model = QSO
        #fields = ["time", "ownNo", "band", "call", "reportTX", "reportRX", "otherNo", "refStr", "remarks"]
        fields = ["time", "ownNo", "band", "call", "reportTX", "reportRX", "refStr", "remarks"]

class ShadowCallAddForm(forms.ModelForm):

    class Meta:
        model = ShadowCall
        fields = ['username']

    def __init__(self, *args, **kwargs):
        super(ShadowCallAddForm, self).__init__(*args, **kwargs)

        self.helper = FormHelper()
        self.helper.form_class = "form-inline "
        self.helper.form_style = 'inline'
        self.helper.field_template = "bootstrap3/layout/inline_field.html"
        self.helper.action = reverse("contest:registerRefs")
        self.helper.add_input(Submit('submit', 'Add shadow'))
        self.helper.layout = Layout('username')

    def clean_username(self):
        data = self.cleaned_data["username"]
        if User.objects.filter(username=data).count() > 0:
            raise forms.ValidationError("A user with this call already exists, this is not a shadow!")

        return data