45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
|
from django.contrib.auth.forms import UserCreationForm
|
||
|
|
from django.contrib.auth.models import User
|
||
|
|
from django import forms
|
||
|
|
from django.utils.translation import gettext_lazy as _
|
||
|
|
|
||
|
|
from registration.models import StudentRegistration, CoachRegistration, AdminRegistration
|
||
|
|
|
||
|
|
|
||
|
|
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 StudentRegistrationForm(forms.ModelForm):
|
||
|
|
class Meta:
|
||
|
|
model = StudentRegistration
|
||
|
|
fields = ('student_class', 'school', 'give_contact_to_animath',)
|
||
|
|
|
||
|
|
|
||
|
|
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',)
|