Фильтр в 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()
[...]