168 lines
4.8 KiB
Python
168 lines
4.8 KiB
Python
from corres2math.tokens import email_validation_token
|
|
from django.contrib.sites.models import Site
|
|
from django.db import models
|
|
from django.template import loader
|
|
from django.utils.crypto import get_random_string
|
|
from django.utils.encoding import force_bytes
|
|
from django.utils.http import urlsafe_base64_encode
|
|
from django.utils.translation import gettext_lazy as _
|
|
from polymorphic.models import PolymorphicModel
|
|
|
|
|
|
class Registration(PolymorphicModel):
|
|
user = models.OneToOneField(
|
|
"auth.User",
|
|
on_delete=models.CASCADE,
|
|
verbose_name=_("user"),
|
|
)
|
|
|
|
give_contact_to_animath = models.BooleanField(
|
|
default=False,
|
|
verbose_name=_("Grant Animath to contact me in the future about other actions"),
|
|
)
|
|
|
|
email_confirmed = models.BooleanField(
|
|
default=False,
|
|
verbose_name=_("email confirmed"),
|
|
)
|
|
|
|
def send_email_validation_link(self):
|
|
subject = "[Corres2math] " + str(_("Activate your Correspondances account"))
|
|
token = email_validation_token.make_token(self.user)
|
|
uid = urlsafe_base64_encode(force_bytes(self.user.pk))
|
|
site = Site.objects.first()
|
|
message = loader.render_to_string('registration/mails/email_validation_email.txt',
|
|
{
|
|
'user': self.user,
|
|
'domain': site.domain,
|
|
'token': token,
|
|
'uid': uid,
|
|
})
|
|
html = loader.render_to_string('registration/mails/email_validation_email.html',
|
|
{
|
|
'user': self.user,
|
|
'domain': site.domain,
|
|
'token': token,
|
|
'uid': uid,
|
|
})
|
|
self.user.email_user(subject, message, html_message=html)
|
|
|
|
@property
|
|
def type(self):
|
|
raise NotImplementedError
|
|
|
|
@property
|
|
def form_class(self):
|
|
raise NotImplementedError
|
|
|
|
@property
|
|
def participates(self):
|
|
return isinstance(self, StudentRegistration) or isinstance(self, CoachRegistration)
|
|
|
|
@property
|
|
def is_admin(self):
|
|
return isinstance(self, AdminRegistration) or self.user.is_superuser
|
|
|
|
def __str__(self):
|
|
return f"{self.user.first_name} {self.user.last_name}"
|
|
|
|
class Meta:
|
|
verbose_name = _("registration")
|
|
verbose_name_plural = _("registrations")
|
|
|
|
|
|
def get_random_filename(instance, filename):
|
|
return "authorization/photo/" + get_random_string(64)
|
|
|
|
|
|
class StudentRegistration(Registration):
|
|
team = models.ForeignKey(
|
|
"participation.Team",
|
|
related_name="students",
|
|
on_delete=models.PROTECT,
|
|
null=True,
|
|
default=None,
|
|
verbose_name=_("team"),
|
|
)
|
|
|
|
student_class = models.IntegerField(
|
|
choices=[
|
|
(12, _("12th grade")),
|
|
(11, _("11th grade")),
|
|
(10, _("10th grade or lower")),
|
|
],
|
|
verbose_name=_("student class"),
|
|
)
|
|
|
|
school = models.CharField(
|
|
max_length=255,
|
|
verbose_name=_("school"),
|
|
)
|
|
|
|
photo_authorization = models.FileField(
|
|
verbose_name=_("photo authorization"),
|
|
upload_to=get_random_filename,
|
|
blank=True,
|
|
default="",
|
|
)
|
|
|
|
@property
|
|
def type(self):
|
|
return _("student")
|
|
|
|
@property
|
|
def form_class(self):
|
|
from registration.forms import StudentRegistrationForm
|
|
return StudentRegistrationForm
|
|
|
|
class Meta:
|
|
verbose_name = _("student registration")
|
|
verbose_name_plural = _("student registrations")
|
|
|
|
|
|
class CoachRegistration(Registration):
|
|
team = models.ForeignKey(
|
|
"participation.Team",
|
|
related_name="coachs",
|
|
on_delete=models.PROTECT,
|
|
null=True,
|
|
default=None,
|
|
verbose_name=_("team"),
|
|
)
|
|
|
|
professional_activity = models.TextField(
|
|
verbose_name=_("professional activity"),
|
|
)
|
|
|
|
@property
|
|
def type(self):
|
|
return _("coach")
|
|
|
|
@property
|
|
def form_class(self):
|
|
from registration.forms import CoachRegistrationForm
|
|
return CoachRegistrationForm
|
|
|
|
class Meta:
|
|
verbose_name = _("coach registration")
|
|
verbose_name_plural = _("coach registrations")
|
|
|
|
|
|
class AdminRegistration(Registration):
|
|
role = models.TextField(
|
|
verbose_name=_("role of the administrator"),
|
|
)
|
|
|
|
@property
|
|
def type(self):
|
|
return _("admin")
|
|
|
|
@property
|
|
def form_class(self):
|
|
from registration.forms import AdminRegistrationForm
|
|
return AdminRegistrationForm
|
|
|
|
class Meta:
|
|
verbose_name = _("admin registration")
|
|
verbose_name_plural = _("admin registrations")
|