Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Django admin queryset filtering by foreign key backwards relation

I have models A and B, where B has a FK to A.

I use django 1.3 and I need two django admin filters:

1) a.b_set.exists() # (True/False)

2) not a.b_set.filter(some_condition=False).exists() # (True/False)

How can I achieve that? Sadly, I couldn't find any solutions by googling.

like image 765
aemdy Avatar asked Nov 30 '12 12:11

aemdy


2 Answers

You need to read this: Custom Filter in Django Admin on Django 1.3 or below

This is my first attempt without any testing, but you should see more or less how its done -

from django.db import models
from django.contrib.admin.filterspecs import FilterSpec, ChoicesFilterSpec
from django.utils.encoding import smart_unicode
from django.utils.translation import ugettext as _

class BNullSetFilterSpec(FilterSpec):

    def __init__(self, f, request, params, model, model_admin):
        super(BSetFilterSpec, self).__init__(f, request, params, model, model_admin)

        self.links = (
            ('Yes', {'b__isnull': False}),
            ('No', {}))

    def title(self):
        return _('B Set')

# registering the filter
FilterSpec.filter_specs.insert(0, (lambda f: getattr(f, 'empty_bset', False), BNullSetFilterSpec))
like image 153
Francis Yaconiello Avatar answered Oct 05 '22 23:10

Francis Yaconiello


What @ptrck mentioned is right. Couldn't you switch to 1.4 ?

In this case, as in the doc described, without need to change the lookups, do something like (for the first here and then the second accordingly):

from django.contrib.admin import BooleanFieldListFilter

class BInA(BooleanFieldListFilter):
    def queryset(self, request, queryset):
        if self.value() is True:
            return queryset.filter(a.b_set.exists())
        else:
            return queryset.filter(a.b_set.exists() is False)
like image 39
Rmatt Avatar answered Oct 05 '22 23:10

Rmatt