Redirect successful login to profile page if logged in from activation complete.

1 files changed, 7 insertions, 2 deletions

diff --git a/pytask/profile/views.py b/pytask/profile/views.py
index a84531d..b4fce21 100755
--- a/pytask/profile/views.py
+++ b/pytask/profile/views.py
@@ -204,12 +204,17 @@ def login_proceed(request):
 
     # Check if the request came from logout page, if so set
     # authentication to redirect to home page
-    if reverse('auth_logout') == urlparse.urlsplit(
-      request.META['HTTP_REFERER'])[2]:
+    referer_path = urlparse.urlsplit(request.META['HTTP_REFERER'])[2]
+    if referer_path == reverse('auth_logout'):
       response = {
         'authentication': 'success',
         'redirect': reverse('home_page'),
         }
+    elif referer_path == reverse('registration_activation_complete'):
+      response = {
+        'authentication': 'success',
+        'redirect': reverse('view_profile'),
+        }
     else:
         response = {
           'authentication': 'success',