Sith/core/forms.py

from django.contrib.auth.forms import UserCreationForm, AuthenticationForm, UserChangeForm
from django import forms
from django.contrib.auth import logout, login, authenticate
import logging

from .models import User, Page

class RegisteringForm(UserCreationForm):
    error_css_class = 'error'
    required_css_class = 'required'
    class Meta:
        model = User
        fields = ('first_name', 'last_name', 'email', 'date_of_birth')

    def save(self, commit=True):
        user = super(RegisteringForm, self).save(commit=False)
        user.set_password(self.cleaned_data["password1"])
        user.generate_username()
        if commit:
            user.save()
        return user

class LoginForm(AuthenticationForm):
    def login(self):
        u = authenticate(username=self.request.POST['username'],
                         password=self.request.POST['password'])
        if u is not None:
            if u.is_active:
                login(self.request, u)
                logging.debug("Logging in "+str(u))
            else:
                raise forms.ValidationError(
                        self.error_messages['invalid_login'],
                        code='inactive',
                        params={'username': self.username_field.verbose_name},
                    )
        else:
            logging.debug("Login failed")
            raise forms.ValidationError(
                    self.error_messages['invalid_login'],
                    code='invalid_login',
                    params={'username': self.username_field.verbose_name},
                )

class EditUserForm(UserChangeForm):
    error_css_class = 'error'
    required_css_class = 'required'
    class Meta:
        model = User
        fields = ('first_name', 'last_name', 'email', 'date_of_birth', 'groups', 'user_permissions')


class PagePropForm(forms.ModelForm):
    error_css_class = 'error'
    required_css_class = 'required'
    parent = forms.ModelChoiceField(queryset=Page.objects.all())

    def __init__(self, *args, **kwargs):
        super(PagePropForm, self).__init__(*args, **kwargs)
        self.fields['parent'].required = False

    class Meta:
        model = Page
        fields = ['parent', 'name', 'owner_group', 'edit_group', 'view_group']

    def save(self, commit=True):
        page = super(PagePropForm, self).save(commit=False)
        if commit:
            page.save()
        return page


class PageEditForm(forms.ModelForm):
    error_css_class = 'error'
    required_css_class = 'required'

    def __init__(self, *args, **kwargs):
        super(PageEditForm, self).__init__(*args, **kwargs)

    class Meta:
        model = Page
        fields = ['title', 'content', ]

    def save(self, commit=True):
        page = super(PageEditForm, self).save(commit=False)
        if commit:
            page.save()
        return page