Add email_verified decorator to check whether email is verified

1 files changed, 25 insertions, 0 deletions

diff --git a/yaksh/decorators.py b/yaksh/decorators.py
new file mode 100644
index 0000000..f0d354c
--- /dev/null
+++ b/yaksh/decorators.py
@@ -0,0 +1,25 @@
+from django.shortcuts import render_to_response
+from django.conf import settings
+from django.template import RequestContext
+
+
+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
+    """
+
+    def is_email_verified(request, *args, **kwargs):
+        ci = RequestContext(request)
+        user = request.user
+        context = {}
+        if not settings.IS_DEVELOPMENT:
+            if user.is_authenticated() and hasattr(user, 'profile'):
+                if not user.profile.is_email_verified:
+                    context['success'] = False
+                    context['msg'] = "Your account is not verified. \
+                                        Please verify your account"
+                    return render_to_response('yaksh/activation_status.html',
+                                                context, context_instance=ci)
+        return func(request, *args, **kwargs)
+    return is_email_verified
\ No newline at end of file