Просмотр исходного кода

Added exception handling to profile view decorator. #37

Rafał Pitoń 11 лет назад
Родитель
Сommit
847a57c53f
1 измененных файлов с 5 добавлено и 1 удалено
  1. 5 1
      misago/apps/profiles/decorators.py

+ 5 - 1
misago/apps/profiles/decorators.py

@@ -2,6 +2,7 @@ from functools import wraps
 from django.conf import settings
 from django.core.urlresolvers import reverse
 from django.shortcuts import redirect
+from misago.acl.exceptions import ACLError403, ACLError404
 from misago.apps.errors import error404
 from misago.models import User
 from misago.utils.strings import slugify
@@ -23,7 +24,10 @@ def profile_view(fallback='user'):
                 return f(request, user, *args, **kwargs)
             except User.DoesNotExist:
                 return error404(request)
-    
+            except ACLError404:
+                return error404(request)
+            except ACLError403 as e:
+                return error404(request, e.message)
         return wraps(f)(inner_decorator)
     return outer_decorator