How do I change a Django Template based on the User's Group?

Right now I have two different groups of users on my site: customers and businesses.

Right now I am only using one login that allows both user groups to see their profile page.

However, there are portions of the profile page I only want the Customer to see and portions I only want the business to see. How can I 开发者_高级运维go about limiting what each group sees on this page?

Should I do it in the template with some sort of if statement? or is there some other solution anyone can let me know about?

If you want to avoid adding anything to your view functions, and you’re using the auth context processor (django.contrib.auth.context_processors.auth) and RequestContext as per @thyagx’s answer, then you could use a template snippet like that suggested in this Google Groups post:

{% for group in user.groups.all %}
    {% if group.name == 'customers' %}
        {% comment %}Customer-specific code goes here{% endcomment %}
    {% endif %}
{% endfor %}

It’s a bit verbose, but it does mean you don’t have to do anything in your view (aside from using RequestContext), or write a custom context processor.

This is probably too old for you to care any more, but I stumbled here myself before figuring it out on my own. For posterity, I found the following solution:

In your view, add something like this:

is_customer = request.user.groups.filter(name='Customers').exists()

In your template:
{% if is_customer %} customer stuff here {% endif %}

It relies on the fact that an if clause in a template will be evaluate to false for an empty list.

What I did to solve this is:

1. I created a custom context processor which basically inserts new variables for you to use in your templates and added it to my settings. See more @ https://docs.djangoproject.com/en/1.3/ref/templates/api/#django.template.RequestContext:

   TEMPLATE_CONTEXT_PROCESSORS = (
       'django.core.context_processors.debug',
       'django.core.context_processors.i18n',
       'django.core.context_processors.media',
       'django.core.context_processors.static',
       'django.contrib.auth.context_processors.auth',
       'django.contrib.messages.context_processors.messages',
       'common.user_context.user_context'
   )
   

2. I went on to write the function user_context inside the file user_context, mine is like so:

   def user_context(request):
       if request.user.is_authenticated():
           is_admin = is_local_admin(request.user)
       else:
           is_admin = False
   
       return {
           'is_local_admin': is_admin
       }
   

   is_local_admin is just a function that checks if the user belongs to the Admins group or not.

3. Whenever I need this is_local_admin information in my template I use this to render it in my view, for example:

   return render_to_response('create_user.html', {
       'form': form
   }, context_instance=RequestContext(request))
   

The important part is the RequestContext, which loads the custom context processor we build in step 1.

Now in your template you can just use:

{% if is_local_admin %}
    <h1>You can see this</h1>
{% else %}
    <h1>Nothing here</h1>
{% endif %}

Hope this helps someone. In summary: take a look at custom context processors, they are worth the read.

I implemented this through a template tag, based on what I found here. Perhaps it can be useful to someone.

In utils/utils_extras.py:

from django import template
from django.contrib.auth.models import Group

register = template.Library()
@register.filter(name='has_group')
def has_group(user, group_name):
    try:
        group = Group.objects.get(name=group_name)
    except:
        return False  # group doesn't exist, so for sure the user isn't part of the group

    # for superuser or staff, always return True
    if user.is_superuser or user.is_staff:
        return True

    return user.groups.filter(name=group_name).exists()

Then in the template itself:

{% load utils_extras %}
{% if user|has_group:"mygroup" %}

{% if  'group_name' in user.groups.get.name %} do smth {% endif %}