def get_sites(self, perm='bluusites.view_bluusite'):
from bluusites.models import BluuSite
from grontextual.shortcuts import get_objects_for_user
if self.has_perm(perm):
return BluuSite.objects.all()
return get_objects_for_user(self, perm)
def get_queryset(self):
if self.request.user.has_perm('companies.view_company'):
return super(CompanyListView, self).get_queryset()
return get_objects_for_user(self.request.user, 'companies.view_company')
def get_companies(self, perm='companies.view_company'):
from companies.models import Company
from grontextual.shortcuts import get_objects_for_user
if self.has_perm(perm):
return Company.objects.all()
return get_objects_for_user(self, perm)
