Mercurial > public > sg101
view accounts/views.py @ 1037:7e0c3cbd3cda
Fix bad select_related call.
In Django 1.8, select_related now throws an error if you give it an invalid
field. This was happening. Fix that query.
Also noticed an extra query generated in the display_post template. Fixed.
author | Brian Neal <bgneal@gmail.com> |
---|---|
date | Tue, 29 Dec 2015 22:21:42 -0600 |
parents | be233ba7ca31 |
children | 6dd1f0065859 |
line wrap: on
line source
""" Views for the accounts application. """ import json import logging from django.http import HttpResponse from django.shortcuts import render, redirect from django.conf import settings from accounts.models import PendingUser from accounts.forms import RegisterForm from accounts.forms import RegisterCodeForm from accounts.forms import ForgotUsernameForm from accounts import create_new_user from antispam.decorators import log_auth_failures logger = logging.getLogger('auth') ####################################################################### @log_auth_failures('Register') def register(request): if request.user.is_authenticated(): return redirect(settings.LOGIN_REDIRECT_URL) if request.method == 'POST': form = RegisterForm(request.POST, ip=request.META.get('REMOTE_ADDR', '?')) if form.is_valid(): form.save(request) return redirect('accounts-register1') else: form = RegisterForm() return render(request, 'accounts/register.html', {'form': form}) ####################################################################### def register1(request): """Displays the registration code.""" if request.user.is_authenticated(): return redirect(settings.LOGIN_REDIRECT_URL) return render(request, 'accounts/register1.html') ####################################################################### @log_auth_failures('Register') def register2(request): """Processes the registration code and creates the user.""" if request.user.is_authenticated(): return redirect(settings.LOGIN_REDIRECT_URL) if request.method == 'POST': form = RegisterCodeForm(request.POST, session=request.session, ip=request.META.get('REMOTE_ADDR', '?')) if form.is_valid(): form.save() return redirect('accounts-register_thanks') else: form = RegisterCodeForm() return render(request, 'accounts/register2.html', {'form': form}) ####################################################################### def get_code(request): code = {'code': 'FAIL-123'} reg_info = request.session.get('reg_info') if reg_info: code['code'] = reg_info.get('code', code['code']) return HttpResponse(json.dumps(code), content_type='application/json') ####################################################################### def register_thanks(request): if request.user.is_authenticated(): return redirect(settings.LOGIN_REDIRECT_URL) return render(request, 'accounts/register_thanks.html') ####################################################################### def register_confirm(request, username, key): if request.user.is_authenticated(): return redirect(settings.LOGIN_REDIRECT_URL) # purge expired users PendingUser.objects.purge_expired() ip = request.META.get('REMOTE_ADDR', '?') try: pending_user = PendingUser.objects.get(username = username) except PendingUser.DoesNotExist: logger.error('Accounts register_confirm [%s]: user does not exist: %s', ip, username) return render(request, 'accounts/register_failure.html', {'username': username}) if pending_user.key != key: logger.error('Accounts register_confirm [%s]: key error: %s', ip, username) return render(request, 'accounts/register_failure.html', {'username': username}) create_new_user(pending_user, ip) return render(request, 'accounts/register_success.html', {'username': username}) ####################################################################### def username_query(request): """This view handles forgotten username queries.""" if request.user.is_authenticated(): return redirect(settings.LOGIN_REDIRECT_URL) if request.method == 'POST': form = ForgotUsernameForm(data=request.POST) if form.is_valid(): form.save() return redirect('accounts-username_sent') else: form = ForgotUsernameForm() return render(request, 'accounts/username_query.html', {'form': form})