mirror of
https://github.com/pinry/pinry.git
synced 2025-11-15 09:25:50 +01:00
53f05dbb6d
Removed pins django app, and moved code to the core. Moved user related code out of core to the users app.
44 lines
1.6 KiB
Python
44 lines
1.6 KiB
Python
from django.conf import settings
|
|
from django.contrib import messages
|
|
from django.contrib.auth import authenticate, login, logout
|
|
from django.contrib.auth.decorators import login_required
|
|
from django.contrib.auth.models import Permission
|
|
from django.core.urlresolvers import reverse
|
|
from django.http import HttpResponseRedirect
|
|
from django.utils.functional import lazy
|
|
from django.views.generic import CreateView
|
|
|
|
from .forms import UserCreationForm
|
|
from pinry.users.models import User
|
|
|
|
|
|
reverse_lazy = lambda name=None, *args: lazy(reverse, str)(name, args=args)
|
|
|
|
|
|
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')) |