Pinry/pinry/core/views.py

from django.contrib.auth.models import Permission
from django.template.response import TemplateResponse
from django.http import HttpResponseRedirect
from django.core.urlresolvers import reverse
from django.contrib.auth.decorators import login_required
from django.contrib.auth import logout, authenticate, login
from django.contrib import messages
from django.conf import settings
from django.utils.functional import lazy
from django.views.generic import CreateView

from .forms import UserCreationForm
from .models import User


reverse_lazy = lambda name=None, *args : lazy(reverse, str)(name, args=args)


def home(request):
    return HttpResponseRedirect(reverse('pins:recent-pins'))


def private(request):
    return TemplateResponse(request, 'user/private.html', None)


class CreateUser(CreateView):
    template_name = 'user/register.html'
    model = User
    form_class = UserCreationForm
    success_url = reverse_lazy('pins:recent-pins')

    def get(self, request, *args, **kwargs):
        if not settings.ALLOW_NEW_REGISTRATIONS:
            messages.error(request, "The admin of this service is not allowing new registrations.")
            return HttpResponseRedirect(reverse('pins:recent-pins'))
        return super(CreateUser, self).get(request, *args, **kwargs)

    def form_valid(self, form):
        redirect = super(CreateUser, self).form_valid(form)
        permissions = Permission.objects.filter(codename__in=['add_pin', 'add_image'])
        user = authenticate(username=form.cleaned_data['username'],
                            password=form.cleaned_data['password'])
        user.user_permissions = permissions
        login(self.request, user)
        return redirect


@login_required
def logout_user(request):
    logout(request)
    messages.success(request, 'You have successfully logged out.')
    return HttpResponseRedirect(reverse('core:home'))