from django import forms
from django.utils.translation import gettext_lazy as _

from userena.forms import SignupForm


class SignupFormExtra(SignupForm):
    """
    A form to demonstrate how to add extra fields to the signup form, in this
    case adding the first and last name.


    """

    first_name = forms.CharField(
        label=_("First name"), max_length=30, required=False
    )

    last_name = forms.CharField(
        label=_("Last name"), max_length=30, required=False
    )

    def __init__(self, *args, **kw):
        """

        A bit of hackery to get the first name and last name at the top of the
        form instead at the end.

        """
        super().__init__(*args, **kw)
        # Put the first and last name at the top
        new_order = self.fields.keyOrder[:-2]
        new_order.insert(0, "first_name")
        new_order.insert(1, "last_name")
        self.fields.keyOrder = new_order

    def save(self):
        """
        Override the save method to save the first and last name to the user
        field.

        """
        # First save the parent form and get the user.
        new_user = super().save()

        new_user.first_name = self.cleaned_data["first_name"]
        new_user.last_name = self.cleaned_data["last_name"]
        new_user.save()

        # Userena expects to get the new user from this form, so return the new
        # user.
        return new_user
