pbp_com/apps/accounts/forms.py

from django import forms
from django.contrib.auth import authenticate, login
from django.contrib.auth.models import User
from django.contrib.sites.models import Site
from django.utils.translation import ugettext_lazy as _

from . import models
from . import profile as profile_utils
from . import constants
from . import utils

REQUIRED = (
    u"There was missing information on your order. Please correct " u"the errors below."
)


class BasePersonForm(forms.Form):
    """ There's a common theme of form data for person information. """

    first_name = forms.CharField(
        label=_(u"First name"),
        max_length=30,
        required=True,
        widget=forms.TextInput(attrs={"type": "text", "size": 25}),
    )

    last_name = forms.CharField(
        label=_(u"Last name"),
        max_length=30,
        required=True,
        widget=forms.TextInput(attrs={"type": "text", "size": 25}),
    )

    company = forms.CharField(
        label=_(u"Company"),
        max_length=30,
        required=False,
        widget=forms.TextInput(attrs={"type": "text", "size": 50}),
    )

    address1 = forms.CharField(
        label=_(u"Address 1"),
        max_length=30,
        required=True,
        widget=forms.TextInput(attrs={"type": "text", "size": 50}),
    )

    address2 = forms.CharField(
        label=_(u"Address 2"),
        max_length=30,
        required=False,
        widget=forms.TextInput(attrs={"type": "text", "size": 50}),
    )

    city = forms.CharField(
        label=_(u"City"),
        max_length=30,
        required=True,
        widget=forms.TextInput(attrs={"type": "text", "size": 50}),
    )

    region = forms.CharField(
        label=_(u"Region"),
        max_length=30,
        required=True,
        widget=forms.Select(
            choices=constants.STATE_CHOICES,
            attrs={"width": "100px", "onchange": "calculate_price();"},
        ),
    )

    country = forms.CharField(
        label=_(u"Country"),
        max_length=50,
        required=True,
        widget=forms.Select(
            choices=constants.COUNTRIES,
            attrs={"onchange": "check_state(); calculate_price();"},
        ),
    )

    country_other = forms.CharField(
        label=_(u"Other"),
        max_length=30,
        required=False,
        widget=forms.TextInput(attrs={"type": "text", "size": 25}),
    )

    postal_code = forms.CharField(
        label=_(u"Postal Code"),
        max_length=15,
        required=True,
        widget=forms.TextInput(
            attrs={"type": "text", "size": 15, "onchange": "calculate_price();"}
        ),
    )

    plus_four = forms.CharField(
        label=_(u"Plus Four"),
        max_length=15,
        required=False,
        widget=forms.TextInput(attrs={"type": "text", "size": 15}),
    )

    phone = forms.CharField(
        label=_(u"Phone"),
        max_length=30,
        required=True,
        widget=forms.TextInput(attrs={"type": "text", "size": 50}),
    )


class EmailRequiredPersonForm(BasePersonForm):
    """ In some cases, we need the email address to be required """

    email = forms.EmailField(
        label=_(u"Email"),
        max_length=30,
        required=True,
        widget=forms.TextInput(attrs={"type": "text", "size": 50}),
    )


class EmailNotRequiredPersonForm(BasePersonForm):
    """ In some cases, we need the email address to not be required """

    email = forms.EmailField(
        label=_(u"Email"),
        max_length=30,
        required=False,
        widget=forms.TextInput(attrs={"type": "text", "size": 50}),
    )


class NonRequiredBasePersonForm(forms.Form):

    first_name = forms.CharField(
        label=_(u"First name"),
        max_length=30,
        required=False,
        widget=forms.TextInput(attrs={"type": "text", "size": 25}),
    )

    last_name = forms.CharField(
        label=_(u"Last name"),
        max_length=30,
        required=False,
        widget=forms.TextInput(attrs={"type": "text", "size": 25}),
    )

    company = forms.CharField(
        label=_(u"Company"),
        max_length=30,
        required=False,
        widget=forms.TextInput(attrs={"type": "text", "size": 50}),
    )

    address1 = forms.CharField(
        label=_(u"Address 1"),
        max_length=30,
        required=False,
        widget=forms.TextInput(attrs={"type": "text", "size": 50}),
    )

    address2 = forms.CharField(
        label=_(u"Address 2"),
        max_length=30,
        required=False,
        widget=forms.TextInput(attrs={"type": "text", "size": 50}),
    )

    city = forms.CharField(
        label=_(u"City"),
        max_length=30,
        required=False,
        widget=forms.TextInput(attrs={"type": "text", "size": 50}),
    )

    region = forms.ChoiceField(
        label=_(u"Region"),
        required=False,
        choices=constants.STATE_CHOICES,
        widget=forms.Select(
            choices=constants.STATE_CHOICES, attrs={"onchange": "calculate_price();"}
        ),
    )

    country = forms.ChoiceField(
        label=_(u"Country"),
        required=False,
        choices=(("", "----------"),) + constants.COUNTRIES,
        widget=forms.Select(
            choices=constants.STATE_CHOICES,
            attrs={"onchange": "check_state(); calculate_price();"},
        ),
    )

    country_other = forms.CharField(
        label=_(u"Other"),
        max_length=30,
        required=False,
        widget=forms.TextInput(attrs={"type": "text", "size": 25}),
    )

    postal_code = forms.CharField(
        label=_(u"Postal Code"),
        max_length=15,
        required=False,
        widget=forms.TextInput(
            attrs={"type": "text", "size": 15, "onchange": "calculate_price();"}
        ),
    )

    plus_four = forms.CharField(
        label=_(u"Plus Four"),
        max_length=4,
        required=False,
        widget=forms.TextInput(attrs={"type": "text", "size": 15}),
    )

    phone = forms.CharField(
        label=_(u"Phone"),
        max_length=30,
        required=False,
        widget=forms.TextInput(attrs={"type": "text", "size": 50}),
    )


class ProfileCreateForm(EmailRequiredPersonForm):
    """ Form used for creating a new user """

    password1 = forms.CharField(
        label=_("Password"), widget=forms.PasswordInput, required=True
    )

    password2 = forms.CharField(
        label=_("Password confirmation"), widget=forms.PasswordInput, required=True
    )

    def save(self, request=None):
        """ Creates a new user and account. Returns the newly created user. """
        password = self.cleaned_data["password1"]
        username, email, password = (
            self.cleaned_data["email"],
            self.cleaned_data["email"],
            password,
        )
        user = User.objects.create_user(username, email, password)
        user.first_name = self.cleaned_data["first_name"]
        user.last_name = self.cleaned_data["last_name"]
        user.save()

        profile = profile_utils.get_profile(user)
        profile.company = self.cleaned_data["company"]
        profile.address1 = self.cleaned_data["address1"]
        profile.address2 = self.cleaned_data["address2"]
        profile.city = self.cleaned_data["city"]
        profile.region = self.cleaned_data["region"]
        profile.phone = self.cleaned_data["phone"]
        profile.country = self.cleaned_data["country"]
        profile.postal_code = self.cleaned_data["postal_code"]
        profile.plus_four = self.cleaned_data["plus_four"]
        profile.save()

        user = authenticate(username=username, password=password)
        if request:
            login(request, user)
            site = Site.objects.get_current()
            utils.send_new_account_email(email, password, user, profile, site)

        return user

    def clean(self):
        # We want a single error message.  We don't display the validation
        # errors for each part of the form
        for field in self.fields:
            if self.fields[field].required and not self.cleaned_data.get(field, None):
                raise forms.ValidationError(_(REQUIRED))
        self.clean_email()

        return self.cleaned_data

    def clean_email(self):
        """ Validate that the e-mail address is unique. """
        if User.objects.filter(
            email__iexact=self.cleaned_data["email"]
        ) or User.objects.filter(username__iexact=self.cleaned_data["email"]):
            raise forms.ValidationError(
                _("This email is already in use. " "Please supply a different email.")
            )
        return self.cleaned_data["email"]

    def clean_password2(self):
        password1 = self.cleaned_data.get("password1", "")
        password2 = self.cleaned_data["password2"]
        if password1 != password2:
            raise forms.ValidationError(
                "Your passwords did not match. Please re-enter your passwords."
            )
        return password2


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

    def __init__(self, user, *args, **kwargs):
        self.user = user
        super(ProfileUpdateForm, self).__init__(*args, **kwargs)

    def clean_email(self):
        """ Validate that the email is not already registered with another user
        """
        if User.objects.filter(email__iexact=self.cleaned_data["email"]).exclude(
            email__iexact=self.user.email
        ):
            raise forms.ValidationError(
                _(u"This email is already in use. Please supply a different " u"email.")
            )
        return self.cleaned_data["email"]

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

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

        profile = profile_utils.get_profile(user)
        profile.company = self.cleaned_data["company"]
        profile.address1 = self.cleaned_data["address1"]
        profile.address2 = self.cleaned_data["address2"]
        profile.city = self.cleaned_data["city"]
        profile.region = self.cleaned_data["region"]
        profile.phone = self.cleaned_data["phone"]
        profile.country = self.cleaned_data["country"]
        profile.country_other = self.cleaned_data["country_other"]
        profile.postal_code = self.cleaned_data["postal_code"]
        profile.plus_four = self.cleaned_data["plus_four"]
        profile.save()

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


class ProfileSecondaryAddressForm(NonRequiredBasePersonForm):

    address_from = forms.DateField(required=False)
    address_to = forms.DateField(required=False)

    class Meta:
        model = models.SecondaryProfileAddress
        exclude = "profile"

    def __init__(self, *args, **kw):
        super(ProfileSecondaryAddressForm, self).__init__(*args, **kw)

    def save(self, *args, **kwargs):
        user = kwargs.pop("user", None)
        profile_address = profile_utils.get_profile_secondary_address(user)
        profile_address.profile = profile_utils.get_profile(user)
        for k, v in self.cleaned_data.iteritems():
            setattr(profile_address, k, v)
        profile_address.save()
        return profile_address