75 lines
2.4 KiB
Python
75 lines
2.4 KiB
Python
from django import forms
|
|
from django.contrib.auth.forms import UserCreationForm
|
|
from django.contrib.auth.models import User
|
|
from django.core.exceptions import ValidationError
|
|
from django.forms import FileInput
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from .models import AdminRegistration, CoachRegistration, StudentRegistration
|
|
|
|
|
|
class SignupForm(UserCreationForm):
|
|
role = forms.ChoiceField(
|
|
label=lambda: _("role").capitalize(),
|
|
choices=lambda: [
|
|
("participant", _("participant").capitalize()),
|
|
("coach", _("coach").capitalize()),
|
|
],
|
|
)
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.fields["first_name"].required = True
|
|
self.fields["last_name"].required = True
|
|
self.fields["email"].required = True
|
|
|
|
class Meta:
|
|
model = User
|
|
fields = ('first_name', 'last_name', 'email', 'password1', 'password2', 'role',)
|
|
|
|
|
|
class UserForm(forms.ModelForm):
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.fields["first_name"].required = True
|
|
self.fields["last_name"].required = True
|
|
self.fields["email"].required = True
|
|
|
|
class Meta:
|
|
model = User
|
|
fields = ('first_name', 'last_name', 'email',)
|
|
|
|
|
|
class StudentRegistrationForm(forms.ModelForm):
|
|
class Meta:
|
|
model = StudentRegistration
|
|
fields = ('student_class', 'school', 'give_contact_to_animath',)
|
|
|
|
|
|
class PhotoAuthorizationForm(forms.ModelForm):
|
|
def clean_photo_authorization(self):
|
|
file = self.files["photo_authorization"]
|
|
if file.content_type not in ["application/pdf", "image/png", "image/jpeg"]:
|
|
raise ValidationError(_("The uploaded file must be a PDF, PNG of JPEG file."))
|
|
return self.cleaned_data["photo_authorization"]
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.fields["photo_authorization"].widget = FileInput()
|
|
|
|
class Meta:
|
|
model = StudentRegistration
|
|
fields = ('photo_authorization',)
|
|
|
|
|
|
class CoachRegistrationForm(forms.ModelForm):
|
|
class Meta:
|
|
model = CoachRegistration
|
|
fields = ('professional_activity', 'give_contact_to_animath',)
|
|
|
|
|
|
class AdminRegistrationForm(forms.ModelForm):
|
|
class Meta:
|
|
model = AdminRegistration
|
|
fields = ('role', 'give_contact_to_animath',)
|