mirror of
https://github.com/pinry/pinry.git
synced 2025-11-13 08:35:41 +01:00
e70d9079ec
Rewritten register view using CBV approach, and added a authenticate/login call after form has been validated and user created. Fixes #65
54 lines
1.8 KiB
Python
54 lines
1.8 KiB
Python
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'))
|