PyRIGS/training/forms.py

from django import forms

from training import models
from RIGS.models import Profile


class QualificationForm(forms.ModelForm):
    class Meta:
        model = models.TrainingItemQualification
        fields = '__all__'

    def __init__(self, *args, **kwargs):
        pk = kwargs.pop('pk', None)
        super().__init__(*args, **kwargs)
        self.fields['trainee'].initial = Profile.objects.get(pk=pk)
        self.fields['date'].widget.format = '%Y-%m-%d'

    def clean_date(self):
        date = self.cleaned_data['date']
        if date > date.today():
            raise forms.ValidationError('Qualification date may not be in the future')
        return date

    def clean_supervisor(self):
        supervisor = self.cleaned_data['supervisor']
        if supervisor.pk == self.cleaned_data['trainee'].pk:
            raise forms.ValidationError('One may not supervise oneself...')
        if not supervisor.is_supervisor:
            raise forms.ValidationError('Selected supervisor must actually *be* a supervisor...')
        return supervisor


class RequirementForm(forms.ModelForm):
    depth = forms.ChoiceField(choices=models.TrainingItemQualification.CHOICES)

    class Meta:
        model = models.TrainingLevelRequirement
        fields = '__all__'

    def __init__(self, *args, **kwargs):
        pk = kwargs.pop('pk', None)
        super().__init__(*args, **kwargs)
        self.fields['level'].initial = models.TrainingLevel.objects.get(pk=pk)