29 lines
834 B
Python
29 lines
834 B
Python
|
from django import forms
|
||
|
|
from django.core.exceptions import ValidationError
|
||
|
|
|
||
|
|
from .models import Team
|
||
|
|
|
||
|
|
|
||
|
|
class TeamForm(forms.ModelForm):
|
||
|
|
class Meta:
|
||
|
|
model = Team
|
||
|
|
fields = ('name', 'trigram', 'grant_animath_access_videos',)
|
||
|
|
|
||
|
|
|
||
|
|
class JoinTeamForm(forms.ModelForm):
|
||
|
|
def clean_access_code(self):
|
||
|
|
access_code = self.cleaned_data["access_code"]
|
||
|
|
if not Team.objects.filter(access_code=access_code).exists():
|
||
|
|
raise ValidationError("No team was found with this access code.")
|
||
|
|
return access_code
|
||
|
|
|
||
|
|
def clean(self):
|
||
|
|
cleaned_data = super().clean()
|
||
|
|
if "access_code" in cleaned_data:
|
||
|
|
self.instance = Team.objects.get(access_code=cleaned_data["access_code"])
|
||
|
|
return cleaned_data
|
||
|
|
|
||
|
|
class Meta:
|
||
|
|
model = Team
|
||
|
|
fields = ('access_code',)
|