Фильтрация в администраторе Django через поле 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)
school = models.ForeignKey(School, 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.
Полагаю, мне следует настроить простой фильтр списка. Однако, прежде всего, я немного запутался в том, как получить список школ, прикрепленных к лицам, прикрепленным к проекту.
Моя попытка на данный момент:
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'.
Ответы 2
Вы используете .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.
но что ModelAdmin ты используешь? Для Project или ProjectRole? Обратите внимание, что мы используем p.get_school(), а не p.get_PI_school().
Я добавил _PI в свой код и оставил все как есть. Я использую его на Project, так как фильтрую проекты. Я хочу отфильтровать проекты, связанные с человеком с ролью = «Главный исследователь». Более того, я хочу дополнительные фильтры: проекты, связанные с school и subject этого человека. Надеюсь, это яснее.
@HBMCS: но в понимании списка вы используете p.get_school(), метод в Project — это get_PI_school, а для ProjectRole (то есть в понимании списка) это p.get_school().
Офт, ты прав! И это было в самом сообщении об ошибке. Дох. Хорошо, теперь я получаю правильную информацию в столбце таблицы (list_display). Теперь я хочу это в list_filter. Должен ли я определить class в admin.py?
@HBMCS: Я думаю, что, возможно, было бы лучше задать отдельный вопрос, чтобы другие могли повторно использовать вопросы и чтобы вопрос содержал достаточно элементов (и не слишком много) :).
Собственно самые первые строчки моего вопроса читаются Now, in my admin back-end I'd like to add a nice filtering with list_filter. :) .
Попробовав разные варианты, мне удалось сделать это с помощью 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()
[...]
Спасибо. Теперь я получаю
'PersonRole' object has no attribute 'get_PI_school'. Кроме того, полеtitleздесь не имеет значения: мне нужныsubjectиschool(теперь добавленные к моему вопросу, просто для ясности) человека с ролью «Главный следователь». Наконец, я хочу, чтобы все это использовалось вlist_filter, который принимает только поля и классы. Я изучаю пользовательские формы фильтров.