diff options
| -rw-r--r-- | pytask/profile/utils.py | 7 |
1 files changed, 2 insertions, 5 deletions
diff --git a/pytask/profile/utils.py b/pytask/profile/utils.py index e13bf60..745901c 100644 --- a/pytask/profile/utils.py +++ b/pytask/profile/utils.py @@ -1,3 +1,4 @@ +from django import shortcuts from django.http import Http404 from django.contrib.auth.models import User from pytask.profile.models import Notification @@ -35,13 +36,9 @@ def get_notification(nid, user): def get_user(uid): - try: - user = User.objects.get(id=uid) - except User.DoesNotExist: - raise Http404 + user = shortcuts.get_object_or_404(User, pk=uid) if user.is_active: return user else: raise Http404 - |