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",)

	username = forms.CharField(max_length=50, validators=[CallUsernameValidator()])

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

	def clean_newRefName(self):
		data = self.cleaned_data["newRefName"].strip().upper()
		if Reference.objects.filter(name=data).count() > 0:
			raise forms.ValidationError("Reference already exists")

		return data

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

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

		if existingRef and newRefName:
			raise forms.ValidationError("Select an existing ref or create a new one, not both!")
		if not existingRef and not newRefName:
			raise forms.ValidationError("Select either an existing ref 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"]

	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()
		if Reference.objects.filter(name=data).count() > 0:
			raise forms.ValidationError("Reference already exists")

		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:
			QSO.objects.get(owner=self.user, ownNo=data)
			raise forms.ValidationError("You already logged a QSO with the number %s" % data)
		except:
			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"]

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