summaryrefslogtreecommitdiff
path: root/yaksh/decorators.py
blob: 9e9bc6d9165163e443809f665926377d819aec26 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
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)
        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,
                                    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.
    """

    def is_email_verified(request, *args, **kwargs):
        ci = RequestContext(request)
        user = request.user
        context = {}
        if not settings.IS_DEVELOPMENT:
            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. \
                                        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