Sith/core/forms.py

from django.contrib.auth.forms import UserCreationForm, AuthenticationForm
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 PageForm(forms.ModelForm):
    #parent = forms.ModelChoiceField(queryset=Page.objects.all())
    #def __init__(self, *args, **kwargs):
    #    super(PageForm, self).__init__(*args, **kwargs)

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

    def save(self, commit=True):
        page = super(PageForm, self).save(commit=False)
        if self.cleaned_data['parent'] is not None:
            page.full_name = '/'.join([self.cleaned_data['parent'].full_name, self.cleaned_data['name']])
        else:
            page.full_name = self.cleaned_data['name']
        page.name = self.cleaned_data['name'].split('/')[-1]
        if commit:
            page.save()
        return page