From ceb55baf69c2f5f7346855ee5a6e5e9f77456fcb Mon Sep 17 00:00:00 2001
From: ankitjavalkar
Date: Fri, 21 Jul 2017 17:07:24 +0530
Subject: Add a has_profile decorator

---
 yaksh/decorators.py | 37 ++++++++++++++++++++++++++++++++-----
 1 file changed, 32 insertions(+), 5 deletions(-)

(limited to 'yaksh/decorators.py')

diff --git a/yaksh/decorators.py b/yaksh/decorators.py
index f0d354c..8e5f879 100644
--- a/yaksh/decorators.py
+++ b/yaksh/decorators.py
@@ -1,12 +1,39 @@
-from django.shortcuts import render_to_response
+from django.shortcuts import render_to_response, redirect
 from django.conf import settings
 from django.template import RequestContext
 
+# Local imports
+from yaksh.forms import ProfileForm
+
+
+def user_has_profile(user):
+    return hasattr(user, 'profile')
+
+
+def has_profile(func):
+    """
+    This decorator is used to check if the user account has a profile.
+    If the user does not have a profile then redirect the user to
+    profile edit page.
+    """
+
+    def _wrapped_view(request, *args, **kwargs):
+        if user_has_profile(request.user):
+            return func(request, *args, **kwargs)
+        ci = RequestContext(request)
+        template = 'manage.html' if 'moderator' in request.user.groups.all() else 'user.html'
+        form = ProfileForm(user=request.user, instance=None)
+        context = {'template': template, 'form': form}
+        return render_to_response('yaksh/editprofile.html', context,
+                                    context_instance=ci)
+    return _wrapped_view
+
 
 def email_verified(func):
-    """ This decorator is used to check if email is verified. 
-        If email is not verified then redirect user for email
-        verification
+    """
+    This decorator is used to check if email is verified.
+    If email is not verified then redirect user for email
+    verification.
     """
 
     def is_email_verified(request, *args, **kwargs):
@@ -14,7 +41,7 @@ def email_verified(func):
         user = request.user
         context = {}
         if not settings.IS_DEVELOPMENT:
-            if user.is_authenticated() and hasattr(user, 'profile'):
+            if user.is_authenticated() and user_has_profile(user):
                 if not user.profile.is_email_verified:
                     context['success'] = False
                     context['msg'] = "Your account is not verified. \
-- 
cgit 


From 281e28819d4ab62cc01722d90dd4951e417e16cb Mon Sep 17 00:00:00 2001
From: ankitjavalkar
Date: Thu, 3 Aug 2017 15:44:39 +0530
Subject: Add review changes and more tests

---
 yaksh/decorators.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'yaksh/decorators.py')

diff --git a/yaksh/decorators.py b/yaksh/decorators.py
index 8e5f879..9e9bc6d 100644
--- a/yaksh/decorators.py
+++ b/yaksh/decorators.py
@@ -21,7 +21,10 @@ def has_profile(func):
         if user_has_profile(request.user):
             return func(request, *args, **kwargs)
         ci = RequestContext(request)
-        template = 'manage.html' if 'moderator' in request.user.groups.all() else 'user.html'
+        if request.user.groups.filter(name='moderator').exists():
+            template = 'manage.html'
+        else:
+            template = 'user.html'
         form = ProfileForm(user=request.user, instance=None)
         context = {'template': template, 'form': form}
         return render_to_response('yaksh/editprofile.html', context,
-- 
cgit