def sitefilter(context):
site_filter = get_current_site_filter(context['request'])
return {
'object_list': Site.objects.all(),
'selected': site_filter.sites.all()
}
def queryset(self, request):
queryset = super(SiteFilterMixin, self).queryset(request)
sites = get_current_site_filter(request).sites.all()
if sites:
try:
return queryset.filter(site__in=sites).distinct()
except FieldError:
return queryset.filter(sites__in=sites).distinct()
return queryset
def manage(request, id):
if not request.user.is_staff:
raise Http404
sitefilter = get_current_site_filter(request)
site = Site.objects.get(id=id)
if site in sitefilter.sites.all():
sitefilter.sites.remove(site)
else:
sitefilter.sites.add(site)
return HttpResponseRedirect(request.META['HTTP_REFERER'])
