From 28b0d40bb88542c6f0f2c58603696fca8b1c07e8 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Stefan=20K=C3=B6gl?= Date: Thu, 28 Jul 2011 16:06:38 +0000 Subject: [PATCH] @requires_token returns 404 for unknown username --- mygpo/decorators.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/mygpo/decorators.py b/mygpo/decorators.py index b9d0d2f6..c2e86b14 100644 --- a/mygpo/decorators.py +++ b/mygpo/decorators.py @@ -17,6 +17,7 @@ # along with this program. If not, see . # +from django.http import Http404 from django.shortcuts import render_to_response, get_object_or_404 from django.template import RequestContext from django.contrib.auth.models import User @@ -42,7 +43,12 @@ def requires_token(token_name, denied_template=None): def decorator(fn): def tmp(request, username, *args, **kwargs): from mygpo import migrate - user = User.objects.get(username=username) + + try: + user = User.objects.get(username=username) + except User.DoesNotExist: + raise Http404 + user = migrate.get_or_migrate_user(user) token = getattr(user, token_name, '') -- 2.11.4.GIT