Nicer display of currently logged-in user

[climbing.kerkeslager.com] / src / user_profile / views.py

diff --git a/src/user_profile/views.py b/src/user_profile/views.py
index 41498b6..39557a9 100644 (file)
--- 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()