Фильтр в Django Admin через поле ManyToMany с помощью Through

Мой models.py:

class Subject(models.Model):
    name = models.CharField(max_length=200)

class Person(models.Model):
    subject = models.ForeignKey(Subject, on_delete=models.CASCADE, blank=True, null=True)

class PersonRole(models.Model):
    project = models.ForeignKey('Project', on_delete=models.CASCADE)
    person = models.ForeignKey(Person, on_delete=models.CASCADE)

class Project(models.Model):

    title = models.CharField(max_length=200)
    person = models.ManyToManyField(Person, through=PersonRole)

Теперь в бэкэнд админки я хочу добавить красивую фильтрацию с помощью list_filter. Этот фильтр должен позволить фильтровать по школам, к которым человек прикреплен к проекту. Другими словами, если Джон (принадлежащий к школе '№ 1') прикреплен к проекту № 3, я хочу, чтобы в таблице проектов на бэкенде отображался только проект № 3. Полагаю, мне следует настроить simplelistfilter. Но прежде всего я немного запутался в том, как получить список школ, прикрепленных к лицам, прикрепленным к проекту.

Моя попытка на данный момент такова:

class PersonRole(models.Model):

    [...]

    def get_school(self):
        return self.person.school


class Project(models.Model):

    @admin.display(description='PI School')
    def get_PI_school(self):
        return [p for p in self.person.get_school()] 

admin.py:

class ProjectAdmin(admin.ModelAdmin):
    list_display = ("get_PI_school",) #This is just to see if the field is populated

При этом я получаю 'ManyRelatedManager' object has no attribute 'get_school'.

Вы используете .get_school() на объекте PersonRole:

class Project(models.Model):
    @admin.display(description='PI School')
    def get_PI_school(self):
        return [p.get_school() for p in self.personrole_set.all()]

Мы можем повысить эффективность, извлекая связанные Person и School в одном и том же запросе:

class Project(models.Model):
    @admin.display(description='PI School')
    def get_PI_school(self):
        return [
            p.get_school()
            for p in self.personrole_set.select_related('person__school')
        ]

и вы можете отфильтровать по роли PI с помощью:

class Project(models.Model):
    @admin.display(description='PI School')
    def get_PI_school(self):
        return [
            p.get_school()
            for p in self.personrole_set.filter(
                title__contains="Principal investigator"
            )
        ]

учитывая, конечно, что ваша PersonRole модель все еще имеет title поле.

После перебора различных вариантов мне удалось сделать это с помощью SimpleListFilter, и это оказалось проще, чем я думал! Мне даже не нужно определять properties и тому подобное в models.py. Вот код, который переходит в admin.py:

class PISchoolFilter(admin.SimpleListFilter):

   title = 'PI School'
   parameter_name = 'school'

   def lookups(self, request, model_admin):
        return (
                ('Arts & Humanities Admin',('Arts & Humanities Admin')),
    [...]
                )

   def queryset(self, request, queryset):
    if self.value() == 'Arts & Humanities Admin':
        return queryset.filter(
            person__personrole__person_role__contains="Principal investigator",
            person__personrole__person__school=5
        ).distinct()
    [...]
Вернуться на верх