from django import forms
from django.utils.translation import gettext_lazy as _
from allauth.account.forms import SignupForm


class CustomSignupForm(SignupForm):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

        # Add description on email field
        self.fields["email"].help_text = _(
            "Please enter a valid email address ending with `@crans.org` or "
            "`@ens-paris-saclay.fr`."
        )

    def clean_email(self):
        """
        Check that the email address ends with a trusted domain.
        """
        email = self.cleaned_data.get("email")
        if not email.endswith("@crans.org") and not email.endswith("@ens-paris-saclay.fr"):
            raise forms.ValidationError(
                _("Must end with `@crans.org` or `@ens-paris-saclay.fr`.")
            )
        return email