X-Git-Url: https://code.kerkeslager.com/?p=climbing.kerkeslager.com;a=blobdiff_plain;f=src%2Fuser_profile%2Fviews.py;fp=src%2Fuser_profile%2Fviews.py;h=39557a9de6c652d5ab7f9b50d85edc1b05237e0e;hp=41498b63981b403b1de3d3548709b13d0bf7c30b;hb=2db9b400dc1dbd5d0e1f3fa829d3d9029540b051;hpb=4ea2032901c06985e8616e16e7d0bba6bb3c9a43 diff --git a/src/user_profile/views.py b/src/user_profile/views.py index 41498b6..39557a9 100644 --- a/src/user_profile/views.py +++ b/src/user_profile/views.py @@ -1,4 +1,5 @@ from django.contrib.auth.models import User +from django.http import Http404 from django.shortcuts import get_object_or_404 from django.views.generic.detail import DetailView @@ -6,6 +7,14 @@ class UserProfileDetailView(DetailView): model = User def get_object(self): - return get_object_or_404(User, username=self.kwargs['username']) + username = self.kwargs.get('username') + + if username: + return get_object_or_404(User, username=username) + + if self.request.user.is_authenticated: + return self.request.user + + raise Http404() user_profile_detail = UserProfileDetailView.as_view()